diff options
author | Shadab Zafar <dufferzafar0@gmail.com> | 2016-02-08 20:07:38 +0530 |
---|---|---|
committer | Shadab Zafar <dufferzafar0@gmail.com> | 2016-02-08 20:18:25 +0530 |
commit | 3374be9f4b1dbeba3e9abbea7a447c957dfeda51 (patch) | |
tree | 9c7c9068b3643ddc84b86aa2518fd687d3b11f63 /libmproxy/flow_export.py | |
parent | 545014dcb6c2abd1e3b3113fb90aea17272ca116 (diff) | |
download | mitmproxy-3374be9f4b1dbeba3e9abbea7a447c957dfeda51.tar.gz mitmproxy-3374be9f4b1dbeba3e9abbea7a447c957dfeda51.tar.bz2 mitmproxy-3374be9f4b1dbeba3e9abbea7a447c957dfeda51.zip |
Move exporters to a separate file
Diffstat (limited to 'libmproxy/flow_export.py')
-rw-r--r-- | libmproxy/flow_export.py | 70 |
1 files changed, 70 insertions, 0 deletions
diff --git a/libmproxy/flow_export.py b/libmproxy/flow_export.py new file mode 100644 index 00000000..584b3be4 --- /dev/null +++ b/libmproxy/flow_export.py @@ -0,0 +1,70 @@ +import urllib +import netlib.http + + +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 = """import requests + +url = '{url}' +{headers}{params}{data} +response = requests.request( + method='{method}', + url=url,{args} +) + +print(response.text)""" + + 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 |