From d99d61e8ef197744a448d767194d57de8b6209f5 Mon Sep 17 00:00:00 2001 From: Shadab Zafar Date: Tue, 19 Jul 2016 12:21:39 +0530 Subject: Update save_marked_flows to use the new attribute --- mitmproxy/console/master.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/mitmproxy/console/master.py b/mitmproxy/console/master.py index fad4c375..b89d2368 100644 --- a/mitmproxy/console/master.py +++ b/mitmproxy/console/master.py @@ -616,10 +616,7 @@ class ConsoleMaster(flow.FlowMaster): return self._write_flows(path, self.state.view) def save_marked_flows(self, path): - marked_flows = [] - for f in self.state.view: - if self.state.flow_marked(f): - marked_flows.append(f) + marked_flows = [f for f in self.state.view if f.marked] return self._write_flows(path, marked_flows) def load_flows_callback(self, path): -- cgit v1.2.3