Skip to content

Commit 966926d

Browse files
committed
Remove unneeded FloodWait check
1 parent 46e19c3 commit 966926d

1 file changed

Lines changed: 10 additions & 28 deletions

File tree

pyrogram/client/ext/utils.py

Lines changed: 10 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -17,12 +17,10 @@
1717
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
1818

1919
import logging
20-
import time
2120
from base64 import b64decode, b64encode
2221
from struct import pack
2322
from weakref import proxy
2423

25-
from pyrogram.api.errors import FloodWait
2624
from pyrogram.client import types as pyrogram_types
2725
from ...api import types, functions
2826
from ...api.errors import StickersetInvalid
@@ -635,19 +633,11 @@ def parse_messages(
635633
m.caption.init(m._client, m.caption_entities or [])
636634

637635
if message.reply_to_msg_id and replies:
638-
while True:
639-
try:
640-
m.reply_to_message = client.get_messages(
641-
m.chat.id,
642-
reply_to_message_ids=message.id,
643-
replies=replies - 1
644-
)
645-
except FloodWait as e:
646-
log.warning("get_messages flood: waiting {} seconds".format(e.x))
647-
time.sleep(e.x)
648-
continue
649-
else:
650-
break
636+
m.reply_to_message = client.get_messages(
637+
m.chat.id,
638+
reply_to_message_ids=message.id,
639+
replies=replies - 1
640+
)
651641
elif isinstance(message, types.MessageService):
652642
action = message.action
653643

@@ -748,19 +738,11 @@ def parse_messages(
748738
)
749739

750740
if isinstance(action, types.MessageActionPinMessage):
751-
while True:
752-
try:
753-
m.pinned_message = client.get_messages(
754-
m.chat.id,
755-
reply_to_message_ids=message.id,
756-
replies=0
757-
)
758-
except FloodWait as e:
759-
log.warning("get_messages flood: waiting {} seconds".format(e.x))
760-
time.sleep(e.x)
761-
continue
762-
else:
763-
break
741+
m.pinned_message = client.get_messages(
742+
m.chat.id,
743+
reply_to_message_ids=message.id,
744+
replies=0
745+
)
764746
else:
765747
m = pyrogram_types.Message(message_id=message.id, client=proxy(client))
766748

0 commit comments

Comments
 (0)