feat: полноценный мультипользовательский режим (users, JWT, изолированные задачи)

2026-08-09 10:50:04 +03:00
parent a47514648c
commit 8a3f3c749f
7 changed files with 581 additions and 195 deletions
@@ -1,129 +1,110 @@
 import sqlite3
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional

-DB_NAME = "/home/puh/scud_orion_ai_v2/data/scud_orion_ai.db"
+DB_PATH = "/home/puh/scud_context_api/scud_orion_ai.db"

-def db_get_active_system_prompt() -> str:
-    try:
-        conn = sqlite3.connect(DB_NAME)
-        cursor = conn.cursor()
-        cursor.execute("SELECT prompt_text FROM system_prompts WHERE name = 'main_agent' AND is_active = 1 LIMIT 1")
-        row = cursor.fetchone()
-        conn.close()
-        if row and row[0]:
-            return row[0].replace("**", "")
-    except Exception as e:
-        print(f"Ошибка чтения системного промпта из БД: {e}")
+def get_db_connection():
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    return conn

-    return "Ты — интеллектуальный ИИ-ассистент и архитектурный координатор проекта SCUD Orion AI."
+# === ЗАДАЧИ С ФИЛЬТРАЦИЕЙ ПО USER_ID ===

-def db_add_system_prompt(name: str = "main_agent", prompt_text: str = "") -> dict:
-    clean_text = prompt_text.replace("**", "")
-    conn = sqlite3.connect(DB_NAME)
+def db_get_tasks(user_id: int) -> List[Dict[str, Any]]:
+    conn = get_db_connection()
    cursor = conn.cursor()
    cursor.execute("""
-        INSERT INTO system_prompts (name, prompt_text, is_active)
-        VALUES (?, ?, 1)
-        ON CONFLICT(name) DO UPDATE SET prompt_text=excluded.prompt_text, updated_at=CURRENT_TIMESTAMP
-    """, (name, clean_text))
+        SELECT id, task_id, module, title, priority, status, due_date, created_at 
+        FROM tasks 
+        WHERE user_id = ?
+        ORDER BY id DESC
+    """, (user_id,))
+    rows = cursor.fetchall()
+    conn.close()
+    return [dict(r) for r in rows]
+
+def db_add_task(user_id: int, module: str, title: str, priority: str = "MEDIUM", due_date: Optional[str] = None) -> Dict[str, Any]:
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    cursor.execute("SELECT MAX(id) FROM tasks")
+    max_id = cursor.fetchone()[0] or 0
+    new_task_id = f"TASK-{(max_id + 1):02d}"
+
+    cursor.execute("""
+        INSERT INTO tasks (task_id, module, title, priority, status, due_date, user_id)
+        VALUES (?, ?, ?, ?, 'BACKLOG', ?, ?)
+    """, (new_task_id, module, title, priority.upper(), due_date, user_id))
+    
    conn.commit()
    conn.close()
-    return {"status": "success", "message": f"Системный промпт '{name}' успешно обновлен в базе!"}
+    return {"status": "success", "task_id": new_task_id, "message": f"Задача {new_task_id} создана"}

-def db_get_tasks(status: str = None) -> list:
-    conn = sqlite3.connect(DB_NAME)
-    conn.row_factory = sqlite3.Row
+def db_update_task_status(user_id: int, task_id: str, status: str, due_date: Optional[str] = None) -> Dict[str, Any]:
+    conn = get_db_connection()
    cursor = conn.cursor()
-    if status:
-        cursor.execute("SELECT * FROM tasks WHERE status = ? ORDER BY task_id ASC", (status.upper(),))
-    else:
-        cursor.execute("SELECT * FROM tasks ORDER BY task_id ASC")
-    rows = [dict(r) for r in cursor.fetchall()]
-    conn.close()
-    return rows
-
-def db_update_task_status(task_id: str, status: str = None, due_date: str = None) -> dict:
-    conn = sqlite3.connect(DB_NAME)
-    cursor = conn.cursor()
-    updates = []
-    params = []
-
-    if status:
-        status_upper = status.upper()
-        updates.append("status = ?")
-        params.append(status_upper)
-        if status_upper == "COMPLETED":
-            updates.append("completed_at = CURRENT_TIMESTAMP")
-        else:
-            updates.append("completed_at = NULL")
-
+    
+    task_id_upper = task_id.upper().strip()
+    
    if due_date:
-        updates.append("due_date = ?")
-        params.append(due_date)
-
-    if not updates:
-        conn.close()
-        return {"error": "Не указаны параметры для обновления"}
-
-    params.append(task_id.upper())
-    query = f"UPDATE tasks SET {', '.join(updates)} WHERE task_id = ?"
-    cursor.execute(query, tuple(params))
+        cursor.execute("""
+            UPDATE tasks 
+            SET status = ?, due_date = ? 
+            WHERE UPPER(task_id) = ? AND user_id = ?
+        """, (status.upper(), due_date, task_id_upper, user_id))
+    else:
+        cursor.execute("""
+            UPDATE tasks 
+            SET status = ? 
+            WHERE UPPER(task_id) = ? AND user_id = ?
+        """, (status.upper(), task_id_upper, user_id))

    if cursor.rowcount == 0:
        conn.close()
-        return {"error": f"Задача {task_id} не найдена"}
+        return {"error": f"Задача {task_id_upper} не найдена или принадлежит другому пользователю"}

    conn.commit()
    conn.close()
-    return {"status": "success", "message": f"Задача {task_id} обновлена"}
+    return {"status": "success", "message": f"Статус {task_id_upper} обновлен"}

-def db_delete_task(task_id: str) -> dict:
-    conn = sqlite3.connect(DB_NAME)
-    cursor = conn.cursor()
-    cursor.execute("DELETE FROM tasks WHERE task_id = ?", (task_id.upper(),))
-    deleted_count = cursor.rowcount
-    conn.commit()
-    conn.close()
-
-    if deleted_count == 0:
-        return {"error": f"Задача {task_id} не найдена"}
-    return {"status": "success", "message": f"Задача {task_id} удалена"}
-
-def db_add_task(module: str = "general", title: str = "", priority: str = "MEDIUM", due_date: str = None, task_id: str = None) -> dict:
-    conn = sqlite3.connect(DB_NAME)
+def db_delete_task(user_id: int, task_id: str) -> Dict[str, Any]:
+    conn = get_db_connection()
    cursor = conn.cursor()
+    task_id_upper = task_id.upper().strip()
    
-    if not task_id:
-        cursor.execute("SELECT task_id FROM tasks WHERE task_id LIKE 'TASK-%'")
-        rows = cursor.fetchall()
-        max_num = 0
-        for r in rows:
-            try:
-                num = int(r[0].replace("TASK-", ""))
-                if num > max_num:
-                    max_num = num
-            except ValueError:
-                pass
-        task_id = f"TASK-{max_num + 1:02d}"
-
-    try:
-        cursor.execute(
-            "INSERT INTO tasks (task_id, module, title, status, priority, due_date) VALUES (?, ?, ?, 'BACKLOG', ?, ?)",
-            (task_id.upper(), module, title, priority.upper(), due_date)
-        )
-        conn.commit()
-        res = {"status": "success", "message": f"Задача {task_id} создана", "task_id": task_id}
-    except sqlite3.IntegrityError:
-        res = {"error": f"Задача с ID {task_id} уже существует"}
-    finally:
+    cursor.execute("DELETE FROM tasks WHERE UPPER(task_id) = ? AND user_id = ?", (task_id_upper, user_id))
+    
+    if cursor.rowcount == 0:
        conn.close()
-    return res
+        return {"error": f"Задача {task_id_upper} не найдена или принадлежит другому пользователю"}

-def db_get_rules() -> list:
-    conn = sqlite3.connect(DB_NAME)
-    conn.row_factory = sqlite3.Row
-    cursor = conn.cursor()
-    cursor.execute("SELECT id, rule_text, added_by FROM ai_knowledge_base ORDER BY id")
-    rows = [dict(r) for r in cursor.fetchall()]
+    conn.commit()
    conn.close()
-    return rows
+    return {"status": "success", "message": f"Задача {task_id_upper} удалена"}
+
+# === ОБЩИЕ СИСТЕМНЫЕ РЕСУРСЫ ===
+
+def db_get_active_system_prompt() -> str:
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT prompt_text FROM system_prompts WHERE is_active = 1 ORDER BY id DESC LIMIT 1")
+    row = cursor.fetchone()
+    conn.close()
+    return row["prompt_text"] if row else "Ты — ИИ-ассистент SCUD Orion AI."
+
+def db_add_system_prompt(name: str, prompt_text: str) -> Dict[str, Any]:
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("UPDATE system_prompts SET is_active = 0")
+    cursor.execute("INSERT INTO system_prompts (name, prompt_text, is_active) VALUES (?, ?, 1)", (name, prompt_text))
+    conn.commit()
+    conn.close()
+    return {"status": "success", "message": "Системный промпт обновлен"}
+
+def db_get_rules() -> List[Dict[str, Any]]:
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT id, rule_text FROM ai_knowledge_base ORDER BY id ASC")
+    rows = cursor.fetchall()
+    conn.close()
+    return [dict(r) for r in rows]