diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2016-12-12 12:18:33 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2016-12-12 12:18:33 +1300 |
commit | 5e2a80fba1589604f7f310ea143ff24ea2dd0aa9 (patch) | |
tree | 0c0dd016713e19e6f7be121018d90743e6b8704f | |
parent | a4ac5b158f512206ae1bb1682052195b0082b01a (diff) | |
download | mitmproxy-5e2a80fba1589604f7f310ea143ff24ea2dd0aa9.tar.gz mitmproxy-5e2a80fba1589604f7f310ea143ff24ea2dd0aa9.tar.bz2 mitmproxy-5e2a80fba1589604f7f310ea143ff24ea2dd0aa9.zip |
Minor fixes
-rw-r--r-- | mitmproxy/tools/console/master.py | 9 |
1 files 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]) |