yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2025-12-31 20:54:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow on_disconnect to be used as a static decorator

Browse source
This commit is contained in:
Dan 2018-10-13 10:47:39 +02:00
parent 1fdc757f2a
commit 54296a6fda
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pyrogram/client/methods/decorators/on_disconnect.py
View file

@ -28,7 +28,11 @@ class OnDisconnect(BaseClient):
"""
def decorator(func):
self.add_handler(pyrogram.DisconnectHandler(func))
return func
handler = pyrogram.DisconnectHandler(func)
if self is not None:
self.add_handler(handler)
return handler
return decorator