hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-09 05:19:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Raise FloodWaitError instead sleeping

Browse Source
This commit is contained in:
Lonami Exo
2017-05-29 16:46:17 +02:00
parent 7f84374e98
commit 63b1881c83
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
telethon/network/mtproto_sender.py
View File

@@ -5,7 +5,7 @@ from time import sleep, time
from .. import helpers as utils
from ..crypto import AES
from ..errors import (BadMessageError, RPCError,
from ..errors import (BadMessageError, FloodWaitError, RPCError,
InvalidDCError, ReadCancelledError)
from ..tl.all_tlobjects import tlobjects
from ..tl.functions import PingRequest
@@ -367,10 +367,7 @@ class MtProtoSender:
if error.message.startswith('FLOOD_WAIT_'):
self.updates_thread_sleep = error.additional_data
print('Should wait {}s. Sleeping until then.'.format(
error.additional_data))
sleep(error.additional_data)
raise FloodWaitError(seconds=error.additional_data)
elif '_MIGRATE_' in error.message:
raise InvalidDCError(error)