add IsReplyFilter

This commit is contained in:
Arwichok 2019-09-16 14:50:23 +03:00
parent 4984c2313a
commit f7fdaa1153
2 changed files with 21 additions and 1 deletions

View file

@ -10,7 +10,7 @@ from aiohttp.helpers import sentinel
from aiogram.utils.deprecated import renamed_argument
from .filters import Command, ContentTypeFilter, ExceptionsFilter, FiltersFactory, HashTag, Regexp, \
RegexpCommandsFilter, StateFilter, Text, IDFilter, AdminFilter
RegexpCommandsFilter, StateFilter, Text, IDFilter, AdminFilter, IsReplyFilter
from .handler import Handler
from .middlewares import MiddlewareManager
from .storage import BaseStorage, DELTA, DisabledStorage, EXCEEDED_COUNT, FSMContext, \
@ -145,6 +145,12 @@ class Dispatcher(DataMixin, ContextInstanceMixin):
self.callback_query_handlers,
self.inline_query_handlers,
])
filters_factory.bind(IsReplyFilter, event_handlers=[
self.message_handlers,
self.edited_message_handlers,
self.channel_post_handlers,
self.edited_channel_post_handlers,
])
def __del__(self):
self.stop_polling()

View file

@ -625,3 +625,17 @@ class AdminFilter(Filter):
admins = [member.user.id for chat_id in chat_ids for member in await obj.bot.get_chat_administrators(chat_id)]
return user_id in admins
class IsReplyFilter(BoundFilter):
"""
Check if message is replied and send reply message to handler
"""
key = 'is_reply'
def __init__(self, is_reply):
self.is_reply = is_reply
async def check(self, msg: ats.Message):
if msg.reply_to_message:
return {'reply': msg.reply_to_message}