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

Keep focus on same flow when disabling marked view

Browse Source
This commit is contained in:
Shadab Zafar 2016-07-21 15:54:24 +05:30
parent 426a62299c
commit 484dbd8a7b
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mitmproxy/console/master.py
View File

@ -155,7 +155,7 @@ class ConsoleState(flow.State):
self.last_filter = self.limit_txt self.last_filter = self.limit_txt
self.set_limit(marked_filter) self.set_limit(marked_filter)
# Set Focus # Restore Focus
if last_focus.marked: if last_focus.marked:
self.set_focus_flow(last_focus) self.set_focus_flow(last_focus)
else: else:
@ -164,10 +164,22 @@ class ConsoleState(flow.State):
self.mark_filter = True self.mark_filter = True
def disable_marked_filter(self): def disable_marked_filter(self):
marked_filter = "~%s" % FMarked.code
# Save Focus
last_focus, _ = self.get_focus()
nearest_marked = self.get_nearest_matching_flow(last_focus, marked_filter)
self.set_limit(self.last_filter) self.set_limit(self.last_filter)
self.last_filter = "" self.last_filter = ""
# Restore Focus
if last_focus.marked:
self.set_focus_flow(last_focus)
else:
self.set_focus_flow(nearest_marked)
self.mark_filter = False self.mark_filter = False
self.set_focus(0)
def clear(self): def clear(self):
marked_flows = [f for f in self.state.view if f.marked] marked_flows = [f for f in self.state.view if f.marked]