Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove unnecessary method call in get_media_group (#860)

Browse Source
This commit is contained in:
Danipulok 2022-01-11 17:42:04 +02:00 committed by GitHub
parent de9705f126
commit 10c512d39c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pyrogram/methods/messages/get_media_group.py
View File

@ -51,13 +51,10 @@ class GetMediaGroup(Scaffold):
In case target message doesn't belong to a media group. In case target message doesn't belong to a media group.
""" """
# There can be maximum 10 items in a media group.
messages = await self.get_messages(chat_id, [msg_id for msg_id in range(message_id - 9, message_id + 10)],
replies=0)
if message_id <= 0: if message_id <= 0:
raise ValueError("Passed message_id is negative or equal to zero.") raise ValueError("Passed message_id is negative or equal to zero.")
# Get messages with id from `id - 9` to `id + 10` to get all possible media group messages.
messages = await self.get_messages( messages = await self.get_messages(
chat_id=chat_id, chat_id=chat_id,
message_ids=[msg_id for msg_id in range(message_id - 9, message_id + 10)], message_ids=[msg_id for msg_id in range(message_id - 9, message_id + 10)],
@ -65,7 +62,7 @@ class GetMediaGroup(Scaffold):
) )
# There can be maximum 10 items in a media group. # There can be maximum 10 items in a media group.
# The if/else condition to fix the problem of getting correct `media_group_id` when it has `message_id` less then 10. # If/else condition to fix the problem of getting correct `media_group_id` when `message_id` is less than 10.
media_group_id = messages[9].media_group_id if len(messages) == 19 else messages[message_id - 1].media_group_id media_group_id = messages[9].media_group_id if len(messages) == 19 else messages[message_id - 1].media_group_id
if media_group_id is None: if media_group_id is None: