Handle dispatcher CancelledError during disconnect

LonamiWebs · Oct 18, 2023 · 854d374 · 854d374
1 parent c49f082
commit 854d374
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/client/src/telethon/_impl/client/client/net.py b/client/src/telethon/_impl/client/client/net.py
@@ -210,6 +210,8 @@ async def disconnect(self: Client) -> None:
     self._dispatcher.cancel()
     try:
         await self._dispatcher
+    except asyncio.CancelledError:
+        pass
     except Exception:
         self._logger.exception(
             "unhandled exception when cancelling dispatcher; this is a bug"

diff --git a/client/src/telethon/_impl/client/client/updates.py b/client/src/telethon/_impl/client/client/updates.py
@@ -130,7 +130,7 @@ async def dispatcher(client: Client) -> None:
         try:
             await dispatch_next(client)
         except asyncio.CancelledError:
-            return
+            raise
         except Exception as e:
             if isinstance(e, RuntimeError) and loop.is_closed():
                 # User probably forgot to call disconnect.