diff options
Diffstat (limited to 'examples')
-rw-r--r-- | examples/simple/custom_contentview.py | 7 | ||||
-rw-r--r-- | examples/simple/io_read_dumpfile.py | 3 | ||||
-rw-r--r-- | examples/simple/io_write_dumpfile.py | 3 |
3 files changed, 9 insertions, 4 deletions
diff --git a/examples/simple/custom_contentview.py b/examples/simple/custom_contentview.py index 289e1efe..71f92575 100644 --- a/examples/simple/custom_contentview.py +++ b/examples/simple/custom_contentview.py @@ -3,7 +3,10 @@ This example shows how one can add a custom contentview to mitmproxy. The content view API is explained in the mitmproxy.contentviews module. """ from mitmproxy import contentviews -from typing import Tuple, Iterable, AnyStr, List +import typing + + +CVIEWSWAPCASE = typing.Tuple[str, typing.Iterable[typing.List[typing.Tuple[str, typing.AnyStr]]]] class ViewSwapCase(contentviews.View): @@ -14,7 +17,7 @@ class ViewSwapCase(contentviews.View): prompt = ("swap case text", "z") content_types = ["text/plain"] - def __call__(self, data: bytes, **metadata) -> Tuple[str, Iterable[List[Tuple[str, AnyStr]]]]: + def __call__(self, data: typing.AnyStr, **metadata) -> CVIEWSWAPCASE: return "case-swapped text", contentviews.format_text(data.swapcase()) diff --git a/examples/simple/io_read_dumpfile.py b/examples/simple/io_read_dumpfile.py index 87d37c0f..ea544cc4 100644 --- a/examples/simple/io_read_dumpfile.py +++ b/examples/simple/io_read_dumpfile.py @@ -1,8 +1,9 @@ #!/usr/bin/env python + +# type: ignore # # Simple script showing how to read a mitmproxy dump file # - from mitmproxy import io from mitmproxy.exceptions import FlowReadException import pprint diff --git a/examples/simple/io_write_dumpfile.py b/examples/simple/io_write_dumpfile.py index 7c4c6a7a..cf7c4f52 100644 --- a/examples/simple/io_write_dumpfile.py +++ b/examples/simple/io_write_dumpfile.py @@ -8,12 +8,13 @@ to multiple files in parallel. import random import sys from mitmproxy import io, http +import typing # noqa class Writer: def __init__(self, path: str) -> None: if path == "-": - f = sys.stdout # type: io.TextIO + f = sys.stdout # type: typing.IO[typing.Any] else: f = open(path, "wb") self.w = io.FlowWriter(f) |