LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-22 01:16:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Handle connection errors during auth key generation

Browse Source 
This should help with spurious server-side disconnects during
auth_key generation, which happen most commonly on user DC
migrations.
This commit is contained in:
Lonami Exo 2019-10-24 13:48:29 +02:00
parent 08b78f0c47
commit 5dcc30dcc6
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
telethon/network/mtprotosender.py
View File

@ -206,8 +206,23 @@ class MTProtoSender:
continue # skip auth key generation until we're connected
if not self.auth_key:
if not await self._try_gen_auth_key(attempt):
continue # keep retrying until we have the auth key
try:
if not await self._try_gen_auth_key(attempt):
continue # keep retrying until we have the auth key
except (IOError, asyncio.TimeoutError) as e:
# Sometimes, specially during user-DC migrations,
# Telegram may close the connection during auth_key
# generation. If that's the case, we will need to
# connect again.
self._log.warning('Connection error %d during auth_key gen: %s: %s',
attempt, type(e).__name__, e)
# Whatever the IOError was, make sure to disconnect so we can
# reconnect cleanly after.
await self._connection.disconnect()
connected = False
await asyncio.sleep(self._delay, loop=self._loop)
continue # next iteration we will try to reconnect
break # all steps done, break retry loop
else: