Fix get_messages and usages

delivrance · delivrance · commit ff90baffb3cc · 2022-05-14T21:12:37.000+02:00
diff --git a/pyrogram/methods/messages/get_messages.py b/pyrogram/methods/messages/get_messages.py
@@ -116,4 +116,4 @@ async def get_messages(
 
         messages = await utils.parse_messages(self, r, replies=replies)
 
-        return messages if is_list else messages[0]
+        return messages if is_list else messages[0] if messages else None
diff --git a/pyrogram/utils.py b/pyrogram/utils.py
@@ -113,7 +113,7 @@ async def parse_messages(client, messages: "raw.types.messages.Messages", replie
 
             reply_messages = await client.get_messages(
                 chat_id,
-                reply_to_message_ids=messages_with_replies.keys(),
+                reply_to_message_ids=list(messages_with_replies.keys()),
                 replies=replies - 1
             )
 

Original file line number	Diff line number	Diff line change
`@@ -116,4 +116,4 @@ async def get_messages(`
`116`	`116`
`117`	`117`	`messages = await utils.parse_messages(self, r, replies=replies)`
`118`	`118`
`119`		`- return messages if is_list else messages[0]`
	`119`	`+ return messages if is_list else messages[0] if messages else None`
Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ async def parse_messages(client, messages: "raw.types.messages.Messages", replie`
`113`	`113`
`114`	`114`	`reply_messages = await client.get_messages(`
`115`	`115`	`chat_id,`
`116`		`- reply_to_message_ids=messages_with_replies.keys(),`
	`116`	`+ reply_to_message_ids=list(messages_with_replies.keys()),`
`117`	`117`	`replies=replies - 1`
`118`	`118`	`)`
`119`	`119`