Allow on_disconnect to be used as a static decorator

delivrance · delivrance · commit 54296a6fdaa9 · 2018-10-13T10:47:39.000+02:00
diff --git a/pyrogram/client/methods/decorators/on_disconnect.py b/pyrogram/client/methods/decorators/on_disconnect.py
@@ -28,7 +28,11 @@ def on_disconnect(self):
         """
 
         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
