Telegram/modules/commands/spoiler.py

from app import app
from pyrogram import filters
from pyrogram.types import Message, ReplyKeyboardMarkup
from pyrogram.client import Client
from classes.errors.holo_user import UserNotFoundError, UserInvalidError
from classes.holo_user import HoloUser
from modules.logging import logWrite
from modules.utils import locale
from modules.database import col_spoilers, col_applications
from modules import custom_filters


@app.on_message(
    custom_filters.enabled_spoilers
    & ~filters.scheduled
    & filters.private
    & ~custom_filters.banned
    & filters.command(["spoiler"], prefixes=["/"])
)
async def cmd_spoiler(app: Client, msg: Message):
    try:
        holo_user = HoloUser(msg.from_user)
    except (UserInvalidError, UserNotFoundError):
        return

    if col_applications.find_one({"user": holo_user.id}) is None:
        await msg.reply_text(locale("not_member", "message", locale=msg.from_user))
        return

    if (
        holo_user.application_state()[0] != "fill"
        and holo_user.sponsorship_state()[0] != "fill"
    ):
        if col_spoilers.find_one({"user": holo_user.id, "completed": False}) is None:
            col_spoilers.insert_one(
                {
                    "user": holo_user.id,
                    "completed": False,
                    "category": None,
                    "description": None,
                    "photo": None,
                    "video": None,
                    "audio": None,
                    "animation": None,
                    "document": None,
                    "caption": None,
                    "text": None,
                }
            )

            await msg.reply_text(
                locale("spoiler_started", "message", locale=msg.from_user),
                reply_markup=ReplyKeyboardMarkup(
                    locale("spoiler_categories", "keyboard"),
                    resize_keyboard=True,
                    one_time_keyboard=True,
                ),
            )
            logWrite(f"User {msg.from_user.id} started creating new spoiler")

        else:
            await msg.reply_text(
                locale("spoiler_unfinished", "message", locale=msg.from_user)
            )