Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

suppress typing complaints in tests

Browse Source
This commit is contained in:
Maximilian Hils 2017-06-28 17:26:45 +02:00
parent 2a68029b21
commit d8621553b1
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mitmproxy/proxy/protocol2/events.py
View File

@ -58,10 +58,10 @@ class CommandReply(Event):
Emitted when a command has been finished, e.g.
when the master has replied or when we have established a server connection.
"""
command: commands.Command
command: typing.Union[commands.Command, int]
reply: typing.Any
def __init__(self, command: commands.Command, reply: typing.Any):
def __init__(self, command: typing.Union[commands.Command, int], reply: typing.Any):
self.command = command
self.reply = reply
@ -72,16 +72,20 @@ class CommandReply(Event):
class OpenConnectionReply(CommandReply):
command: commands.OpenConnection
command: typing.Union[commands.OpenConnection, int]
reply: str
def __init__(self, command: commands.OpenConnection, err: typing.Optional[str]):
def __init__(
self,
command: typing.Union[commands.OpenConnection, int],
err: typing.Optional[str]
):
super().__init__(command, err)
class HookReply(CommandReply):
command: commands.Hook
command: typing.Union[commands.Hook, int]
reply: typing.Any
def __init__(self, command: commands.Hook, reply: typing.Any):
def __init__(self, command: typing.Union[commands.Hook, int], reply: typing.Any):
super().__init__(command, reply)