Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 02:10:59 +00:00
Code Issues Projects Releases Wiki Activity

Add command to write request or response body to file.

Browse Source
This commit is contained in:
Aldo Cortesi 2011-01-27 15:14:46 +13:00
parent 345ac0f2a0
commit edab82f7d8
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
libmproxy/console.py
View File

@ -282,6 +282,21 @@ class ConnectionView(WWrap):
self.flow.request.method = i[0].upper()
self.master.refresh_connection(self.flow)
def save_body(self, path):
if not path:
return
if self.viewing == self.REQ:
c = self.flow.request
else:
c = self.flow.response
path = os.path.expanduser(path)
try:
f = file(path, "wb")
f.write(str(c.content))
f.close()
except IOError, v:
self.master.statusbar.message(str(v))
def edit(self, part):
if self.viewing == self.REQ:
conn = self.flow.request
@ -391,6 +406,11 @@ class ConnectionView(WWrap):
self.master.ui._curs_set(1)
self.master.ui.clear()
os.unlink(name)
elif key == "w":
if self.viewing == self.REQ:
self.master.prompt("Save request body: ", self.save_body)
else:
self.master.prompt("Save response body: ", self.save_body)
return key
@ -815,6 +835,7 @@ class ConsoleMaster(controller.Master):
("e", "edit response/request"),
("s", "save this flow"),
("v", "view contents in external viewer"),
("w", "save request or response body"),
("tab", "toggle response/request view"),
]
text.extend(format_keyvals(keys, key="key", val="text", indent=4))