65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
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)
|
|
)
|