example_usage.py

#!/usr/bin/env python3
"""
Example usage of the GitHub Pull Request Creation API
This demonstrates the exact format you specified in your request.
"""

import requests
import json

def create_pull_request_example():
    """
    Example of creating a pull request with the exact format you specified:
    {
        "pr_title": "title",
        "body": "Pr body", 
        "files_and_content": {"/src/readme.md": "readme content"}
    }
    """
    
    # The exact format you specified
    data = {
        "pr_title": "Update README with new documentation",
        "body": "This PR adds comprehensive documentation for the new API endpoints and includes usage examples for developers.",
        "files_and_content": {
            "/src/readme.md": "# API Documentation\n\n## Overview\n\nThis API allows you to create pull requests programmatically.\n\n## Usage\n\n```bash\ncurl -X POST http://localhost:5000/create-pr \\\n  -H \"Content-Type: application/json\" \\\n  -d '{\n    \"pr_title\": \"Your Title\",\n    \"body\": \"Your description\",\n    \"files_and_content\": {\n      \"/path/to/file\": \"content\"\n    }\n  }'\n```\n\n## Features\n\n- Create PRs with multiple file changes\n- Automatic branch creation\n- Comprehensive error handling\n- RESTful API design"
        }
    }
    
    print("🚀 Creating Pull Request...")
    print(f"Title: {data['pr_title']}")
    print(f"Body: {data['body']}")
    print(f"Files to modify: {list(data['files_and_content'].keys())}")
    print("-" * 50)
    
    try:
        response = requests.post(
            "http://localhost:5000/create-pr",
            json=data,
            headers={"Content-Type": "application/json"}
        )
        
        print(f"Status Code: {response.status_code}")
        print(f"Response: {json.dumps(response.json(), indent=2)}")
        
        if response.status_code == 201:
            result = response.json()
            print("\n✅ Pull Request Created Successfully!")
            print(f"PR URL: {result['pr_url']}")
            print(f"PR Number: {result['pr_number']}")
            print(f"Branch Name: {result['branch_name']}")
            print(f"Commit SHA: {result['commit_sha']}")
        else:
            print(f"\n❌ Failed to create pull request: {response.json().get('error', 'Unknown error')}")
            
    except requests.exceptions.ConnectionError:
        print("❌ Error: Could not connect to server.")
        print("Make sure the server is running: python app.py")
    except Exception as e:
        print(f"❌ Error: {str(e)}")

def create_multiple_files_example():
    """
    Example of creating a pull request with multiple files
    """
    
    data = {
        "pr_title": "Add new feature with multiple files",
        "body": "This PR implements a new feature across multiple files including the main implementation, tests, and documentation.",
        "files_and_content": {
            "/src/new_feature.py": "def new_feature():\n    \"\"\"\n    A new feature that does something amazing.\n    \"\"\"\n    return \"Feature implemented!\"\n\n\ndef helper_function():\n    return \"Helper function for the new feature\"\n",
            "/tests/test_new_feature.py": "import pytest\nfrom src.new_feature import new_feature, helper_function\n\n\ndef test_new_feature():\n    assert new_feature() == \"Feature implemented!\"\n\n\ndef test_helper_function():\n    assert helper_function() == \"Helper function for the new feature\"\n",
            "/docs/new_feature.md": "# New Feature Documentation\n\n## Overview\n\nThis document describes the new feature that was implemented.\n\n## Usage\n\n```python\nfrom src.new_feature import new_feature\n\nresult = new_feature()\nprint(result)  # Output: Feature implemented!\n```\n\n## Testing\n\nRun the tests with:\n```bash\npytest tests/test_new_feature.py\n```"
        }
    }
    
    print("\n" + "=" * 60)
    print("🚀 Creating Pull Request with Multiple Files...")
    print(f"Title: {data['pr_title']}")
    print(f"Files to modify: {list(data['files_and_content'].keys())}")
    print("-" * 50)
    
    try:
        response = requests.post(
            "http://localhost:5000/create-pr",
            json=data,
            headers={"Content-Type": "application/json"}
        )
        
        print(f"Status Code: {response.status_code}")
        
        if response.status_code == 201:
            result = response.json()
            print("\n✅ Multi-file Pull Request Created Successfully!")
            print(f"PR URL: {result['pr_url']}")
            print(f"PR Number: {result['pr_number']}")
            print(f"Branch Name: {result['branch_name']}")
        else:
            print(f"\n❌ Failed to create pull request: {response.json().get('error', 'Unknown error')}")
            
    except requests.exceptions.ConnectionError:
        print("❌ Error: Could not connect to server.")
        print("Make sure the server is running: python app.py")
    except Exception as e:
        print(f"❌ Error: {str(e)}")

if __name__ == "__main__":
    print("GitHub Pull Request Creation API - Example Usage")
    print("=" * 60)
    
    # Check if server is running
    try:
        health_response = requests.get("http://localhost:5000/health")
        if health_response.status_code == 200:
            print("✅ Server is running!")
        else:
            print("❌ Server is not responding properly")
            exit(1)
    except requests.exceptions.ConnectionError:
        print("❌ Server is not running. Please start it with: python app.py")
        exit(1)
    
    # Run examples
    create_pull_request_example()
    create_multiple_files_example()
    
    print("\n" + "=" * 60)
    print("🎉 Examples completed!")
    print("Check your GitHub repository for the created pull requests.")