alemi/aiocraft
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix for on_packet

Browse source
This commit is contained in:
əlemi 2021-11-17 17:09:14 +01:00
parent cd402f668a
commit 54f1c1d760
1 changed files with 4 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
aiocraft/client.py
View file

@ -76,18 +76,11 @@ class Client:
def connected(self) -> bool: def connected(self) -> bool:
return self.started and self.dispatcher.connected return self.started and self.dispatcher.connected
def on(self, hook): def on_packet(self, packet:Type[Packet], *args) -> Callable: # receive *args for retro compatibility
def wrapper(fun): def wrapper(fun):
pass # TODO if packet not in self._packet_callbacks:
return wrapper self._packet_callbacks[packet] = []
self._packet_callbacks[packet].append(fun)
def on_packet(self, packet:Type[Packet], state:ConnectionState) -> Callable:
def wrapper(fun):
if state not in self._packet_callbacks:
self._packet_callbacks[state] = {}
if packet not in self._packet_callbacks[state]:
self._packet_callbacks[state][packet] = []
self._packet_callbacks[state][packet].append(fun)
return fun return fun
return wrapper return wrapper