Add clear_handlers parameter in stop/restart methods

Author: KurimuzonAkuma

pyrogram/dispatcher.py

@@ -274,7 +274,7 @@ async def start(self):
             if not self.client.skip_updates:
                 await self.client.recover_gaps()
 
-    async def stop(self, clear: bool = True):
+    async def stop(self, clear_handlers: bool = True):
         if callable(self.client.stop_handler):
             try:
                 await self.client.stop_handler(self.client)
@@ -288,7 +288,7 @@ async def stop(self, clear: bool = True):
             for i in self.handler_worker_tasks:
                 await i
 
-            if clear:
+            if clear_handlers:
                 self.handler_worker_tasks.clear()
                 self.groups.clear()
 

pyrogram/methods/auth/terminate.py

@@ -27,12 +27,18 @@
 class Terminate:
     async def terminate(
         self: "pyrogram.Client",
+        clear_handlers: bool = True
     ):
         """Terminate the client by shutting down workers.
 
         This method does the opposite of :meth:`~pyrogram.Client.initialize`.
         It will stop the dispatcher and shut down updates and download workers.
 
+        Parameters:
+            clear_handlers (``bool``, *optional*):
+                Clear the already existing handlers on restart the client.
+                Default to True.
+
         Raises:
             ConnectionError: In case you try to terminate a client that is already terminated.
         """
@@ -44,7 +50,7 @@ async def terminate(
             log.info("Takeout session %s finished", self.takeout_id)
 
         await self.storage.save()
-        await self.dispatcher.stop()
+        await self.dispatcher.stop(clear_handlers=clear_handlers)
 
         for media_session in self.media_sessions.values():
             await media_session.stop()

pyrogram/methods/utilities/restart.py

@@ -22,7 +22,8 @@
 class Restart:
     async def restart(
         self: "pyrogram.Client",
-        block: bool = True
+        block: bool = True,
+        clear_handlers: bool = False
     ):
         """Restart the Client.
 
@@ -35,6 +36,10 @@ async def restart(
                 you want to restart the own client within an handler in order not to cause a deadlock.
                 Defaults to True.
 
+            clear_handlers (``bool``, *optional*):
+                Clear the already existing handlers on restart the client.
+                Default to False.
+
         Returns:
             :obj:`~pyrogram.Client`: The restarted client itself.
 
@@ -62,7 +67,7 @@ async def main():
         """
 
         async def do_it():
-            await self.stop()
+            await self.stop(clear_handlers=clear_handlers)
             await self.start()
 
         if block:

pyrogram/methods/utilities/stop.py

@@ -22,7 +22,8 @@
 class Stop:
     async def stop(
         self: "pyrogram.Client",
-        block: bool = True
+        block: bool = True,
+        clear_handlers: bool = True
     ):
         """Stop the Client.
 
@@ -34,6 +35,10 @@ async def stop(
                 you want to stop the own client *within* a handler in order not to cause a deadlock.
                 Defaults to True.
 
+            clear_handlers (``bool``, *optional*):
+                Clear the already existing handlers on stop the client.
+                Default to True.
+
         Returns:
             :obj:`~pyrogram.Client`: The stopped client itself.
 
@@ -60,7 +65,7 @@ async def main():
         """
 
         async def do_it():
-            await self.terminate()
+            await self.terminate(clear_handlers=clear_handlers)
             await self.disconnect()
 
         if block:

pyrogram/qrlogin.py

@@ -88,7 +88,7 @@ async def raw_handler(client, update, users, chats):
             await asyncio.wait_for(event.wait(), timeout=timeout)
         finally:
             self.client.remove_handler(*handler)
-            await self.client.dispatcher.stop(clear=False)
+            await self.client.dispatcher.stop(clear_handlers=False)
 
         r = await self.client.invoke(
             raw.functions.auth.ExportLoginToken(