Skip to content

Commit 7516f4e

Browse files
authored
Support get_messages without specifying chat_id
Partially Fixes #116
1 parent b3ebea5 commit 7516f4e

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pyrogram/methods/messages/get_messages.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,21 @@ async def get_messages(
9898
Raises:
9999
ValueError: In case of invalid arguments.
100100
"""
101+
102+
if not chat_id and message_ids:
103+
is_iterable = not isinstance(message_ids, int)
104+
ids = list(message_ids) if is_iterable else [message_ids]
105+
ids = [raw.types.InputMessageID(id=i) for i in ids]
106+
rpc = raw.functions.messages.GetMessages(id=ids)
107+
r = await self.invoke(rpc, sleep_threshold=-1)
108+
messages = await utils.parse_messages(
109+
self,
110+
r,
111+
is_scheduled=is_scheduled,
112+
replies=replies
113+
)
114+
return messages if is_iterable else messages[0] if messages else None
115+
101116
if chat_id:
102117
ids, ids_type = (
103118
(message_ids, raw.types.InputMessageID) if message_ids

0 commit comments

Comments
 (0)