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

Allow specifying a threshold to handle flood waits

Browse Source
This commit is contained in:
Lonami Exo 2017-10-08 16:15:30 +02:00
parent 83677fc927
commit 15e90dcb69
2 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
telethon/telegram_bare_client.py
View File

@ -534,11 +534,17 @@ class TelegramBareClient:
'[ERROR] Telegram is having some internal issues', e
)
except FloodWaitError:
except FloodWaitError as e:
if e.seconds > self.session.flood_sleep_threshold | 0:
sender.disconnect()
self.disconnect()
raise
self._logger.debug(
'Sleep of %d seconds below threshold, sleeping' % e.seconds
)
sleep(e.seconds)
# Some really basic functionality
def is_user_authorized(self):

2
telethon/tl/session.py
View File

@ -37,6 +37,7 @@ class Session:
self.lang_pack = session.lang_pack
self.report_errors = session.report_errors
self.save_entities = session.save_entities
self.flood_sleep_threshold = session.flood_sleep_threshold
else: # str / None
self.session_user_id = session_user_id
@ -50,6 +51,7 @@ class Session:
self.lang_pack = ''
self.report_errors = True
self.save_entities = True
self.flood_sleep_threshold = 60
# Cross-thread safety
self._seq_no_lock = Lock()