Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2025-02-07 02:28:50 +00:00
Code Issues Projects Releases Wiki Activity

Refactor copy_as_python_code

Browse Source 
Make it easier to support other HTTP request types
This commit is contained in:
Shadab Zafar 2016-02-05 23:25:25 +05:30
parent ae4a1dd6de
commit 9efda4d138
1 changed files with 22 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
libmproxy/console/common.py
View File

@ -309,20 +309,32 @@ def copy_as_python_code(flow):
signals.status_message.send(message="Currently, only GET methods are supported") signals.status_message.send(message="Currently, only GET methods are supported")
return return
data = ("import requests\n" code = """import requests
"headers = {%s}\n"
"url = '%s'\n"
"resp = requests.get(url, headers=headers)")
headers = "\n" url = '{url}'
for k, v in flow.request.headers.fields: {headers}
headers += " '%s': '%s',\n" % (k, v) response = requests.request(
method='{method}',
url=url,{args}
)
full_url = flow.request.scheme + "://" + flow.request.host + flow.request.path print(response.text)"""
data = data % (headers, full_url) 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,"
copy_to_clipboard_or_prompt(data) code = code.format(
url=flow.request.pretty_url,
headers=headers,
method=flow.request.method,
args=args,
)
copy_to_clipboard_or_prompt(code)
def copy_as_raw_request(flow): def copy_as_raw_request(flow):