LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-09 08:00:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

Handle dispatcher CancelledError during disconnect

Browse Source
This commit is contained in:
Lonami Exo 2023-10-18 20:38:45 +02:00
parent c49f0827b1
commit 854d3745b3
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
client/src/telethon/_impl/client/client/net.py
View File

@ -210,6 +210,8 @@ async def disconnect(self: Client) -> None:
self._dispatcher.cancel() self._dispatcher.cancel()
try: try:
await self._dispatcher await self._dispatcher
except asyncio.CancelledError:
pass
except Exception: except Exception:
self._logger.exception( self._logger.exception(
"unhandled exception when cancelling dispatcher; this is a bug" "unhandled exception when cancelling dispatcher; this is a bug"

2
client/src/telethon/_impl/client/client/updates.py
View File

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