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

Unify retry loops in mtprotosender._connect

Browse Source 
Now the retry count is not twice its value.
This commit is contained in:
Lonami Exo 2019-10-24 13:40:09 +02:00
parent 3039915ce9
commit 08b78f0c47
1 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
telethon/network/mtprotosender.py
View File

@ -197,20 +197,26 @@ class MTProtoSender:
receive loops.
"""
self._log.info('Connecting to %s...', self._connection)
for attempt in retry_range(self._retries):
if await self._try_connect(attempt):
break
else:
raise ConnectionError('Connection to Telegram failed %d time(s)', self._retries)
if not self.auth_key:
for attempt in retry_range(self._retries):
if await self._try_gen_auth_key(attempt):
break
else:
e = ConnectionError('auth_key generation failed %d time(s)', self._retries)
await self._disconnect(error=e)
raise e
connected = False
for attempt in retry_range(self._retries):
if not connected:
connected = await self._try_connect(attempt)
if not connected:
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
break # all steps done, break retry loop
else:
if not connected:
raise ConnectionError('Connection to Telegram failed %d time(s)', self._retries)
e = ConnectionError('auth_key generation failed %d time(s)', self._retries)
await self._disconnect(error=e)
raise e
self._log.debug('Starting send loop')
self._send_loop_handle = self._loop.create_task(self._send_loop())