kylefang/mitmproxy
1
0
Fork 0
mirror of https://github.com/zhigang1992/mitmproxy.git synced 2026-05-28 23:41:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

mitmproxy prompted input now display previously set value.

Browse Source 
E.g. if you set a limit, then re-enter the limit prompt, you start with the
currently set value.
This commit is contained in:
Aldo Cortesi
2011-03-13 11:24:49 +13:00
parent 9a9a2d9125
commit c901cbbada
4 changed files with 63 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
libmproxy/console.py
View File

@@ -165,7 +165,12 @@ class ConnectionItem(WWrap):
self.state.revert(self.flow)
self.master.sync_list_view()
elif key == "w":
self.master.prompt("Save this flow: ", self.master.save_one_flow, self.flow)
self.master.prompt(
"Save this flow: ",
self.state.last_saveload,
self.master.save_one_flow,
self.flow
)
elif key == "z":
self.master.kill_connection(self.flow)
elif key == "enter":
@@ -384,6 +389,7 @@ class ConnectionView(WWrap):
def save_body(self, path):
if not path:
return
self.state.last_saveload = path
if self.state.view_flow_mode == VIEW_FLOW_REQUEST:
c = self.flow.request
else:
@@ -516,7 +522,12 @@ class ConnectionView(WWrap):
self.state.revert(self.flow)
self.master.refresh_connection(self.flow)
elif key == "w":
self.master.prompt("Save this flow: ", self.master.save_one_flow, self.flow)
self.master.prompt(
"Save this flow: ",
self.state.last_saveload,
self.master.save_one_flow,
self.flow
)
elif key == "v":
if self.state.view_flow_mode == VIEW_FLOW_REQUEST:
conn = self.flow.request
@@ -528,9 +539,17 @@ class ConnectionView(WWrap):
self.master.spawn_external_viewer(conn.content, t)
elif key == "b":
if self.state.view_flow_mode == VIEW_FLOW_REQUEST:
self.master.prompt("Save request body: ", self.save_body)
self.master.prompt(
"Save request body: ",
self.state.last_saveload,
self.save_body
)
else:
self.master.prompt("Save response body: ", self.save_body)
self.master.prompt(
"Save response body: ",
self.state.last_saveload,
self.save_body
)
elif key == " ":
self.master.view_next_flow(self.flow)
elif key == "|":
@@ -632,7 +651,7 @@ class ActionBar(WWrap):
self.w = PathEdit(prompt, text)
def prompt(self, prompt, text = ""):
self.w = urwid.Edit(prompt, text)
self.w = urwid.Edit(prompt, text or "")
def message(self, message):
self.w = urwid.Text(message)
@@ -1100,8 +1119,8 @@ class ConsoleMaster(flow.FlowMaster):
self.view.set_focus("footer")
self.prompting = (callback, args)
def prompt(self, prompt, callback, *args):
self.statusbar.prompt(prompt)
def prompt(self, prompt, text, callback, *args):
self.statusbar.prompt(prompt, text)
self.view.set_focus("footer")
self.prompting = (callback, args)
@@ -1129,7 +1148,7 @@ class ConsoleMaster(flow.FlowMaster):
prompt.extend(mkup)
prompt.append(")? ")
self.onekey = "".join([i[1] for i in keys])
self.prompt(prompt, callback)
self.prompt(prompt, "", callback)
def prompt_done(self):
self.prompting = False
@@ -1153,21 +1172,10 @@ class ConsoleMaster(flow.FlowMaster):
self.state.accept_all()
def set_limit(self, txt):
if txt:
f = filt.parse(txt)
if not f:
return "Invalid filter expression."
self.state.set_limit(f)
else:
self.state.set_limit(None)
return self.state.set_limit(txt)
def set_intercept(self, txt):
if txt:
self.state.intercept = filt.parse(txt)
if not self.state.intercept:
return "Invalid filter expression."
else:
self.state.intercept = None
return self.state.set_intercept(txt)
def set_beep(self, txt):
if txt:
@@ -1214,15 +1222,19 @@ class ConsoleMaster(flow.FlowMaster):
if k == "?":
self.view_help()
elif k == "l":
self.prompt("Limit: ", self.set_limit)
self.prompt("Limit: ", self.state.limit_txt, self.set_limit)
self.sync_list_view()
k = None
elif k == "i":
self.prompt("Intercept: ", self.set_intercept)
self.prompt(
"Intercept: ",
self.state.intercept_txt,
self.set_intercept
)
self.sync_list_view()
k = None
elif k == "B":
self.prompt("Beep: ", self.set_beep)
self.prompt("Beep: ", "", self.set_beep)
k = None
elif k == "j":
k = "down"