diff options
| author | Shadab Zafar <dufferzafar0@gmail.com> | 2016-07-21 11:00:33 +0530 |
|---|---|---|
| committer | Shadab Zafar <dufferzafar0@gmail.com> | 2016-07-23 10:41:57 +0530 |
| commit | 4a3e9c0563a20d82da31d6b80f961b938029786d (patch) | |
| tree | b030a5e14ab4d4f4e98b12c92378e3c39bbb98e7 | |
| parent | 40b50e9398eeaf2bdcc565abb872f1bbb1ff1292 (diff) | |
| download | mitmproxy-4a3e9c0563a20d82da31d6b80f961b938029786d.tar.gz mitmproxy-4a3e9c0563a20d82da31d6b80f961b938029786d.tar.bz2 mitmproxy-4a3e9c0563a20d82da31d6b80f961b938029786d.zip | |
Keep focus on same flow when enabling marked view
...if it is marked.
| -rw-r--r-- | mitmproxy/console/master.py | 20 |
1 files changed, 15 insertions, 5 deletions
diff --git a/mitmproxy/console/master.py b/mitmproxy/console/master.py index 271fd64a..7570d2f7 100644 --- a/mitmproxy/console/master.py +++ b/mitmproxy/console/master.py @@ -124,14 +124,24 @@ class ConsoleState(flow.State): return ret def enable_marked_filter(self): - self.last_filter = self.limit_txt marked_flows = [f for f in self.flows if f.marked] - if len(marked_flows) > 0: - self.last_filter = self.limit_txt - self.set_limit("~%s" % FMarked.code) - self.mark_filter = True + if not marked_flows: + return + + # Save Focus + last_focus, _ = self.get_focus() + + self.last_filter = self.limit_txt + self.set_limit("~%s" % FMarked.code) + + # Set Focus + if last_focus.marked: + self.set_focus_flow(last_focus) + else: self.set_focus(0) + self.mark_filter = True + def disable_marked_filter(self): self.set_limit(self.last_filter) self.last_filter = "" |