python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update Text filter

Browse source 
made the Text filter ready for lazy_gettext
This commit is contained in:
Nikita 2019-03-30 19:44:22 +05:00
parent a977ae3318
commit 585bbfaee7
1 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aiogram/dispatcher/filters/builtin.py
View file

@ -4,6 +4,8 @@ from contextvars import ContextVar
from dataclasses import dataclass, field from dataclasses import dataclass, field
from typing import Any, Dict, Iterable, Optional, Union from typing import Any, Dict, Iterable, Optional, Union
from babel.support import LazyProxy
from aiogram import types from aiogram import types
from aiogram.dispatcher.filters.filters import BoundFilter, Filter from aiogram.dispatcher.filters.filters import BoundFilter, Filter
from aiogram.types import CallbackQuery, Message, InlineQuery from aiogram.types import CallbackQuery, Message, InlineQuery
@ -116,10 +118,10 @@ class Text(Filter):
""" """
def __init__(self, def __init__(self,
equals: Optional[str] = None, equals: Optional[Union[str, LazyProxy]] = None,
contains: Optional[str] = None, contains: Optional[Union[str, LazyProxy]] = None,
startswith: Optional[str] = None, startswith: Optional[Union[str, LazyProxy]] = None,
endswith: Optional[str] = None, endswith: Optional[Union[str, LazyProxy]] = None,
ignore_case=False): ignore_case=False):
""" """
Check text for one of pattern. Only one mode can be used in one filter. Check text for one of pattern. Only one mode can be used in one filter.
@ -173,13 +175,13 @@ class Text(Filter):
text = text.lower() text = text.lower()
if self.equals: if self.equals:
return text == self.equals return text == str(self.equals)
elif self.contains: elif self.contains:
return self.contains in text return str(self.contains) in text
elif self.startswith: elif self.startswith:
return text.startswith(self.startswith) return text.startswith(str(self.startswith))
elif self.endswith: elif self.endswith:
return text.endswith(self.endswith) return text.endswith(str(self.endswith))
return False return False