diff options
author | Maximilian Hils <git@maximilianhils.com> | 2017-05-26 16:14:20 +0200 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2017-05-26 16:14:20 +0200 |
commit | eb5f37a7d19d74c88eed25d4b90f98bec1afa4be (patch) | |
tree | 9262e2549b8a3bd4cb2872041c8cc651d31c4d23 /examples/simple/io_write_dumpfile.py | |
parent | 9f58b190320515835af5e7ba75bd3b1fdd9445e0 (diff) | |
download | mitmproxy-eb5f37a7d19d74c88eed25d4b90f98bec1afa4be.tar.gz mitmproxy-eb5f37a7d19d74c88eed25d4b90f98bec1afa4be.tar.bz2 mitmproxy-eb5f37a7d19d74c88eed25d4b90f98bec1afa4be.zip |
fix mypy annotations
Diffstat (limited to 'examples/simple/io_write_dumpfile.py')
-rw-r--r-- | examples/simple/io_write_dumpfile.py | 10 |
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])] |