diff options
Diffstat (limited to 'mitmproxy/eventsequence.py')
-rw-r--r-- | mitmproxy/eventsequence.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mitmproxy/eventsequence.py b/mitmproxy/eventsequence.py index d263727b..ef3b6db4 100644 --- a/mitmproxy/eventsequence.py +++ b/mitmproxy/eventsequence.py @@ -80,11 +80,11 @@ def _iterate_tcp(f: tcp.TCPFlow) -> TEventGenerator: yield "tcp_end", f -_iterate_map = { +_iterate_map: typing.Dict[typing.Type[flow.Flow], typing.Callable[[typing.Any], TEventGenerator]] = { http.HTTPFlow: _iterate_http, websocket.WebSocketFlow: _iterate_websocket, tcp.TCPFlow: _iterate_tcp, -} # type: typing.Dict[typing.Type[flow.Flow], typing.Callable[[typing.Any], TEventGenerator]] +} def iterate(f: flow.Flow) -> TEventGenerator: |