Skip to content

Commit

Permalink
Support get_messages without specifying chat_id
Browse files Browse the repository at this point in the history
Partially Fixes #116
  • Loading branch information
SpEcHiDe authored Jan 27, 2025
1 parent b3ebea5 commit 7516f4e
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions pyrogram/methods/messages/get_messages.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,6 +98,21 @@ async def get_messages(
Raises:
ValueError: In case of invalid arguments.
"""

if not chat_id and message_ids:
is_iterable = not isinstance(message_ids, int)
ids = list(message_ids) if is_iterable else [message_ids]
ids = [raw.types.InputMessageID(id=i) for i in ids]
rpc = raw.functions.messages.GetMessages(id=ids)
r = await self.invoke(rpc, sleep_threshold=-1)
messages = await utils.parse_messages(
self,
r,
is_scheduled=is_scheduled,
replies=replies
)
return messages if is_iterable else messages[0] if messages else None

if chat_id:
ids, ids_type = (
(message_ids, raw.types.InputMessageID) if message_ids
Expand Down

0 comments on commit 7516f4e

Please sign in to comment.