from ..parser.models import ParsedCurl


def generate_python(parsed: ParsedCurl) -> str:
    """Generate Python requests code from parsed curl command."""
    lines = [
        "import requests",
        "from requests.auth import HTTPBasicAuth",
        "import json",
        "",
        "",
        "def make_request():",
        f'    url = "{parsed.url}"',
        "",
    ]
    
    if parsed.auth and parsed.auth.username and parsed.auth.password:
        lines.append(f'    auth = HTTPBasicAuth("{parsed.auth.username}", "{parsed.auth.password}")')
    else:
        lines.append("    auth = None")
    
    headers_dict = {h.key: h.value for h in parsed.headers}
    if parsed.user_agent:
        headers_dict["User-Agent"] = parsed.user_agent
    
    lines.append(f"    headers = {headers_dict}")
    lines.append("")
    
    method_name = parsed.method.value.lower()
    if parsed.data:
        if parsed.data.data_type.value == "json":
            lines.append(f"    json_data = {parsed.data.json_data}")
            lines.append(f"    response = requests.{method_name}(url, json=json_data, headers=headers, auth=auth)")
        else:
            lines.append(f"    data = {repr(parsed.data.content)}")
            lines.append(f"    response = requests.{method_name}(url, data=data, headers=headers, auth=auth)")
    else:
        lines.append(f"    response = requests.{method_name}(url, headers=headers, auth=auth)")
    
    if parsed.timeout:
        lines[-1] = lines[-1][:-1] + f", timeout={parsed.timeout})")
    
    lines.extend([
        "",
        "    try:",
        "        response.raise_for_status()",
        "        print(response.json())",
        "    except requests.exceptions.RequestException as e:",
        '        print(f"Error: {e}")',
        "    except ValueError as e:",
        '        print(f"JSON decode error: {e}")',
        "",
        "",
        "if __name__ == \"__main__\":",
        "    make_request()",
    ])
    
    return "\n".join(lines)