mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 03:14:22 +00:00
74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
import urllib
|
|
import netlib.http
|
|
from textwrap import dedent
|
|
|
|
|
|
def curl_command(flow):
|
|
data = "curl "
|
|
|
|
for k, v in flow.request.headers.fields:
|
|
data += "-H '%s:%s' " % (k, v)
|
|
|
|
if flow.request.method != "GET":
|
|
data += "-X %s " % flow.request.method
|
|
|
|
full_url = flow.request.scheme + "://" + flow.request.host + flow.request.path
|
|
data += "'%s'" % full_url
|
|
|
|
if flow.request.content:
|
|
data += " --data-binary '%s'" % flow.request.content
|
|
|
|
return data
|
|
|
|
|
|
def python_code(flow):
|
|
code = dedent("""
|
|
import requests
|
|
|
|
url = '{url}'
|
|
{headers}{params}{data}
|
|
response = requests.request(
|
|
method='{method}',
|
|
url=url,{args}
|
|
)
|
|
|
|
print(response.text)
|
|
""").strip()
|
|
|
|
components = map(lambda x: urllib.quote(x, safe=""), flow.request.path_components)
|
|
url = flow.request.scheme + "://" + flow.request.host + "/" + "/".join(components)
|
|
|
|
args = ""
|
|
headers = ""
|
|
if flow.request.headers:
|
|
lines = [" '%s': '%s',\n" % (k, v) for k, v in flow.request.headers.fields]
|
|
headers += "\nheaders = {\n%s}\n" % "".join(lines)
|
|
args += "\n headers=headers,"
|
|
|
|
params = ""
|
|
if flow.request.query:
|
|
lines = [" '%s': '%s',\n" % (k, v) for k, v in flow.request.query]
|
|
params = "\nparams = {\n%s}\n" % "".join(lines)
|
|
args += "\n params=params,"
|
|
|
|
data = ""
|
|
if flow.request.body:
|
|
data = "\ndata = '''%s'''\n" % flow.request.body
|
|
args += "\n data=data,"
|
|
|
|
code = code.format(
|
|
url=url,
|
|
headers=headers,
|
|
params=params,
|
|
data=data,
|
|
method=flow.request.method,
|
|
args=args,
|
|
)
|
|
|
|
return code
|
|
|
|
|
|
def raw_request(flow):
|
|
data = netlib.http.http1.assemble_request(flow.request)
|
|
return data
|