diff options
| author | Guillem Anguera <ganguera@gmail.com> | 2016-01-22 22:11:03 +0100 |
|---|---|---|
| committer | Guillem Anguera <ganguera@gmail.com> | 2016-01-22 22:11:03 +0100 |
| commit | b030ff090ac3a36b2e2bf386fa04c1453bd58140 (patch) | |
| tree | 2e0f13ae9b0825291d544a393949dcba837bb8ff /libmproxy/console/__init__.py | |
| parent | 4024721c7b1762325d90146044f2fdc01d07caa0 (diff) | |
| download | mitmproxy-b030ff090ac3a36b2e2bf386fa04c1453bd58140.tar.gz mitmproxy-b030ff090ac3a36b2e2bf386fa04c1453bd58140.tar.bz2 mitmproxy-b030ff090ac3a36b2e2bf386fa04c1453bd58140.zip | |
Added CLI argument to enable "follow flow list" mode
Diffstat (limited to 'libmproxy/console/__init__.py')
| -rw-r--r-- | libmproxy/console/__init__.py | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/libmproxy/console/__init__.py b/libmproxy/console/__init__.py index 31edca81..0df10256 100644 --- a/libmproxy/console/__init__.py +++ b/libmproxy/console/__init__.py @@ -105,25 +105,25 @@ class ConsoleState(flow.State): for f in self.flows: if self.flow_marked(f): marked_flows.append(f) - + super(ConsoleState, self).clear() - + for f in marked_flows: self.add_flow(f) self.set_flow_marked(f, True) - + if len(self.flows.views) == 0: self.focus = None else: self.focus = 0 self.set_focus(self.focus) - + def flow_marked(self, flow): return self.get_flow_setting(flow, "marked", False) - + def set_flow_marked(self, flow, marked): self.add_flow_setting(flow, "marked", marked) - + class Options(object): attributes = [ @@ -134,6 +134,7 @@ class Options(object): "anticomp", "client_replay", "eventlog", + "follow", "keepserving", "kill", "intercept", @@ -212,6 +213,7 @@ class ConsoleMaster(flow.FlowMaster): self.eventlog = options.eventlog self.eventlist = urwid.SimpleListWalker([]) + self.follow = options.follow if options.client_replay: self.client_playback_path(options.client_replay) @@ -562,6 +564,9 @@ class ConsoleMaster(flow.FlowMaster): else: body = flowlist.FlowListBox(self) + if self.follow: + self.toggle_follow_flows() + signals.push_view_state.send( self, window = window.Window( @@ -604,7 +609,7 @@ class ConsoleMaster(flow.FlowMaster): def save_flows(self, path): return self._write_flows(path, self.state.view) - + def save_marked_flows(self, path): marked_flows = [] for f in self.state.view: |