diff options
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..7c4c6a7a 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) | 
