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

Handle cancellation on the receive loop

Browse Source
This commit is contained in:
Lonami Exo 2018-10-03 15:31:49 +02:00
parent e319fa3aa9
commit 37b9922f64
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
telethon/network/connection/connection.py
View File

@ -113,6 +113,8 @@ class Connection(abc.ABC):
while not self._disconnected.is_set(): while not self._disconnected.is_set():
data = await self._recv() data = await self._recv()
await self._recv_queue.put((True, data)) await self._recv_queue.put((True, data))
except asyncio.CancelledError:
pass
except Exception as e: except Exception as e:
await self._recv_queue.put((False, e)) await self._recv_queue.put((False, e))
self.disconnect() self.disconnect()