yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-03 14:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix get_message getting infinite replies

Browse source
This commit is contained in:
Dan 2018-12-31 12:02:15 +01:00
parent a02846e900
commit 490b8bf579
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pyrogram/client/methods/messages/get_messages.py
View file

@ -78,6 +78,6 @@ class GetMessages(BaseClient):
else: else:
rpc = functions.messages.GetMessages(id=ids) rpc = functions.messages.GetMessages(id=ids)
messages = pyrogram.Messages._parse(self, self.send(rpc)) messages = pyrogram.Messages._parse(self, self.send(rpc), replies)
return messages if is_iterable else messages.messages[0] return messages if is_iterable else messages.messages[0]

4
pyrogram/client/types/messages_and_media/messages.py
View file

@ -47,13 +47,13 @@ class Messages(PyrogramType):
self.messages = messages self.messages = messages
@staticmethod @staticmethod
def _parse(client, messages: types.messages.Messages) -> "Messages": def _parse(client, messages: types.messages.Messages, replies: int = 1) -> "Messages":
users = {i.id: i for i in messages.users} users = {i.id: i for i in messages.users}
chats = {i.id: i for i in messages.chats} chats = {i.id: i for i in messages.chats}
return Messages( return Messages(
total_count=getattr(messages, "count", len(messages.messages)), total_count=getattr(messages, "count", len(messages.messages)),
messages=[Message._parse(client, message, users, chats) for message in messages.messages], messages=[Message._parse(client, message, users, chats, replies) for message in messages.messages],
client=client client=client
) )