59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
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)
|