доб проверку на макс промпты в ap и sp калбеке

2 files changed, 20 insertions, 5 deletions

diff --git a/callbacks.py b/callbacks.py
index 6a45658..8fe02c2 100644
--- a/callbacks.py
+++ b/callbacks.py
@@ -766,7 +766,10 @@ async def set_prompt_callback(callback: CallbackQuery):
         return
 
     if prompt_user_id != 0:
-        await add_prompt_callback(callback, f"0%{user_id}%{prompt_user_id}%{data[3]}", alo=False)
+        if not len(config["prompts"]) >= MAX_PROMPTS:
+            await add_prompt_callback(callback, f"0%{user_id}%{prompt_user_id}%{data[3]}", alo=False)
+        else:
+            return await callback.answer("достигнуто максимальное количество промптов врат")
             
     config = get_user_config(user_id)
     set_user_config(
@@ -812,6 +815,10 @@ async def add_prompt_callback(callback: CallbackQuery, dataalo="", alo=True):
         return await callback.answer("это стандартный промпт suka.")
 
     config = get_user_config(user_id)
+
+
+    if len(config["prompts"]) >= MAX_PROMPTS:
+        return await callback.answer("достигнуто максимальное количество промптов врат")
     
     new_prompts = config["prompts"]
 
diff --git a/commands.py b/commands.py
index beecfd1..d50f605 100644
--- a/commands.py
+++ b/commands.py
@@ -331,15 +331,23 @@ async def start_handler(message: Message, command: CommandObject):
                     )
                 )
             elif prompt_pub:
-                return await message.reply(
-                    f"✅ вот `{prompt.get('name', 'говно')}` промпт крутой da:"
-                    f"```\n{prompt.get('text')}\n```",
-                    reply_markup=InlineKeyboardMarkup(
+                if len(config["prompts"]) => MAX_PROMPTS:
+                    keyboard = InlineKeyboardMarkup(
+                        inline_keyboard=[
+                            [InlineKeyboardButton(text="достигнут лимит промптов", callback_data=f"hamza")],
+                        ]
+                    )
+                else:
+                    InlineKeyboardMarkup(
                         inline_keyboard=[
                             [InlineKeyboardButton(text="✅ добавить", callback_data=f"ap%{user_id}%{prompt_user_id}%{prompt_id}")],
                             [InlineKeyboardButton(text="✅🔗 доб. + установить", callback_data=f"sp%{user_id}%{prompt_user_id}%{prompt_id}%2")],
                         ]
                     )
+                return await message.reply(
+                    f"✅ вот `{prompt.get('name', 'говно')}` промпт крутой da:"
+                    f"```\n{prompt.get('text')}\n```",
+                    reply_markup=keyboard
                 )
         except Exception as e: 
             print(f"{YELLOW} -- prompt - {e}{RESET}")