First commit

bot/filters/__init__.py

@@ -0,0 +1,5 @@
+from .callback import *
+from .chat_rights import *
+from .chat_type import *
+from .message_content import *
+from .subscrided import *

bot/filters/callback.py

@@ -0,0 +1,35 @@
+from aiogram.filters import BaseFilter
+from aiogram.types import CallbackQuery
+
+# Настройка экспорта в модули
+__all__ = ("CallbackDataStartsWith", "CallbackStartsWith")
+
+
+class CallbackDataStartsWith(BaseFilter):
+    """
+    Фильтр для callback_data, начинающихся с префикса.
+
+    Example:
+        @router.callback_query(CallbackDataStartsWith("menu:"))
+        async def handler(cb: CallbackQuery):
+            await cb.answer("Это callback из меню ✅")
+    """
+
+    def __init__(self, prefix: str) -> None:
+        self.prefix = prefix
+
+    async def __call__(self, callback: CallbackQuery) -> bool:
+        return bool(callback.data and callback.data.startswith(self.prefix))
+
+
+class CallbackStartsWith(BaseFilter):
+    """
+    Фильтр для callback_data, которое начинается с команды из списка.
+    Игнорирует регистр.
+    """
+    def __init__(self, commands: list[str]):
+        self.commands = [cmd.casefold() for cmd in commands]
+
+    async def __call__(self, callback: CallbackQuery) -> bool:
+        data = callback.data.casefold() if callback.data else ""
+        return any(data.startswith(cmd) for cmd in self.commands)

bot/filters/chat_rights.py

@@ -0,0 +1,152 @@
+from typing import Any
+
+from aiogram import Bot
+from aiogram.exceptions import TelegramBadRequest, TelegramForbiddenError
+from aiogram.filters import BaseFilter
+from aiogram.types import Message, ResultChatMemberUnion, CallbackQuery
+
+from configs import ImportantID
+
+# Настройка экспорта в модули
+__all__ = ("IsChatCreator", "IsAdmin", "IsModerator", "IsOwner",)
+
+
+class IsOwner(BaseFilter):
+    """
+    Фильтр для проверки, является ли пользователь владельцем бота.
+
+    Args:
+        send_error_message (bool): Если True, при попытке не- владельца выполнить команду,
+                                   бот отправит сообщение об ошибке.
+
+    Returns:
+        bool | dict[str, Any]:
+            - False, если пользователь не владелец и send_error_message=False
+            - True, если пользователь является владельцем
+            - dict с информацией о пользователе, если send_error_message=True
+
+    Example:
+        @router.message(IsOwner())
+        async def cmd_handler(message: Message):
+            ...
+
+        @router.message(IsOwner(send_error_message=True))
+        async def admin_only(message: Message):
+            ...
+    """
+
+    def __init__(self, send_error_message: bool = False) -> None:
+        """
+        Инициализация фильтра.
+
+        Args:
+            send_error_message: Нужно ли отправлять сообщение при запрещенном доступе
+        """
+        self.send_error_message: bool = send_error_message
+
+    async def __call__(self, update: Message | CallbackQuery, bot: Bot) -> bool | dict[str, Any]:
+        """
+        Проверяет, является ли пользователь владельцем.
+
+        Args:
+            update: Объект Message или CallbackQuery
+            bot: Экземпляр бота (не используется, но требуется сигнатурой)
+
+        Returns:
+            bool | dict[str, Any]: Результат фильтра. Если пользователь владелец,
+                                    возвращается True или dict с info. Иначе False
+        """
+        if not update.from_user:
+            # Без from_user невозможно определить владельца
+            return False
+
+        user_id: int = update.from_user.id
+        is_owner: bool = user_id in ImportantID.OWNERS_ID
+
+        if not is_owner and self.send_error_message:
+            # Отправляем предупреждение о доступе
+            if isinstance(update, Message):
+                await update.answer(text="⛔ Эта команда доступна только владельцу бота!")
+            elif isinstance(update, CallbackQuery):
+                await update.answer(text="⛔ Доступно только владельцу бота!", show_alert=True)
+            return False
+
+        # Если пользователь владелец — возвращаем словарь с дополнительной информацией
+        if is_owner:
+            return {
+                "is_owner": True,
+                "user_id": user_id,
+                "owner_ids": ImportantID.OWNERS_ID
+            }
+
+        # Если не владелец и send_error_message=False
+        return False
+
+
+class IsChatCreator(BaseFilter):
+    """
+    Пользователь является создателем чата.
+
+    Example:
+        @router.message(IsChatCreator())
+        async def handler(msg: Message):
+            await msg.answer("Ты создатель этого чата 👑")
+    """
+
+    async def __call__(self, message: Message, bot: Bot) -> bool:
+        try:
+            member: ResultChatMemberUnion = await bot.get_chat_member(message.chat.id, message.from_user.id)
+            return member.status == "creator"
+        except (TelegramBadRequest, TelegramForbiddenError):
+            return False
+
+
+class IsAdmin(BaseFilter):
+    """
+    Пользователь является администратором (или создателем).
+
+    Example:
+        @router.message(IsAdmin())
+        async def handler(msg: Message):
+            await msg.answer("Ты админ ✅")
+    """
+
+    async def __call__(self, message: Message, bot: Bot) -> bool:
+        try:
+            member: ResultChatMemberUnion = await bot.get_chat_member(message.chat.id, message.from_user.id)
+            return member.status in {"administrator", "creator"}
+        except (TelegramBadRequest, TelegramForbiddenError):
+            return False
+
+
+class IsModerator(BaseFilter):
+    """
+    Администратор с модераторскими правами:
+    - удаление сообщений
+    - ограничение пользователей
+    - закрепление сообщений
+
+    Example:
+        @router.message(IsModerator())
+        async def handler(msg: Message):
+            await msg.answer("Ты модератор ✅")
+    """
+
+    async def __call__(self, message: Message, bot: Bot) -> bool:
+        try:
+            member: ResultChatMemberUnion = await bot.get_chat_member(message.chat.id, message.from_user.id)
+
+            if member.status == "creator":
+                return True
+            if member.status != "administrator":
+                return False
+
+            required_rights: list[bool] = [
+                getattr(member, "can_delete_messages", False),
+                getattr(member, "can_restrict_members", False),
+                getattr(member, "can_pin_messages", False),
+            ]
+            return all(required_rights)
+
+        except (TelegramBadRequest, TelegramForbiddenError):
+            return False

