Fix on_disconnect decorator not working (pyrogram#1134)

01101sam · web-flow · commit b423730dcbd5 · 2022-11-13T14:18:42.000+01:00
* Fix on_disconnect decorator not working

* Remove useless if else statment in on_raw_update
diff --git a/pyrogram/methods/decorators/on_disconnect.py b/pyrogram/methods/decorators/on_disconnect.py
@@ -32,6 +32,11 @@ def on_disconnect(self=None) -> Callable:
         def decorator(func: Callable) -> Callable:
             if isinstance(self, pyrogram.Client):
                 self.add_handler(pyrogram.handlers.DisconnectHandler(func))
+            else:
+                if not hasattr(func, "handlers"):
+                    func.handlers = []
+
+                func.handlers.append((pyrogram.handlers.DisconnectHandler(func), 0))
 
             return func
 
diff --git a/pyrogram/methods/decorators/on_raw_update.py b/pyrogram/methods/decorators/on_raw_update.py
@@ -46,7 +46,7 @@ def decorator(func: Callable) -> Callable:
                 func.handlers.append(
                     (
                         pyrogram.handlers.RawUpdateHandler(func),
-                        group if self is None else group
+                        group
                     )
                 )
 

Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,7 @@ def decorator(func: Callable) -> Callable:`
`46`	`46`	`func.handlers.append(`
`47`	`47`	`(`
`48`	`48`	`pyrogram.handlers.RawUpdateHandler(func),`
`49`		`- group if self is None else group`
	`49`	`+ group`
`50`	`50`	`)`
`51`	`51`	`)`
`52`	`52`