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

Create a separate method to start reconnections

Browse Source
This commit is contained in:
Lonami Exo 2018-06-24 10:44:31 +02:00
parent 0d0e78e10b
commit 184424957d
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
telethon/network/mtprotosender.py
View File

@ -305,6 +305,11 @@ class MTProtoSender:
__log__.error('Failed to reconnect automatically.') __log__.error('Failed to reconnect automatically.')
await self._disconnect(error=ConnectionError()) await self._disconnect(error=ConnectionError())
def _start_reconnect(self):
"""Starts a reconnection in the background."""
if self._user_connected:
self._loop.create_task(self._reconnect())
def _clean_containers(self, msg_ids): def _clean_containers(self, msg_ids):
""" """
Helper method to clean containers from the pending messages Helper method to clean containers from the pending messages
@ -378,7 +383,7 @@ class MTProtoSender:
__log__.exception('Unhandled exception while receiving') __log__.exception('Unhandled exception while receiving')
await asyncio.sleep(1) await asyncio.sleep(1)
self._loop.create_task(self._reconnect()) self._start_reconnect()
break break
else: else:
# Remove the cancelled messages from pending # Remove the cancelled messages from pending
@ -417,7 +422,7 @@ class MTProtoSender:
__log__.exception('Unhandled exception while receiving') __log__.exception('Unhandled exception while receiving')
await asyncio.sleep(1) await asyncio.sleep(1)
self._loop.create_task(self._reconnect()) self._start_reconnect()
break break
# TODO Check salt, session_id and sequence_number # TODO Check salt, session_id and sequence_number
@ -436,7 +441,7 @@ class MTProtoSender:
# an actually broken authkey? # an actually broken authkey?
__log__.warning('Broken authorization key?: {}'.format(e)) __log__.warning('Broken authorization key?: {}'.format(e))
self.state.auth_key = None self.state.auth_key = None
self._loop.create_task(self._reconnect()) self._start_reconnect()
break break
except SecurityError as e: except SecurityError as e:
# A step while decoding had the incorrect data. This message # A step while decoding had the incorrect data. This message