LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-25 10:53:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Trigger automatic reconnection if server kicks us (#94)

Browse Source
This commit is contained in:
Lonami Exo 2017-05-30 10:11:18 +02:00
parent 6e1798d6ac
commit fb4885aed0
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
telethon/telegram_client.py
View File

@ -214,6 +214,13 @@ class TelegramClient:
self._reconnect_to_dc(error.new_dc)
return self.invoke(request, timeout=timeout, throw_invalid_dc=True)
except ConnectionResetError:
self._logger.info('Server disconnected us. Reconnecting and '
'resending request...')
self.reconnect()
self.invoke(request, timeout=timeout,
throw_invalid_dc=throw_invalid_dc)
except FloodWaitError:
self.disconnect()
raise
@ -842,9 +849,12 @@ class TelegramClient:
for handler in self.update_handlers:
handler(result)
except ConnectionResetError:
self._logger.info('Server disconnected us. Reconnecting...')
self.reconnect()
except TimeoutError:
self._logger.debug('Receiving updates timed out')
self.reconnect()
except ReadCancelledError:
self._logger.info('Receiving updates cancelled')