python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-11 18:01:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

regexp filter for callback query #19

Browse source
This commit is contained in:
Alex Root Junior 2018-03-19 01:59:59 +02:00
parent d2d8f4b51c
commit 5467d91cf8
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aiogram/dispatcher/filters.py
View file

@ -2,7 +2,7 @@ import asyncio
import inspect import inspect
import re import re
from ..types import ContentType from ..types import CallbackQuery, ContentType, Message
from ..utils import context from ..utils import context
from ..utils.helper import Helper, HelperMode, Item from ..utils.helper import Helper, HelperMode, Item
@ -127,9 +127,12 @@ class RegexpFilter(Filter):
def __init__(self, regexp): def __init__(self, regexp):
self.regexp = re.compile(regexp, flags=re.IGNORECASE | re.MULTILINE) self.regexp = re.compile(regexp, flags=re.IGNORECASE | re.MULTILINE)
def check(self, message): def check(self, obj):
if message.text: if isinstance(obj, Message) and obj.text:
return bool(self.regexp.search(message.text)) return bool(self.regexp.search(obj.text))
elif isinstance(obj, CallbackQuery) and obj.data:
return bool(self.regexp.search(obj.data))
return False
class RegexpCommandsFilter(AsyncFilter): class RegexpCommandsFilter(AsyncFilter):