Caption and category added for spoilers

modules/commands/reapply.py

@@ -15,7 +15,7 @@ async def cmd_reapply(app: Client, msg: Message):
     holo_user = HoloUser(msg.from_user)
 
     # Check if user has approved/rejected tmp application
-    if (holo_user.application_state()[0] in ["approved", "rejected"]) or (holo_user.application_state()[0] == "none"):
+    if ((holo_user.application_state()[0] in ["approved", "rejected"]) or (holo_user.application_state()[0] == "none")) and holo_user.spoiler_state() is False:
 
         # Check if user's tmp application is already completed or even sent
         if ((holo_user.application_state()[1] is True) and (not col_tmp.find_one({"user": holo_user.id, "type": "application"})["sent"])) or (holo_user.application_state()[0] == "none"):
@@ -51,6 +51,10 @@ async def cmd_reapply(app: Client, msg: Message):
                 ]
             ]))
 
+    elif holo_user.spoiler_state() is True:
+
+        await msg.reply_text(locale("spoiler_in_progress", "message", locale=holo_user))
+
     else:
 
         if (holo_user.application_state()[0] == "fill") and (col_tmp.find_one({"user": holo_user.id, "type": "application"})["sent"] is True):

modules/commands/spoiler.py

@@ -1,6 +1,6 @@
 from app import app
 from pyrogram import filters
-from pyrogram.types import Message, ForceReply
+from pyrogram.types import Message, ReplyKeyboardMarkup
 from pyrogram.client import Client
 from classes.holo_user import HoloUser, UserInvalidError, UserNotFoundError
 from modules.logging import logWrite
@@ -25,15 +25,17 @@ async def cmd_spoiler(app: Client, msg: Message):
                 {
                     "user": msg.from_user.id,
                     "completed": False,
+                    "category": None,
                     "description": None,
                     "photo": None,
                     "video": None,
                     "animation": None,
+                    "caption": None,
                     "text": None
                 }
             )
 
-            await msg.reply_text(locale("spoiler_started", "message", locale=msg.from_user), reply_markup=ForceReply(placeholder=locale("spoiler_content", "force_reply", locale=msg.from_user)))
+            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:

modules/commands/sponsorship.py

@@ -10,9 +10,13 @@ from modules.database import col_applications
 # Sponsorship command ==========================================================================================================
 @app.on_message(custom_filters.enabled_sponsorships & ~filters.scheduled & filters.command(["sponsorship"], prefixes=["/"]) & (custom_filters.allowed | custom_filters.admin))
 async def cmd_sponsorship(app: Client, msg: Message):
-    if HoloUser(msg.from_user).application_state()[0] == "fill":
+    holo_user = HoloUser(msg.from_user)
+    if holo_user.application_state()[0] == "fill":
         await msg.reply_text(locale("finish_application", "message", locale=msg.from_user), quote=should_quote(msg))
         return
+    if holo_user.spoiler_state() is True:
+        await msg.reply_text(locale("spoiler_in_progress", "message", locale=holo_user))
+        return
     await msg.reply_text(locale("sponsorship_apply", "message", locale=msg.from_user), reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(text=str(locale("sponsor_apply", "button", locale=msg.from_user)), callback_data=f"sponsor_apply_{msg.from_user.id}")]]), quote=should_quote(msg))
     # else:
     #     await msg.reply_text(locale("sponsorship_application_empty", "message"))

modules/commands/start.py

@@ -33,11 +33,11 @@ async def cmd_start(app: Client, msg: Message):
         try:
             spoiler = col_spoilers.find_one( {"_id": ObjectId(msg.command[1])} )
             if spoiler["photo"] is not None:
-                await msg.reply_photo(spoiler["photo"])
+                await msg.reply_photo(spoiler["photo"], caption=spoiler["caption"])
             if spoiler["video"] is not None:
-                await msg.reply_video(spoiler["video"])
+                await msg.reply_video(spoiler["video"], caption=spoiler["caption"])
             if spoiler["animation"] is not None:
-                await msg.reply_animation(spoiler["animation"])
+                await msg.reply_animation(spoiler["animation"], caption=spoiler["caption"])
             if spoiler["text"] is not None:
                 await msg.reply_text(spoiler["text"])
         except InvalidId: