LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-04 12:10:21 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add docs

Browse Source 
Fix matching against data
This commit is contained in:
painor 2019-06-26 22:48:22 +01:00
parent ce48523be7
commit cfec109461
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
telethon/events/callbackquery.py
View File

@ -24,14 +24,20 @@ class CallbackQuery(EventBuilder):
A UTF-8 string can also be given, a regex or a callable. For A UTF-8 string can also be given, a regex or a callable. For
instance, to check against ``'data_1'`` and ``'data_2'`` you instance, to check against ``'data_1'`` and ``'data_2'`` you
can use ``re.compile(b'data_')``. can use ``re.compile(b'data_')``.
pattern (`str`, `callable`, `Pattern`, optional):
If set, only buttons with payload matching this pattern will be handled.
You can specify a regex-like string which will be matched
against the message, a callable function that returns ``True``
if a message is acceptable, or a compiled regex pattern.
""" """
def __init__( def __init__(
self, chats=None, *, blacklist_chats=False, func=None, data=None, pattern=None): self, chats=None, *, blacklist_chats=False, func=None, data=None, pattern=None):
super().__init__(chats, blacklist_chats=blacklist_chats, func=func) super().__init__(chats, blacklist_chats=blacklist_chats, func=func)
if data is None and pattern is None: if all(x is not None and not x for x in (data, pattern)):
self._log.warn("Please don't use both data and pattern.data will be ignored if you do so") raise ValueError("Only pass either data or pattern not both.")
data = None
if isinstance(data, bytes): if isinstance(data, bytes):
self.data = data self.data = data
@ -99,7 +105,7 @@ class CallbackQuery(EventBuilder):
return None return None
if self.pattern: if self.pattern:
match = self.pattern(event.message.message or '') match = self.pattern(event.query.data)
if not match: if not match:
return return
event.pattern_match = match event.pattern_match = match