From fb3e9a4c3bc9346d5246993c7d86bb049a731655 Mon Sep 17 00:00:00 2001 From: JosXa Date: Wed, 28 Feb 2018 22:16:02 +0100 Subject: [PATCH] Capture match and add it to event --- telethon/events/__init__.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/telethon/events/__init__.py b/telethon/events/__init__.py index c5c87fbd..2da686fa 100644 --- a/telethon/events/__init__.py +++ b/telethon/events/__init__.py @@ -77,6 +77,8 @@ class _EventCommon(abc.ABC): self._input_chat = None self._chat = None + self.pattern_match = None + self.is_private = isinstance(chat_peer, types.PeerUser) self.is_group = ( isinstance(chat_peer, (types.PeerChat, types.PeerChannel)) @@ -251,8 +253,12 @@ class NewMessage(_EventBuilder): return if self.outgoing and not event.message.out: return - if self.pattern and not self.pattern(event.message.message or ''): - return + + if self.pattern: + match = self.pattern(event.text or '') + if not match: + return + event.pattern_match = match return self._filter_event(event)