Add custom filters support

mybot/filters/__init__.py

@@ -0,0 +1,9 @@
+from telebot import TeleBot
+from telebot.custom_filters import StateFilter
+
+from .isowner import IsOwnerFilter
+
+
+def add_custom_filters(bot: TeleBot, config):
+    bot.add_custom_filter(StateFilter(bot))
+    bot.add_custom_filter(IsOwnerFilter(config.OWNER_ID))

mybot/filters/isowner.py

@@ -0,0 +1,12 @@
+from telebot.custom_filters import SimpleCustomFilter
+from telebot.types import Message
+
+
+class IsOwnerFilter (SimpleCustomFilter):
+    key = "is_owner"
+
+    def __init__(self, owner_id: int):
+        self.owner_id: int = owner_id
+
+    def check(self, message: Message):
+        return message.from_user.id == self.owner_id