diff options
| -rw-r--r-- | callbacks.py | 9 | ||||
| -rw-r--r-- | commands.py | 16 |
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}") |