TelegramBot/modules/custom_filters.py

"""Custom message filters"""

from pyrogram import filters
from pyrogram.types import Message

from classes.pyroclient import PyroClient


async def _owner_func(_, __: PyroClient, message: Message):
    return False if message.from_user is None else __.owner == message.from_user.id


async def _context_func(_, __: PyroClient, message: Message):
    return message.from_user.id in __.contexts


owner = filters.create(_owner_func)
context = filters.create(_context_func)