LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-01-24 08:14:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make reconnect private

Browse Source
This commit is contained in:
Lonami Exo 2017-09-22 12:31:41 +02:00
parent bc15b451b5
commit 4245ec5abc
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
telethon/telegram_bare_client.py
View File

@ -181,7 +181,7 @@ class TelegramBareClient:
"""Disconnects from the Telegram server"""
self._sender.disconnect()
def reconnect(self, new_dc=None):
def _reconnect(self, new_dc=None):
"""If 'new_dc' is not set, only a call to .connect() will be made
since it's assumed that the connection has been lost and the
library is reconnecting.
@ -325,7 +325,7 @@ class TelegramBareClient:
except ConnectionResetError:
self._logger.debug('Server disconnected us. Reconnecting and '
'resending request...')
self.reconnect()
self._reconnect()
except FloodWaitError:
self.disconnect()

6
telethon/telegram_client.py
View File

@ -268,14 +268,14 @@ class TelegramClient(TelegramBareClient):
# For normal use cases, this won't happen, because this will only
# be on the very first connection (not authorized, not running),
# but may be an issue for people who actually travel?
self.reconnect(new_dc=e.new_dc)
self._reconnect(new_dc=e.new_dc)
return self.invoke(request)
except ConnectionResetError:
if self._connect_lock.locked():
# We are connecting and we don't want to reconnect there...
raise
while self._user_connected and not self.reconnect():
while self._user_connected and not self._reconnect():
pass # Retry forever until we finally can send the request
# Let people use client(SomeRequest()) instead client.invoke(...)
@ -1065,7 +1065,7 @@ class TelegramClient(TelegramBareClient):
pass
except ConnectionResetError:
self._logger.debug('Server disconnected us. Reconnecting...')
while self._user_connected and not self.reconnect():
while self._user_connected and not self._reconnect():
pass # Retry forever, this is instant messaging
except Exception as e: