mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 13:27:47 +00:00
Code Issues Releases Wiki Activity

Remove unneeded FloodWait check

Browse Source
This commit is contained in:
Dan 2018-11-04 16:53:23 +01:00
parent 46e19c395c
commit 966926d383
1 changed files with 10 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pyrogram/client/ext/utils.py
View File

@ -17,12 +17,10 @@
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>. # along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
import logging import logging
import time
from base64 import b64decode, b64encode from base64 import b64decode, b64encode
from struct import pack from struct import pack
from weakref import proxy from weakref import proxy
from pyrogram.api.errors import FloodWait
from pyrogram.client import types as pyrogram_types from pyrogram.client import types as pyrogram_types
from ...api import types, functions from ...api import types, functions
from ...api.errors import StickersetInvalid from ...api.errors import StickersetInvalid
@ -635,19 +633,11 @@ def parse_messages(
m.caption.init(m._client, m.caption_entities or []) m.caption.init(m._client, m.caption_entities or [])
if message.reply_to_msg_id and replies: if message.reply_to_msg_id and replies:
while True:
try:
m.reply_to_message = client.get_messages( m.reply_to_message = client.get_messages(
m.chat.id, m.chat.id,
reply_to_message_ids=message.id, reply_to_message_ids=message.id,
replies=replies - 1 replies=replies - 1
) )
except FloodWait as e:
log.warning("get_messages flood: waiting {} seconds".format(e.x))
time.sleep(e.x)
continue
else:
break
elif isinstance(message, types.MessageService): elif isinstance(message, types.MessageService):
action = message.action action = message.action
@ -748,19 +738,11 @@ def parse_messages(
) )
if isinstance(action, types.MessageActionPinMessage): if isinstance(action, types.MessageActionPinMessage):
while True:
try:
m.pinned_message = client.get_messages( m.pinned_message = client.get_messages(
m.chat.id, m.chat.id,
reply_to_message_ids=message.id, reply_to_message_ids=message.id,
replies=0 replies=0
) )
except FloodWait as e:
log.warning("get_messages flood: waiting {} seconds".format(e.x))
time.sleep(e.x)
continue
else:
break
else: else:
m = pyrogram_types.Message(message_id=message.id, client=proxy(client)) m = pyrogram_types.Message(message_id=message.id, client=proxy(client))