hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-17 18:46:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix exception ignored warnings and invalid state #935

Browse Source
This commit is contained in:
Lonami Exo 2018-08-04 15:05:22 +02:00
parent d0c566bbdb
commit a688c8a4ce
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
telethon/extensions/tcpclient.py
View File

@ -151,7 +151,14 @@ class TcpClient:
loop=self._loop loop=self._loop
) )
for r in running: for r in running:
r.cancel() if not r.cancelled():
if r.done():
# Retrieve exception to avoid "not retrieved" errors
r.exception()
# Cancel the future despite its state
r.cancel()
if not self.is_connected: if not self.is_connected:
raise self.SocketClosed() raise self.SocketClosed()
if not done: if not done:

2
telethon/network/mtprotosender.py
View File

@ -167,7 +167,7 @@ class MTProtoSender:
self._recv_loop_handle.cancel() self._recv_loop_handle.cancel()
__log__.info('Disconnection from {} complete!'.format(self._ip)) __log__.info('Disconnection from {} complete!'.format(self._ip))
if self._disconnected: if self._disconnected and not self._disconnected.done():
if error: if error:
self._disconnected.set_exception(error) self._disconnected.set_exception(error)
else: else: