13.08.2026 20:25 перед внедрением коррекции поведения ИИ при использовании инструментов (создание флагов мусорных сообщений для очистки)

llm/db/db_tasks.py

@@ -0,0 +1,102 @@
+"""
+Модуль управления бэклогом задач (Task Tracker).
+"""
+import re
+from typing import List, Dict, Any, Optional
+from .connection import get_db_connection
+
+def normalize_task_id(task_id_input: str) -> str:
+    if not task_id_input:
+        return ""
+    clean_id = str(task_id_input).strip().upper().replace("TASK-", "").replace("TASK", "")
+    if clean_id.isdigit():
+        num = int(clean_id)
+        return f"TASK-{(num):02d}" if num < 100 else f"TASK-{(num):03d}"
+    return f"TASK-{clean_id}"
+
+def db_get_tasks(user_id: int) -> List[Dict[str, Any]]:
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        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()
+    
+    # Считываем все существующие TASK-XX и находим реальный максимум
+    cursor.execute("SELECT task_id FROM tasks")
+    rows = cursor.fetchall()
+    
+    max_num = 0
+    for r in rows:
+        t_id = r["task_id"] or ""
+        match = re.search(r'\d+', t_id)
+        if match:
+            num = int(match.group(0))
+            if num > max_num:
+                max_num = num
+                
+    new_task_id = f"TASK-{(max_num + 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", "task_id": new_task_id, "message": f"Задача {new_task_id} создана"}
+
+def db_update_task_status(user_id: int, task_id: str, status: str = "COMPLETED", due_date: Optional[str] = None) -> Dict[str, Any]:
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    formatted_id = normalize_task_id(task_id)
+    
+    if due_date:
+        cursor.execute("""
+            UPDATE tasks 
+            SET status = ?, due_date = ? 
+            WHERE (UPPER(task_id) = ? OR task_id LIKE ?) AND user_id = ?
+        """, (status.upper(), due_date, formatted_id, f"%{task_id.strip()}", user_id))
+    else:
+        cursor.execute("""
+            UPDATE tasks 
+            SET status = ? 
+            WHERE (UPPER(task_id) = ? OR task_id LIKE ?) AND user_id = ?
+        """, (status.upper(), formatted_id, f"%{task_id.strip()}", user_id))
+
+    if cursor.rowcount == 0:
+        conn.close()
+        return {"error": f"Задача {task_id} не найдена или принадлежит другому пользователю"}
+
+    conn.commit()
+    conn.close()
+    return {"status": "success", "message": f"Статус задачи {formatted_id} обновлен на {status.upper()}"}
+
+def db_delete_task(user_id: int, task_id: str) -> Dict[str, Any]:
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    
+    formatted_id = normalize_task_id(task_id)
+    
+    cursor.execute("""
+        DELETE FROM tasks 
+        WHERE (UPPER(task_id) = ? OR task_id LIKE ?) AND user_id = ?
+    """, (formatted_id, f"%{task_id.strip()}", user_id))
+    
+    if cursor.rowcount == 0:
+        conn.close()
+        return {"error": f"Задача {task_id} не найдена"}
+
+    conn.commit()
+    conn.close()
+    return {"status": "success", "message": f"Задача {formatted_id} удалена"}