Merge pull request #4426 from mhils/mypy

update mypy

examples/addons/commands-flows.py

@@ -4,12 +4,14 @@ import typing
 from mitmproxy import command
 from mitmproxy import ctx
 from mitmproxy import flow
+from mitmproxy import http
 
 
 class MyAddon:
     @command.command("myaddon.addheader")
     def addheader(self, flows: typing.Sequence[flow.Flow]) -> None:
         for f in flows:
+            if isinstance(f, http.HTTPFlow):
                 f.request.headers["myheader"] = "value"
         ctx.log.alert("done")
 

examples/addons/commands-paths.py

@@ -4,6 +4,7 @@ import typing
 from mitmproxy import command
 from mitmproxy import ctx
 from mitmproxy import flow
+from mitmproxy import http
 from mitmproxy import types
 
 
@@ -14,8 +15,9 @@ class MyAddon:
         flows: typing.Sequence[flow.Flow],
         path: types.Path,
     ) -> None:
-        totals = {}
+        totals: typing.Dict[str, int] = {}
         for f in flows:
+            if isinstance(f, http.HTTPFlow):
                 totals[f.request.host] = totals.setdefault(f.request.host, 0) + 1
 
         with open(path, "w+") as fp:

examples/addons/http-trailers.py

@@ -36,6 +36,7 @@ def request(flow: http.HTTPFlow):
 
 
 def response(flow: http.HTTPFlow):
+    assert flow.response  # make type checker happy
     if flow.response.trailers:
         print("HTTP Trailers detected! Response contains:", flow.response.trailers)
 

examples/addons/internet_in_mirror.py

@@ -7,5 +7,6 @@ from mitmproxy import http
 
 
 def response(flow: http.HTTPFlow) -> None:
+    assert flow.response  # make type checker happy
     reflector = b"<style>body {transform: scaleX(-1);}</style></head>"
     flow.response.content = flow.response.content.replace(b"</head>", reflector)

examples/addons/io-read-saved-flows.py

@@ -2,7 +2,7 @@
 """
 Read a mitmproxy dump file.
 """
-from mitmproxy import io
+from mitmproxy import io, http
 from mitmproxy.exceptions import FlowReadException
 import pprint
 import sys
@@ -13,6 +13,7 @@ with open(sys.argv[1], "rb") as logfile:
     try:
         for f in freader.stream():
             print(f)
+            if isinstance(f, http.HTTPFlow):
                 print(f.request.host)
             pp.pprint(f.get_state())
             print("")

mitmproxy/certs.py

@@ -393,9 +393,9 @@ class CertStore:
 
         # Dump the certificate in PKCS12 format for Windows devices
         (path / f"{basename}-ca-cert.p12").write_bytes(
-            pkcs12.serialize_key_and_certificates(  # type: ignore
+            pkcs12.serialize_key_and_certificates(
                 name=basename.encode(),
-                key=None,
+                key=None,  # type: ignore
                 cert=ca,
                 cas=None,
                 encryption_algorithm=serialization.NoEncryption(),

mitmproxy/test/tflow.py

@@ -1,15 +1,13 @@
-import io
 import uuid
 
-from mitmproxy.test import tutils
+from mitmproxy import controller
+from mitmproxy import flow
+from mitmproxy import http
 from mitmproxy import tcp
 from mitmproxy import websocket
-from mitmproxy import controller
-from mitmproxy import http
-from mitmproxy import flow
 from mitmproxy.net import http as net_http
 from mitmproxy.proxy import context
-
+from mitmproxy.test import tutils
 from wsproto.frame_protocol import Opcode
 
 
@@ -34,7 +32,6 @@ def ttcpflow(client_conn=True, server_conn=True, messages=True, err=None):
 
 
 def twebsocketflow(client_conn=True, server_conn=True, messages=True, err=None, handshake_flow=True):
-
     if client_conn is True:
         client_conn = tclient_conn()
     if server_conn is True:
@@ -169,7 +166,7 @@ def tclient_conn() -> context.Client:
         alpn_offers=[],
         cipher_list=[],
     ))
-    c.reply = controller.DummyReply()
+    c.reply = controller.DummyReply()  # type: ignore
     return c
 
 
@@ -197,9 +194,7 @@ def tserver_conn() -> context.Server:
         cipher_list=[],
         via2=None,
     ))
-    c.reply = controller.DummyReply()
+    c.reply = controller.DummyReply()  # type: ignore
-    c.rfile = io.BytesIO()
-    c.wfile = io.BytesIO()
     return c
 
 

mitmproxy/test/tutils.py

@@ -21,7 +21,7 @@ def treq(**kwargs) -> http.Request:
         timestamp_end=946681201,
     )
     default.update(kwargs)
-    return http.Request(**default)
+    return http.Request(**default)  # type: ignore
 
 
 def tresp(**kwargs) -> http.Response:
@@ -40,4 +40,4 @@ def tresp(**kwargs) -> http.Response:
         timestamp_end=946681203,
     )
     default.update(kwargs)
-    return http.Response(**default)
+    return http.Response(**default)  # type: ignore

mitmproxy/tools/console/window.py

@@ -18,7 +18,7 @@ from mitmproxy.tools.console import statusbar
 if os.name == "nt":
     from mitmproxy.contrib.urwid import raw_display
 else:
-    from urwid import raw_display
+    from urwid import raw_display  # type: ignore
 
 
 class StackWidget(urwid.Frame):

setup.cfg

@@ -25,6 +25,7 @@ exclude_lines =
 
 [mypy]
 ignore_missing_imports = True
+files = mitmproxy,examples/addons,release
 
 [mypy-mitmproxy.contrib.*]
 ignore_errors = True

tox.ini

@@ -28,9 +28,9 @@ commands =
   python ./test/filename_matching.py
 
 [testenv:mypy]
-deps = mypy==0.790
+deps = mypy==0.800
 commands =
-  mypy . {posargs}
+  mypy {posargs}
   
 [testenv:individual_coverage]
 commands =