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

Handle errors at connection level

Browse Source 
Closes #4042.
This commit is contained in:
Lonami Exo 2023-03-12 17:43:36 +01:00
parent acd3407418
commit be59c36ed3
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
telethon/network/mtprotosender.py
View File

@ -507,6 +507,18 @@ class MTProtoSender:
self._log.info('Connection closed while receiving data') self._log.info('Connection closed while receiving data')
self._start_reconnect(e) self._start_reconnect(e)
return return
except InvalidBufferError as e:
if e.code == 429:
self._log.warning('Server indicated flood error at transport level: %s', e)
await self._disconnect(error=e)
else:
self._log.exception('Server sent invalid buffer')
self._start_reconnect(e)
return
except Exception as e:
self._log.exception('Unhandled error while receiving data')
self._start_reconnect(e)
return
try: try:
message = self._state.decrypt_message_data(body) message = self._state.decrypt_message_data(body)
@ -532,7 +544,7 @@ class MTProtoSender:
self._start_reconnect(e) self._start_reconnect(e)
return return
except Exception as e: except Exception as e:
self._log.exception('Unhandled error while receiving data') self._log.exception('Unhandled error while decrypting data')
self._start_reconnect(e) self._start_reconnect(e)
return return