hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-08 04:52:30 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make disconnect synchronous

Browse Source 
This also fixes a bug when auto-reconnecting in MTProtoSender.
This commit is contained in:
Lonami Exo
2018-10-16 11:56:17 +02:00
parent 477fbd8dc7
commit 9cbc088b76
6 changed files with 28 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
telethon/client/updates.py
View File

@@ -20,7 +20,7 @@ class UpdateMethods(UserMethods):
try:
await self.disconnected
except KeyboardInterrupt:
await self.disconnect()
self.disconnect()
def run_until_disconnected(self):
"""
@@ -37,12 +37,7 @@ class UpdateMethods(UserMethods):
try:
return self.loop.run_until_complete(self.disconnected)
except KeyboardInterrupt:
# Importing the magic sync module turns disconnect into sync.
# TODO Maybe disconnect() should not need the magic module...
if inspect.iscoroutinefunction(self.disconnect):
self.loop.run_until_complete(self.disconnect())
else:
self.disconnect()
self.disconnect()
def on(self, event):
"""
@@ -223,7 +218,7 @@ class UpdateMethods(UserMethods):
except asyncio.TimeoutError:
pass
except asyncio.CancelledError:
await self.disconnect()
self.disconnect()
return
except Exception as e:
continue # Any disconnected exception should be ignored