Merge pull request #2358 from mhils/make-mypy-great-again

Fix mypy annotations, update mypy

1 files changed, 5 insertions, 5 deletions

diff --git a/examples/simple/io_write_dumpfile.py b/examples/simple/io_write_dumpfile.py
index cf7c4f52..be6e4121 100644
--- a/examples/simple/io_write_dumpfile.py
+++ b/examples/simple/io_write_dumpfile.py
@@ -13,15 +13,15 @@ import typing  # noqa
 
 class Writer:
     def __init__(self, path: str) -> None:
-        if path == "-":
-            f = sys.stdout  # type: typing.IO[typing.Any]
-        else:
-            f = open(path, "wb")
-        self.w = io.FlowWriter(f)
+        self.f = open(path, "wb")  # type: typing.IO[bytes]
+        self.w = io.FlowWriter(self.f)
 
     def response(self, flow: http.HTTPFlow) -> None:
         if random.choice([True, False]):
             self.w.add(flow)
 
+    def done(self):
+        self.f.close()
+
 
 addons = [Writer(sys.argv[1])]