bot/filters/chat_type.py

@@ -0,0 +1,33 @@
+from aiogram.filters import BaseFilter
+from aiogram.types import Message
+
+# Настройка экспорта в модули
+__all__ = ("IsPrivate", "IsGroup",)
+
+
+class IsPrivate(BaseFilter):
+    """
+    Сообщение в личке с ботом.
+
+    Example:
+        @router.message(IsPrivate())
+        async def handler(msg: Message):
+            await msg.answer("Это ЛС ✅")
+    """
+
+    async def __call__(self, message: Message) -> bool:
+        return message.chat.type == "private"
+
+
+class IsGroup(BaseFilter):
+    """
+    Сообщение в группе или супергруппе.
+
+    Example:
+        @router.message(IsGroup())
+        async def handler(msg: Message):
+            await msg.answer("Это сообщение в группе ✅")
+    """
+
+    async def __call__(self, message: Message) -> bool:
+        return message.chat.type in {"group", "supergroup"}

bot/filters/message_content.py

@@ -0,0 +1,71 @@
+from aiogram.filters import BaseFilter
+from aiogram.types import Message
+
+# Настройка экспорта в модули
+__all__ = ("IsReply", "IsForwarded", "HasMedia", "ContainsURL",)
+
+
+class IsReply(BaseFilter):
+    """
+    Сообщение является ответом.
+
+    Example:
+        @router.message(IsReply())
+        async def handler(msg: Message):
+            await msg.answer("Это реплай ✅")
+    """
+
+    async def __call__(self, message: Message) -> bool:
+        return message.reply_to_message is not None
+
+
+class IsForwarded(BaseFilter):
+    """
+    Сообщение переслано из другого чата/от пользователя.
+
+    Example:
+        @router.message(IsForwarded())
+        async def handler(msg: Message):
+            await msg.answer("Это пересланное сообщение 🔄")
+    """
+
+    async def __call__(self, message: Message) -> bool:
+        return (message.forward_from is not None) or (message.forward_from_chat is not None)
+
+
+class HasMedia(BaseFilter):
+    """
+    Сообщение содержит медиа (фото, видео, документ и т.д.).
+
+    Example:
+        @router.message(HasMedia())
+        async def handler(msg: Message):
+            await msg.answer("Это медиа ✅")
+    """
+
+    async def __call__(self, message: Message) -> bool:
+        return any([
+            message.photo,
+            message.video,
+            message.document,
+            message.audio,
+            message.voice,
+            message.video_note,
+            message.sticker,
+        ])
+
+
+class ContainsURL(BaseFilter):
+    """
+    Сообщение содержит ссылку (http/https).
+
+    Example:
+        @router.message(ContainsURL())
+        async def handler(msg: Message):
+            await msg.answer("Это сообщение с ссылкой 🔗")
+    """
+
+    async def __call__(self, message: Message) -> bool:
+        if not message.text:
+            return False
+        return "http://" in message.text or "https://" in message.text

bot/filters/subscrided.py

@@ -0,0 +1,41 @@
+from typing import Union
+
+from aiogram import Bot
+from aiogram.exceptions import TelegramBadRequest, TelegramForbiddenError
+from aiogram.filters import BaseFilter
+from aiogram.types import Message, ResultChatMemberUnion
+
+# Настройка экспорта в модули
+__all__ = ("FilterSubscribed",)
+
+
+class FilterSubscribed(BaseFilter):
+    """
+    Фильтр для проверки подписки пользователя на один или несколько каналов.
+    Поддерживает как публичные каналы (username), так и приватные (ID).
+
+    Пример:
+    # Проверка сразу двух каналов: публичный по username и приватный по ID
+    @router.message(FilterSubscribed(["@public_channel", -1001234567890]))
+    async def only_subscribed(message: Message):
+        await message.answer("Ты подписан и на публичный, и на приватный канал ✅")
+    """
+
+    def __init__(self, channels: list[Union[str, int]]) -> None:
+        self.channels = channels
+
+    async def __call__(self, message: Message, bot: Bot) -> bool:
+        for channel in self.channels:
+            try:
+                member: ResultChatMemberUnion = await bot.get_chat_member(
+                    chat_id=channel,
+                    user_id=message.from_user.id
+                )
+                if member.status in ("left", "kicked"):
+                    return False
+
+            except (TelegramBadRequest, TelegramForbiddenError):
+                # Канал недоступен, либо у бота нет прав
+                return False
+
+        return True