From 5e2a80fba1589604f7f310ea143ff24ea2dd0aa9 Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Mon, 12 Dec 2016 12:18:33 +1300 Subject: Minor fixes --- mitmproxy/tools/console/master.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/mitmproxy/tools/console/master.py b/mitmproxy/tools/console/master.py index 536dc03d..d8355324 100644 --- a/mitmproxy/tools/console/master.py +++ b/mitmproxy/tools/console/master.py @@ -386,11 +386,10 @@ class ConsoleMaster(master.Master): ) def _write_flows(self, path, flows): - f = open(path, "wb") - fw = io.FlowWriter(f) - for i in flows: - fw.add(i) - f.close() + with open(path, "wb") as f: + fw = io.FlowWriter(f) + for i in flows: + fw.add(i) def save_one_flow(self, path, flow): return self._write_flows(path, [flow]) -- cgit v1.2.3