diff options
author | harsh vijay <iharsh234@gmail.com> | 2017-04-29 03:26:14 +0530 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-04-29 03:26:14 +0530 |
commit | 36118973d98f1afa81e79a846f595a7c4782ff9a (patch) | |
tree | 36649bb0b9737548b9b33fd7eac2c359ce1d2d83 /examples/simple/io_write_dumpfile.py | |
parent | b537997f4f5d7c33562414a414913b6cd89f99c3 (diff) | |
download | mitmproxy-36118973d98f1afa81e79a846f595a7c4782ff9a.tar.gz mitmproxy-36118973d98f1afa81e79a846f595a7c4782ff9a.tar.bz2 mitmproxy-36118973d98f1afa81e79a846f595a7c4782ff9a.zip |
extend mypy example/simple
Diffstat (limited to 'examples/simple/io_write_dumpfile.py')
-rw-r--r-- | examples/simple/io_write_dumpfile.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/examples/simple/io_write_dumpfile.py b/examples/simple/io_write_dumpfile.py index a0956e33..04d943c0 100644 --- a/examples/simple/io_write_dumpfile.py +++ b/examples/simple/io_write_dumpfile.py @@ -7,18 +7,18 @@ to multiple files in parallel. """ import random import sys -from mitmproxy import io +from mitmproxy import io, http class Writer: - def __init__(self, path): + def __init__(self, path: str) -> None: if path == "-": - f = sys.stdout + f = sys.stdout # type: io.TextIO else: f = open(path, "wb") self.w = io.FlowWriter(f) - def response(self, flow): + def response(self, flow: http.HTTPFlow) -> None: if random.choice([True, False]): self.w.add(flow) |