diff options
| author | Marcelo Glezer <mg@tekii.com.ar> | 2015-01-15 12:57:00 -0300 |
|---|---|---|
| committer | Marcelo Glezer <mg@tekii.com.ar> | 2015-02-05 11:31:24 -0300 |
| commit | 3bcf0cbba894ad12d4cfc438d26cdf22d640416c (patch) | |
| tree | 8c43f10386e1de6d35c4e0cfd2b02821356b4056 /libmproxy/console/flowview.py | |
| parent | 8f66fe1bfc16d0bb63acc267632471640939371e (diff) | |
| download | mitmproxy-3bcf0cbba894ad12d4cfc438d26cdf22d640416c.tar.gz mitmproxy-3bcf0cbba894ad12d4cfc438d26cdf22d640416c.tar.bz2 mitmproxy-3bcf0cbba894ad12d4cfc438d26cdf22d640416c.zip | |
used save_body from common when b key is pressed
Diffstat (limited to 'libmproxy/console/flowview.py')
| -rw-r--r-- | libmproxy/console/flowview.py | 41 |
1 files changed, 14 insertions, 27 deletions
diff --git a/libmproxy/console/flowview.py b/libmproxy/console/flowview.py index bc0b5f83..b964b92a 100644 --- a/libmproxy/console/flowview.py +++ b/libmproxy/console/flowview.py @@ -508,22 +508,6 @@ class FlowView(common.WWrap): self.flow.request.method = i[0].upper() self.master.refresh_flow(self.flow) - def save_body(self, path): - if not path: - return - self.state.last_saveload = path - if self.state.view_flow_mode == common.VIEW_FLOW_REQUEST: - 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(v.strerror) - def set_url(self, url): request = self.flow.request try: @@ -691,17 +675,20 @@ class FlowView(common.WWrap): elif key == "b": if conn: if self.state.view_flow_mode == common.VIEW_FLOW_REQUEST: - self.master.path_prompt( - "Save request body: ", - self.state.last_saveload, - self.save_body - ) - else: - self.master.path_prompt( - "Save response body: ", - self.state.last_saveload, - self.save_body - ) + msg = "Save request body: " + content = self.flow.request.content + else: + msg = "Save response body: " + content = self.flow.response.content + + self.master.path_prompt( + msg, + self.state.last_saveload, + common.save_body, + self.master, + self.state, + content, + ) elif key == "d": if self.state.flow_count() == 1: self.master.view_flowlist() |