From 490b8bf5791ef889a8b5886e0de4a56e98e89312 Mon Sep 17 00:00:00 2001 From: Dan <14043624+delivrance@users.noreply.github.com> Date: Mon, 31 Dec 2018 12:02:15 +0100 Subject: [PATCH] Fix get_message getting infinite replies --- pyrogram/client/methods/messages/get_messages.py | 2 +- pyrogram/client/types/messages_and_media/messages.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pyrogram/client/methods/messages/get_messages.py b/pyrogram/client/methods/messages/get_messages.py index da2f6578..d0c064f9 100644 --- a/pyrogram/client/methods/messages/get_messages.py +++ b/pyrogram/client/methods/messages/get_messages.py @@ -78,6 +78,6 @@ class GetMessages(BaseClient): else: 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] diff --git a/pyrogram/client/types/messages_and_media/messages.py b/pyrogram/client/types/messages_and_media/messages.py index cd401c1b..51d3fbdf 100644 --- a/pyrogram/client/types/messages_and_media/messages.py +++ b/pyrogram/client/types/messages_and_media/messages.py @@ -47,13 +47,13 @@ class Messages(PyrogramType): self.messages = messages @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} chats = {i.id: i for i in messages.chats} return 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 )