Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions telebot/asyncio_filters.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,6 +147,8 @@ async def check(self, obj: Union[types.Message, types.CallbackQuery, types.Inlin
text = obj.question
elif isinstance(obj, types.Message):
text = obj.text or obj.caption
if text is None:
return False
elif isinstance(obj, types.CallbackQuery):
text = obj.data
elif isinstance(obj, types.InlineQuery):
Expand Down Expand Up @@ -239,6 +241,8 @@ async def check(self, message, text):
elif isinstance(text, list) or isinstance(text, tuple):
text = [i for i in text if isinstance(i, str)]

if message.text is None:
return False
return any([i in message.text for i in text])


Expand All @@ -260,6 +264,8 @@ async def check(self, message, text):
"""
:meta private:
"""
if message.text is None:
return False
return message.text.startswith(text)


Expand Down Expand Up @@ -447,4 +453,6 @@ async def check(self, message):
"""
:meta private:
"""
if message.text is None:
return False
return message.text.isdigit()
21 changes: 20 additions & 1 deletion tests/test_custom_filters.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,12 @@

sys.path.append('../')

import asyncio
import json

import pytest

from telebot import types
from telebot import asyncio_filters, types
from telebot.custom_filters import IsDigitFilter, TextContainsFilter, TextStartsFilter


Expand Down Expand Up @@ -50,3 +51,21 @@ def test_text_filters_still_match_text_messages(text_message):
assert TextContainsFilter().check(text_message, "23") is True
assert TextStartsFilter().check(text_message, "12") is True
assert IsDigitFilter().check(text_message) is True


def test_async_text_filters_skip_message_without_text(photo_message):
assert asyncio.run(asyncio_filters.TextContainsFilter().check(photo_message, "23")) is False
assert asyncio.run(asyncio_filters.TextStartsFilter().check(photo_message, "12")) is False
assert asyncio.run(asyncio_filters.IsDigitFilter().check(photo_message)) is False
assert asyncio.run(asyncio_filters.TextFilter(contains=["23"]).check(photo_message)) is False
assert asyncio.run(asyncio_filters.TextFilter(equals="12345", ignore_case=True).check(photo_message)) is False


def test_async_text_filters_still_match_text_messages(text_message):
assert asyncio.run(asyncio_filters.TextContainsFilter().check(text_message, "23")) is True
assert asyncio.run(asyncio_filters.TextStartsFilter().check(text_message, "12")) is True
assert asyncio.run(asyncio_filters.IsDigitFilter().check(text_message)) is True
captioned = make_message(
photo=[{"file_id": "f", "file_unique_id": "u", "width": 1, "height": 1}], caption="12345"
)
assert asyncio.run(asyncio_filters.TextFilter(contains=["23"]).check(captioned)) is True