удалит промпт

1 files changed, 39 insertions, 0 deletions

diff --git a/callbacks.py b/callbacks.py
index a5e31cf..6a45658 100644
--- a/callbacks.py
+++ b/callbacks.py
@@ -291,6 +291,7 @@ async def view_prompt_callback(callback: CallbackQuery, dataalo=""):
     buttons = [
         [
             InlineKeyboardButton(text="✏️ изменить", callback_data=f"ed%{user_id}%{prompt_id}"),
+            InlineKeyboardButton(text="❌ удалить", callback_data=f"dp%{user_id}%{prompt_id}"),
             
         ],
         [
@@ -440,6 +441,44 @@ async def view_text_callback(callback: CallbackQuery, dataalo=""):
     )
 
 
+@router.callback_query(F.data.startswith("dp%"))
+async def delete_prompt_callback(callback: CallbackQuery, dataalo=""):
+    current_user_id = str(callback.from_user.id)
+    if dataalo != "":
+        data = dataalo
+    else:
+        data = callback.data
+
+    data = data.split("%", 2)
+        
+    user_id = data[1]
+    prompt_id = data[2]
+    if user_id != current_user_id:
+        await callback.answer("нет, не трогай это не твое")
+        return
+
+    config = get_user_config(user_id)
+    
+    prompts = config["prompts"]
+
+    del prompts[prompt_id]
+    
+    new_prompts = prompts
+        
+    set_user_config(
+        user_id, 
+        config.get("model"), 
+        config.get("stream"), 
+        config.get("call"), 
+        DEFAULT_PROMPT,
+        config.get("notify"),
+        config.get("markdown"),
+        new_prompts
+    )
+
+    return await user_prompts_callback(callback, dataalo=f"0%{user_id}")
+
+
 @router.callback_query(F.data.startswith("edf%"))
 async def edit_filer_callback(callback: CallbackQuery, dataalo=""):
     current_user_id = str(callback.from_user.id)