Add MediaGroupFilter

This commit is contained in:
Daniil Kovalenko 2021-03-17 18:37:07 +07:00 committed by GitHub
parent cce3a5fbd0
commit 2739680518
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -738,3 +738,20 @@ class ChatTypeFilter(BoundFilter):
return False
return obj.type in self.chat_type
class MediaGroupFilter(BoundFilter):
"""
Check if message is part of a media group.
`is_media_group=True` - the message is part of a media group
`is_media_group=False` - the message is NOT part of a media group
"""
key = "is_media_group"
def __init__(self, is_media_group: bool):
self.is_media_group = is_media_group
async def check(self, message: types.Message) -> bool:
return bool(getattr(message, "media_group_id")) is self.is_media_group