From 37b9922f6482e01ea76e8dd6ff6374dfd5eedbb0 Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Wed, 3 Oct 2018 15:31:49 +0200 Subject: [PATCH] Handle cancellation on the receive loop --- telethon/network/connection/connection.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/telethon/network/connection/connection.py b/telethon/network/connection/connection.py index 661d2de2..5d650a38 100644 --- a/telethon/network/connection/connection.py +++ b/telethon/network/connection/connection.py @@ -113,6 +113,8 @@ class Connection(abc.ABC): while not self._disconnected.is_set(): data = await self._recv() await self._recv_queue.put((True, data)) + except asyncio.CancelledError: + pass except Exception as e: await self._recv_queue.put((False, e)) self.disconnect()