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

Fix edit prompt display, return code editing.

Browse Source
This commit is contained in:
Aldo Cortesi 2011-06-30 14:30:48 +12:00
parent 18d4c3a9e9
commit 37c8d3425d
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libmproxy/console.py
View File

@ -372,7 +372,10 @@ class ConnectionView(WWrap):
def set_resp_code(self, code): def set_resp_code(self, code):
response = self.flow.response response = self.flow.response
response.code = code try:
response.code = int(code)
except ValueError:
return None
import BaseHTTPServer import BaseHTTPServer
if BaseHTTPServer.BaseHTTPRequestHandler.responses.has_key(int(code)): if BaseHTTPServer.BaseHTTPRequestHandler.responses.has_key(int(code)):
response.msg = BaseHTTPServer.BaseHTTPRequestHandler.responses[int(code)][0] response.msg = BaseHTTPServer.BaseHTTPRequestHandler.responses[int(code)][0]
@ -403,7 +406,7 @@ class ConnectionView(WWrap):
elif part == "m" and self.state.view_flow_mode == VIEW_FLOW_REQUEST: elif part == "m" and self.state.view_flow_mode == VIEW_FLOW_REQUEST:
self.master.prompt_onekey("Method", self.methods, self.edit_method) self.master.prompt_onekey("Method", self.methods, self.edit_method)
elif part == "c" and self.state.view_flow_mode == VIEW_FLOW_RESPONSE: elif part == "c" and self.state.view_flow_mode == VIEW_FLOW_RESPONSE:
self.master.prompt_edit("Code", conn.code, self.set_resp_code) self.master.prompt_edit("Code", str(conn.code), self.set_resp_code)
elif part == "m" and self.state.view_flow_mode == VIEW_FLOW_RESPONSE: elif part == "m" and self.state.view_flow_mode == VIEW_FLOW_RESPONSE:
self.master.prompt_edit("Message", conn.msg, self.set_resp_msg) self.master.prompt_edit("Message", conn.msg, self.set_resp_msg)
elif part == "r" and self.state.view_flow_mode == VIEW_FLOW_REQUEST: elif part == "r" and self.state.view_flow_mode == VIEW_FLOW_REQUEST:
@ -1290,7 +1293,7 @@ class ConsoleMaster(flow.FlowMaster):
self.prompting = (callback, args) self.prompting = (callback, args)
def prompt_edit(self, prompt, text, callback): def prompt_edit(self, prompt, text, callback):
self.statusbar.prompt(prompt, text) self.statusbar.prompt(prompt + ": ", text)
self.view.set_focus("footer") self.view.set_focus("footer")
self.prompting = (callback, []) self.prompting = (callback, [])