diff options
| -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]) |