91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
{"""Python code generator."""
|
|
|
|
import json
|
|
import base64
|
|
from curlconverter.parser import ParsedCurl
|
|
from curlconverter.generators import register_generator
|
|
|
|
|
|
def _detect_content_type(headers: dict, data: str) -> str:
|
|
"""Detect content type from headers or data."""
|
|
if "Content-Type" in headers:
|
|
return headers["Content-Type"]
|
|
if "content-type" in headers:
|
|
return headers["content-type"]
|
|
if data:
|
|
try:
|
|
json.loads(data)
|
|
return "application/json"
|
|
except (json.JSONDecodeError, TypeError):
|
|
pass
|
|
return "application/x-www-form-urlencoded"
|
|
|
|
|
|
@register_generator("python")
|
|
def generate_python(parsed: ParsedCurl) -> str:
|
|
"""Generate Python requests code from parsed curl data."""
|
|
lines = []
|
|
lines.append("import requests")
|
|
lines.append("")
|
|
lines.append("url = " + repr(parsed.url))
|
|
lines.append("")
|
|
|
|
kwargs = []
|
|
headers = dict(parsed.headers) if parsed.headers else {}
|
|
|
|
if parsed.user_agent and "User-Agent" not in headers and "user-agent" not in headers:
|
|
headers["User-Agent"] = parsed.user_agent
|
|
|
|
if headers:
|
|
lines.append("headers = " + _format_dict(headers))
|
|
kwargs.append("headers=headers")
|
|
|
|
if parsed.auth:
|
|
if ":" in parsed.auth[0]:
|
|
try:
|
|
encoded = base64.b64encode(f"{parsed.auth[0]}:{parsed.auth[1]}".encode()).decode()
|
|
if "Authorization" not in headers:
|
|
headers["Authorization"] = f"Basic {encoded}"
|
|
except Exception:
|
|
pass
|
|
|
|
if parsed.data:
|
|
content_type = _detect_content_type(headers, parsed.data)
|
|
|
|
if content_type == "application/json":
|
|
try:
|
|
json_data = json.loads(parsed.data)
|
|
lines.append("data = " + json.dumps(json_data, indent=4))
|
|
kwargs.append("json=data")
|
|
except (json.JSONDecodeError, TypeError):
|
|
lines.append("data = " + repr(parsed.data))
|
|
kwargs.append("data=data")
|
|
else:
|
|
lines.append("data = " + repr(parsed.data))
|
|
kwargs.append("data=data")
|
|
|
|
if parsed.cookies:
|
|
lines.append("cookies = " + repr(parsed.cookies))
|
|
kwargs.append("cookies=cookies")
|
|
|
|
if kwargs:
|
|
kwargs_str = ",\n ".join(kwargs)
|
|
lines.append(f"response = requests.{parsed.method.lower()}(url, {kwargs_str})")
|
|
else:
|
|
lines.append(f"response = requests.{parsed.method.lower()}(url)")
|
|
|
|
lines.append("")
|
|
lines.append("print(response.status_code)")
|
|
lines.append("print(response.text)")
|
|
|
|
return "\n".join(lines)
|
|
|
|
|
|
def _format_dict(d: dict) -> str:
|
|
"""Format a dictionary for Python code."""
|
|
if not d:
|
|
return "{}"
|
|
items = []
|
|
for k, v in d.items():
|
|
items.append(f" {repr(k)}: {repr(v)}")
|
|
return "{\n" + ",\n".join(items) + "\n}" |