Compare commits
10
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
27e8055a3d | ||
|
|
fa386b3b79 | ||
|
|
a9680db0aa | ||
|
|
66087d5806 | ||
|
|
4b1b8a3586 | ||
|
|
828ce67817 | ||
|
|
4e3c99b728 | ||
|
|
c0a5145604 | ||
|
|
7ee7520c98 | ||
|
|
0e33e61ab2 |
@@ -8,11 +8,11 @@ DATA_DIR = os.path.join(BASE_DIR, "data")
|
|||||||
SCUD_DIR = os.path.join(DATA_DIR, "scud")
|
SCUD_DIR = os.path.join(DATA_DIR, "scud")
|
||||||
ZUP_1C_DIR = os.path.join(DATA_DIR, "1c")
|
ZUP_1C_DIR = os.path.join(DATA_DIR, "1c")
|
||||||
OUTPUT_DIR = os.path.join(BASE_DIR, "output")
|
OUTPUT_DIR = os.path.join(BASE_DIR, "output")
|
||||||
|
REPORTS_DIR = os.path.join(OUTPUT_DIR, "reports")
|
||||||
|
|
||||||
# Путь к намонтированной сетевой шаре 1С в Linux (вместо \\storage\SCUD\Обмен\Штат)
|
|
||||||
SHARE_1C_DIR = "/mnt/scud_share"
|
SHARE_1C_DIR = "/mnt/scud_share"
|
||||||
|
|
||||||
for folder in [DATA_DIR, SCUD_DIR, ZUP_1C_DIR, OUTPUT_DIR]:
|
for folder in [DATA_DIR, SCUD_DIR, ZUP_1C_DIR, OUTPUT_DIR, REPORTS_DIR]:
|
||||||
os.makedirs(folder, exist_ok=True)
|
os.makedirs(folder, exist_ok=True)
|
||||||
|
|
||||||
NOW = datetime.now()
|
NOW = datetime.now()
|
||||||
@@ -23,32 +23,30 @@ if NOW.weekday() == 0:
|
|||||||
else:
|
else:
|
||||||
DATE_YESTERDAY = (NOW - timedelta(days=1)).strftime("%d.%m.%Y")
|
DATE_YESTERDAY = (NOW - timedelta(days=1)).strftime("%d.%m.%Y")
|
||||||
|
|
||||||
# --- Настройки LLM / Ollama ---
|
OLLAMA_URL = "http://10.121.17.227:11434/api/generate"
|
||||||
OLLAMA_URL = "http://192.168.11.3:11434/api/generate"
|
OLLAMA_MODEL = "qwen2.5:14b-instruct-q8_0"
|
||||||
OLLAMA_MODEL = "qwen2.5:14b"
|
|
||||||
MODEL_NAME = OLLAMA_MODEL
|
MODEL_NAME = OLLAMA_MODEL
|
||||||
|
|
||||||
KNOWLEDGE_BASE_PATH = os.path.join(DATA_DIR, "knowledge_base.json")
|
KNOWLEDGE_BASE_PATH = os.path.join(DATA_DIR, "knowledge_base.json")
|
||||||
EXCEPTIONS_PATH = os.path.join(BASE_DIR, "exceptions.json")
|
EXCEPTIONS_PATH = os.path.join(BASE_DIR, "exceptions.json")
|
||||||
|
|
||||||
# --- Настройки подключения к MS SQL Server (1С:ЗУП 3.1) ---
|
|
||||||
ZUP_SQL_CONFIG = {
|
ZUP_SQL_CONFIG = {
|
||||||
"driver": "{ODBC Driver 18 for SQL Server}", # или "{ODBC Driver 17 for SQL Server}"
|
"driver": "{ODBC Driver 18 for SQL Server}",
|
||||||
"server": os.getenv("ZUP_SQL_SERVER", "ACCOUNT-01"),
|
"server": os.getenv("ZUP_SQL_SERVER", "ACCOUNT-01"),
|
||||||
"database": os.getenv("ZUP_SQL_DB", "ZUP30"), # Замените на точное имя базы ЗУП на ACCOUNT-01
|
"database": os.getenv("ZUP_SQL_DB", "ZUP30"),
|
||||||
"user": os.getenv("ZUP_SQL_USER", "scud_reader"), # Логин SQL Server
|
"user": os.getenv("ZUP_SQL_USER", "scud_reader"),
|
||||||
"password": os.getenv("ZUP_SQL_PASS", "Rhfcysq90"), # Пароль SQL Server
|
"password": os.getenv("ZUP_SQL_PASS", "Rhfcysq90"),
|
||||||
"trust_server_certificate": "yes", # Доверять сертификату сервера
|
"trust_server_certificate": "yes",
|
||||||
"encrypt": "no" # "yes", если на MS SQL включено обязательное SSL-шифрование
|
"encrypt": "no"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
def find_dated_file(prefix, date_str, search_dirs=[ZUP_1C_DIR, SCUD_DIR, DATA_DIR, "."]):
|
def find_dated_file(prefix, date_str, search_dirs=None):
|
||||||
"""
|
if search_dirs is None:
|
||||||
Ищет файлы по префиксу и дате (поддерживает и 30.07.2026, и 30_07_2026).
|
search_dirs = [ZUP_1C_DIR, SCUD_DIR, DATA_DIR, "."]
|
||||||
"""
|
|
||||||
date_dots = date_str
|
date_dots = str(date_str).replace('_', '.')
|
||||||
date_underscores = date_str.replace('.', '_')
|
date_underscores = date_dots.replace('.', '_')
|
||||||
|
|
||||||
for d in search_dirs:
|
for d in search_dirs:
|
||||||
if not os.path.exists(d):
|
if not os.path.exists(d):
|
||||||
@@ -75,6 +73,18 @@ def clean_scud_fio_light(fio_str):
|
|||||||
|
|
||||||
|
|
||||||
def load_exceptions():
|
def load_exceptions():
|
||||||
|
"""
|
||||||
|
Приоритетно читает исключения и белый список из SQLite таблицы exceptions_registry.
|
||||||
|
При отсутствии таблицы или пустой базе выполняет fallback на exceptions.json.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
from services.exceptions_repo import get_all_exceptions_from_db
|
||||||
|
db_exc = get_all_exceptions_from_db()
|
||||||
|
if any(db_exc.values()):
|
||||||
|
return db_exc
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
import json
|
import json
|
||||||
if os.path.exists(EXCEPTIONS_PATH):
|
if os.path.exists(EXCEPTIONS_PATH):
|
||||||
try:
|
try:
|
||||||
@@ -82,4 +92,4 @@ def load_exceptions():
|
|||||||
return json.load(f)
|
return json.load(f)
|
||||||
except Exception:
|
except Exception:
|
||||||
pass
|
pass
|
||||||
return {"fio": [], "departments": [], "positions": [], "position_keywords": []}
|
return {"fio": [], "departments": [], "positions": [], "position_keywords": [], "include_fio": []}
|
||||||
@@ -0,0 +1,27 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: core/connection.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
ROLE: Единый менеджер подключений к базе данных SQLite (WAL mode, timeouts).
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import sqlite3
|
||||||
|
from config import DATA_DIR
|
||||||
|
|
||||||
|
DB_PATH = os.path.join(DATA_DIR, "scud_orion_ai.db")
|
||||||
|
|
||||||
|
|
||||||
|
def get_connection(row_factory: bool = False) -> sqlite3.Connection:
|
||||||
|
"""
|
||||||
|
Создает оптимизированное подключение к SQLite.
|
||||||
|
row_factory=True возвращает sqlite3.Row для доступа к полям по имени.
|
||||||
|
"""
|
||||||
|
conn = sqlite3.connect(DB_PATH, timeout=30.0)
|
||||||
|
if row_factory:
|
||||||
|
conn.row_factory = sqlite3.Row
|
||||||
|
conn.execute("PRAGMA foreign_keys = ON;")
|
||||||
|
conn.execute("PRAGMA journal_mode = WAL;")
|
||||||
|
conn.execute("PRAGMA synchronous = NORMAL;")
|
||||||
|
return conn
|
||||||
+35
-480
@@ -1,480 +1,35 @@
|
|||||||
import os
|
"""
|
||||||
import sqlite3
|
===============================================================================
|
||||||
import json
|
FILE: core/database.py
|
||||||
import pandas as pd
|
ROLE: Фасад ядра базы данных с полной обратной совместимостью импортов.
|
||||||
from datetime import datetime
|
===============================================================================
|
||||||
from config import DATA_DIR
|
"""
|
||||||
|
|
||||||
DB_PATH = os.path.join(DATA_DIR, "scud_orion_ai.db")
|
from core.connection import get_connection, DB_PATH
|
||||||
|
from core.schema import init_all_tables
|
||||||
|
|
||||||
def get_connection():
|
from core.repositories.scud_repo import (
|
||||||
"""Создает подключение к базе данных SQLite с оптимизированными настройками."""
|
has_scud_logs_for_date,
|
||||||
conn = sqlite3.connect(DB_PATH)
|
has_yesterday_final_snapshot,
|
||||||
conn.execute("PRAGMA foreign_keys = ON;")
|
get_or_create_snapshot_id,
|
||||||
conn.execute("PRAGMA journal_mode = WAL;")
|
save_scud_to_db,
|
||||||
return conn
|
get_latest_snapshot_time,
|
||||||
|
load_scud_from_db_by_snapshot,
|
||||||
|
get_available_snapshots,
|
||||||
def init_db():
|
delete_snapshot_by_id,
|
||||||
"""Создает структуру таблиц и индексов в базе данных."""
|
delete_snapshots_by_date
|
||||||
with get_connection() as conn:
|
)
|
||||||
cursor = conn.cursor()
|
|
||||||
|
from core.repositories.zup_repo import (
|
||||||
cursor.execute("""
|
save_staff_to_db,
|
||||||
CREATE TABLE IF NOT EXISTS scud_logs (
|
load_staff_from_db,
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
save_absences_to_db,
|
||||||
log_date TEXT NOT NULL,
|
load_absences_from_db,
|
||||||
fio TEXT NOT NULL,
|
save_anomalies_to_db,
|
||||||
fio_clean TEXT NOT NULL,
|
get_all_rules_from_db,
|
||||||
department TEXT,
|
add_rule_to_db,
|
||||||
position TEXT,
|
get_department_synonyms_dict,
|
||||||
time_in TEXT,
|
add_department_synonym_to_db
|
||||||
first_activity TEXT DEFAULT '—',
|
)
|
||||||
time_out TEXT,
|
|
||||||
time_in_building TEXT,
|
init_db = init_all_tables
|
||||||
is_present INTEGER NOT NULL,
|
|
||||||
anomaly_flag TEXT DEFAULT 'NONE',
|
|
||||||
snapshot_time TEXT DEFAULT NULL,
|
|
||||||
snapshot_id TEXT DEFAULT NULL,
|
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
|
||||||
""")
|
|
||||||
|
|
||||||
cursor.execute("PRAGMA table_info(scud_logs);")
|
|
||||||
cols = [col[1] for col in cursor.fetchall()]
|
|
||||||
if 'snapshot_id' not in cols:
|
|
||||||
cursor.execute("ALTER TABLE scud_logs ADD COLUMN snapshot_id TEXT DEFAULT NULL;")
|
|
||||||
|
|
||||||
cursor.execute("""
|
|
||||||
CREATE TABLE IF NOT EXISTS zup_staff (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
snapshot_date TEXT NOT NULL,
|
|
||||||
fio TEXT NOT NULL,
|
|
||||||
fio_clean TEXT NOT NULL,
|
|
||||||
department TEXT,
|
|
||||||
position TEXT,
|
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
|
||||||
""")
|
|
||||||
|
|
||||||
cursor.execute("""
|
|
||||||
CREATE TABLE IF NOT EXISTS zup_absences (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
absence_date TEXT NOT NULL,
|
|
||||||
fio TEXT NOT NULL,
|
|
||||||
fio_clean TEXT NOT NULL,
|
|
||||||
absence_type TEXT NOT NULL,
|
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
|
||||||
""")
|
|
||||||
|
|
||||||
cursor.execute("""
|
|
||||||
CREATE TABLE IF NOT EXISTS anomalies_history (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
anomaly_date TEXT NOT NULL,
|
|
||||||
fio TEXT NOT NULL,
|
|
||||||
anomaly_type TEXT NOT NULL,
|
|
||||||
details TEXT NOT NULL,
|
|
||||||
human_status TEXT DEFAULT 'Pending',
|
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
|
||||||
""")
|
|
||||||
|
|
||||||
cursor.execute("""
|
|
||||||
CREATE TABLE IF NOT EXISTS ai_knowledge_base (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
rule_text TEXT UNIQUE NOT NULL,
|
|
||||||
added_by TEXT DEFAULT 'Human',
|
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
|
||||||
""")
|
|
||||||
|
|
||||||
cursor.execute("CREATE INDEX IF NOT EXISTS idx_scud_date ON scud_logs(log_date);")
|
|
||||||
cursor.execute("CREATE INDEX IF NOT EXISTS idx_scud_fio ON scud_logs(fio_clean);")
|
|
||||||
cursor.execute("CREATE INDEX IF NOT EXISTS idx_scud_snapshot ON scud_logs(snapshot_time);")
|
|
||||||
|
|
||||||
conn.commit()
|
|
||||||
|
|
||||||
sync_knowledge_base_to_db()
|
|
||||||
|
|
||||||
def has_scud_logs_for_date(date_str):
|
|
||||||
"""Проверяет, есть ли в базе данные СКУД за указанную дату."""
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("SELECT 1 FROM scud_logs WHERE log_date = ? LIMIT 1", (date_str,))
|
|
||||||
return cursor.fetchone() is not None
|
|
||||||
|
|
||||||
def has_yesterday_final_snapshot(date_str):
|
|
||||||
"""Проверяет, зафиксирован ли уже ИТОГОВЫЙ вчерашний снапшот с индексом Y/22:00."""
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute(
|
|
||||||
"SELECT 1 FROM scud_logs WHERE log_date = ? AND (snapshot_id LIKE 'Y%' OR snapshot_time LIKE '%22:00:00') LIMIT 1",
|
|
||||||
(date_str,)
|
|
||||||
)
|
|
||||||
return cursor.fetchone() is not None
|
|
||||||
|
|
||||||
|
|
||||||
def get_or_create_snapshot_id(snapshot_time, date_str=None, is_yesterday=False):
|
|
||||||
"""
|
|
||||||
Генерирует датированный составной ID снапшота (YYYYMMDD-NNN или YYYYMMDD-NNN).
|
|
||||||
Префикс 'Y' присваивается автоматически, если дата логов (date_str) предшествует дате снятия (snapshot_time).
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
dt_snap = datetime.strptime(snapshot_time, "%Y-%m-%d %H:%M:%S").date()
|
|
||||||
date_prefix = dt_snap.strftime("%Y%m%d")
|
|
||||||
except (ValueError, TypeError):
|
|
||||||
dt_snap = datetime.now().date()
|
|
||||||
date_prefix = dt_snap.strftime("%Y%m%d")
|
|
||||||
|
|
||||||
# Автоматическая проверка: если дата логов раньше даты выгрузки — проставляем Y
|
|
||||||
if date_str:
|
|
||||||
try:
|
|
||||||
dt_log = datetime.strptime(date_str, "%d.%m.%Y").date()
|
|
||||||
if dt_log < dt_snap:
|
|
||||||
is_yesterday = True
|
|
||||||
except Exception:
|
|
||||||
pass
|
|
||||||
|
|
||||||
prefix = "Y" if is_yesterday else ""
|
|
||||||
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
|
|
||||||
if date_str:
|
|
||||||
cursor.execute(
|
|
||||||
"SELECT snapshot_id FROM scud_logs WHERE log_date = ? AND snapshot_time = ? AND snapshot_id IS NOT NULL LIMIT 1",
|
|
||||||
(date_str, snapshot_time)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
cursor.execute(
|
|
||||||
"SELECT snapshot_id FROM scud_logs WHERE snapshot_time = ? AND snapshot_id IS NOT NULL LIMIT 1",
|
|
||||||
(snapshot_time,)
|
|
||||||
)
|
|
||||||
|
|
||||||
row = cursor.fetchone()
|
|
||||||
if row and row[0]:
|
|
||||||
return row[0]
|
|
||||||
|
|
||||||
cursor.execute("""
|
|
||||||
SELECT snapshot_id
|
|
||||||
FROM scud_logs
|
|
||||||
WHERE snapshot_id LIKE ? OR snapshot_id LIKE ?
|
|
||||||
ORDER BY snapshot_id DESC LIMIT 1
|
|
||||||
""", (f"{date_prefix}-%", f"Y{date_prefix}-%"))
|
|
||||||
|
|
||||||
last_row = cursor.fetchone()
|
|
||||||
next_seq = 1
|
|
||||||
|
|
||||||
if last_row and last_row[0]:
|
|
||||||
parts = last_row[0].replace("Y", "").split("-")
|
|
||||||
if len(parts) > 1 and parts[1].isdigit():
|
|
||||||
next_seq = int(parts[1]) + 1
|
|
||||||
|
|
||||||
return f"{prefix}{date_prefix}-{next_seq:03d}"
|
|
||||||
|
|
||||||
|
|
||||||
def save_scud_to_db(df_scud, date_str, snapshot_time=None, is_yesterday=False):
|
|
||||||
"""Сохраняет логи СКУД с автоматическим присвоением составного snapshot_id."""
|
|
||||||
if df_scud is None or df_scud.empty:
|
|
||||||
return
|
|
||||||
|
|
||||||
if not snapshot_time:
|
|
||||||
snapshot_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
||||||
|
|
||||||
# Передаем date_str в генератор snapshot_id
|
|
||||||
snapshot_id = get_or_create_snapshot_id(snapshot_time, date_str=date_str, is_yesterday=is_yesterday)
|
|
||||||
|
|
||||||
data_to_insert = []
|
|
||||||
for _, r in df_scud.iterrows():
|
|
||||||
data_to_insert.append((
|
|
||||||
date_str,
|
|
||||||
r.get('Сотрудник', r.get('fio_raw', '')),
|
|
||||||
r.get('fio_clean', ''),
|
|
||||||
r.get('Подразделение', ''),
|
|
||||||
r.get('Должность', ''),
|
|
||||||
r.get('Начало_дня', 'Нет входа'),
|
|
||||||
r.get('Первая_активность', '—'),
|
|
||||||
r.get('Конец_дня', 'Нет выхода'),
|
|
||||||
r.get('Находился_в_здании', '00:00'),
|
|
||||||
1 if r.get('Пришел', False) else 0,
|
|
||||||
r.get('anomaly_flag', 'NONE'),
|
|
||||||
snapshot_time,
|
|
||||||
snapshot_id
|
|
||||||
))
|
|
||||||
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("DELETE FROM scud_logs WHERE log_date = ? AND snapshot_time = ?", (date_str, snapshot_time))
|
|
||||||
cursor.executemany("""
|
|
||||||
INSERT INTO scud_logs (
|
|
||||||
log_date, fio, fio_clean, department, position,
|
|
||||||
time_in, first_activity, time_out, time_in_building,
|
|
||||||
is_present, anomaly_flag, snapshot_time, snapshot_id
|
|
||||||
)
|
|
||||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
||||||
""", data_to_insert)
|
|
||||||
conn.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def get_latest_snapshot_time(date_str=None):
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
if date_str:
|
|
||||||
cursor.execute("SELECT snapshot_time FROM scud_logs WHERE log_date = ? AND snapshot_time IS NOT NULL ORDER BY snapshot_time DESC LIMIT 1", (date_str,))
|
|
||||||
else:
|
|
||||||
cursor.execute("SELECT snapshot_time FROM scud_logs WHERE snapshot_time IS NOT NULL ORDER BY snapshot_time DESC LIMIT 1")
|
|
||||||
row = cursor.fetchone()
|
|
||||||
return row[0] if row else None
|
|
||||||
|
|
||||||
|
|
||||||
def load_scud_from_db_by_snapshot(date_str, snapshot_param=None):
|
|
||||||
"""
|
|
||||||
Загружает логи СКУД по ID снапшота.
|
|
||||||
Если snapshot_param не задан (или искался прошлый день), приоритетно выбирает
|
|
||||||
финальный вечерний снапшот, начинающийся с буквы 'Y' (срез на 22:00) за дату date_str.
|
|
||||||
"""
|
|
||||||
with get_connection() as conn:
|
|
||||||
df = pd.DataFrame()
|
|
||||||
|
|
||||||
# 1. Если передан конкретный ID снапшота (например, '20260810-001')
|
|
||||||
if snapshot_param:
|
|
||||||
df = pd.read_sql_query(
|
|
||||||
"SELECT * FROM scud_logs WHERE snapshot_id = ?",
|
|
||||||
conn, params=(str(snapshot_param),)
|
|
||||||
)
|
|
||||||
|
|
||||||
# 2. Если по snapshot_param ничего не найдено ИЛИ snapshot_param=None (запрос за вчерашний день):
|
|
||||||
if df.empty and date_str:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
|
|
||||||
# ⭐️ ПРИОРИТЕТ 1: Ищем срез, начальный символ которого 'Y' (вечерний зафиксированный Y-снапшот)
|
|
||||||
cursor.execute(
|
|
||||||
"SELECT snapshot_id FROM scud_logs WHERE log_date = ? AND snapshot_id LIKE 'Y%' ORDER BY id DESC LIMIT 1",
|
|
||||||
(date_str,)
|
|
||||||
)
|
|
||||||
row = cursor.fetchone()
|
|
||||||
|
|
||||||
# ⭐️ ПРИОРИТЕТ 2: Если Y-снапшот не найден, берем самый свежий обычный
|
|
||||||
if not row:
|
|
||||||
cursor.execute(
|
|
||||||
"SELECT snapshot_id FROM scud_logs WHERE log_date = ? ORDER BY id DESC LIMIT 1",
|
|
||||||
(date_str,)
|
|
||||||
)
|
|
||||||
row = cursor.fetchone()
|
|
||||||
|
|
||||||
if row and row[0]:
|
|
||||||
target_snap_id = row[0]
|
|
||||||
df = pd.read_sql_query(
|
|
||||||
"SELECT * FROM scud_logs WHERE snapshot_id = ?",
|
|
||||||
conn, params=(target_snap_id,)
|
|
||||||
)
|
|
||||||
|
|
||||||
# Приводим названия колонок к стандарту
|
|
||||||
if not df.empty:
|
|
||||||
rename_map = {
|
|
||||||
'department': 'Подразделение',
|
|
||||||
'position': 'Должность',
|
|
||||||
'fio': 'Сотрудник',
|
|
||||||
'time_in': 'Начало_дня',
|
|
||||||
'first_activity': 'Первая_активность',
|
|
||||||
'time_out': 'Конец_дня',
|
|
||||||
'duration': 'Находился_в_здании',
|
|
||||||
'is_present': 'Пришел'
|
|
||||||
}
|
|
||||||
# Переименовываем имеющиеся колонки
|
|
||||||
df = df.rename(columns={k: v for k, v in rename_map.items() if k in df.columns})
|
|
||||||
|
|
||||||
# Заполняем недостающие колонки значениями по умолчанию, если их не было в БД
|
|
||||||
required_cols = ['Пришел', 'Начало_дня', 'Первая_активность', 'Конец_дня', 'Находился_в_здании', 'anomaly_flag']
|
|
||||||
for col in required_cols:
|
|
||||||
if col not in df.columns:
|
|
||||||
if col == 'Пришел':
|
|
||||||
df[col] = False
|
|
||||||
else:
|
|
||||||
df[col] = '—'
|
|
||||||
|
|
||||||
# Приводим тип флага Пришел к bool
|
|
||||||
if 'Пришел' in df.columns:
|
|
||||||
df['Пришел'] = df['Пришел'].astype(bool)
|
|
||||||
|
|
||||||
return df
|
|
||||||
|
|
||||||
def get_available_snapshots(date_str=None):
|
|
||||||
"""
|
|
||||||
Возвращает список снапшотов.
|
|
||||||
Сортировка по времени snapshot_time DESC.
|
|
||||||
"""
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
if date_str:
|
|
||||||
cursor.execute("""
|
|
||||||
SELECT
|
|
||||||
snapshot_id,
|
|
||||||
log_date,
|
|
||||||
snapshot_time,
|
|
||||||
COUNT(*) as cnt
|
|
||||||
FROM scud_logs
|
|
||||||
WHERE log_date = ? AND snapshot_time IS NOT NULL
|
|
||||||
GROUP BY snapshot_id, log_date, snapshot_time
|
|
||||||
ORDER BY snapshot_time DESC
|
|
||||||
""", (date_str,))
|
|
||||||
else:
|
|
||||||
cursor.execute("""
|
|
||||||
SELECT
|
|
||||||
snapshot_id,
|
|
||||||
log_date,
|
|
||||||
snapshot_time,
|
|
||||||
COUNT(*) as cnt
|
|
||||||
FROM scud_logs
|
|
||||||
WHERE snapshot_time IS NOT NULL
|
|
||||||
GROUP BY snapshot_id, log_date, snapshot_time
|
|
||||||
ORDER BY snapshot_time DESC
|
|
||||||
""")
|
|
||||||
return cursor.fetchall()
|
|
||||||
|
|
||||||
|
|
||||||
def save_staff_to_db(df_staff, date_str):
|
|
||||||
if df_staff is None or df_staff.empty:
|
|
||||||
return
|
|
||||||
data_to_insert = [
|
|
||||||
(date_str, r.get('ФИО', ''), r.get('fio_clean', ''), r.get('Подразделение', ''), r.get('Должность', ''))
|
|
||||||
for _, r in df_staff.iterrows()
|
|
||||||
]
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("DELETE FROM zup_staff WHERE snapshot_date = ?", (date_str,))
|
|
||||||
cursor.executemany("INSERT INTO zup_staff (snapshot_date, fio, fio_clean, department, position) VALUES (?, ?, ?, ?, ?)", data_to_insert)
|
|
||||||
conn.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def save_absences_to_db(df_absent, date_str):
|
|
||||||
if df_absent is None or df_absent.empty:
|
|
||||||
return
|
|
||||||
data_to_insert = [
|
|
||||||
(date_str, r.get('ФИО', r.get('fio_clean', '')), r.get('fio_clean', ''), r.get('Вид_отсутствия', ''))
|
|
||||||
for _, r in df_absent.iterrows()
|
|
||||||
]
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("DELETE FROM zup_absences WHERE absence_date = ?", (date_str,))
|
|
||||||
cursor.executemany("INSERT INTO zup_absences (absence_date, fio, fio_clean, absence_type) VALUES (?, ?, ?, ?)", data_to_insert)
|
|
||||||
conn.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def save_anomalies_to_db(anomalies_list, date_str):
|
|
||||||
if not anomalies_list:
|
|
||||||
return
|
|
||||||
data_to_insert = [
|
|
||||||
(date_str, a.get('fio', ''), a.get('type', ''), a.get('details', ''))
|
|
||||||
for a in anomalies_list
|
|
||||||
]
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("DELETE FROM anomalies_history WHERE anomaly_date = ?", (date_str,))
|
|
||||||
cursor.executemany("INSERT INTO anomalies_history (anomaly_date, fio, anomaly_type, details) VALUES (?, ?, ?, ?)", data_to_insert)
|
|
||||||
conn.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def get_all_rules_from_db():
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("SELECT rule_text FROM ai_knowledge_base")
|
|
||||||
return [r[0] for r in cursor.fetchall()]
|
|
||||||
|
|
||||||
|
|
||||||
def add_rule_to_db(rule_text, added_by="Human"):
|
|
||||||
if not rule_text or not rule_text.strip():
|
|
||||||
return
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
try:
|
|
||||||
cursor.execute("INSERT OR IGNORE INTO ai_knowledge_base (rule_text, added_by) VALUES (?, ?)", (rule_text.strip(), added_by))
|
|
||||||
conn.commit()
|
|
||||||
except Exception as e:
|
|
||||||
print(f"[⚠️] Ошибка записи правила в БД: {e}")
|
|
||||||
|
|
||||||
|
|
||||||
def sync_knowledge_base_to_db():
|
|
||||||
pass
|
|
||||||
|
|
||||||
def delete_snapshot_by_id(snapshot_id: str):
|
|
||||||
"""Удаляет конкретный снапшот из таблицы scud_logs по его ID."""
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("DELETE FROM scud_logs WHERE snapshot_id = ?", (snapshot_id,))
|
|
||||||
deleted_count = cursor.rowcount
|
|
||||||
conn.commit()
|
|
||||||
print(f"[✓] Удален снапшот [{snapshot_id}]. Удалено строк: {deleted_count}")
|
|
||||||
return deleted_count
|
|
||||||
|
|
||||||
def delete_snapshots_by_date(date_str: str):
|
|
||||||
"""Удаляет все снапшоты за указанную дату (например, '04.08.2026')."""
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
# Удаляем по log_date или по дате внутри snapshot_id
|
|
||||||
cursor.execute("DELETE FROM scud_logs WHERE log_date = ? OR snapshot_id LIKE ?", (date_str, f"%{date_str.replace('.', '')}%"))
|
|
||||||
deleted_count = cursor.rowcount
|
|
||||||
conn.commit()
|
|
||||||
print(f"[✓] Удалены все снапшоты за дату [{date_str}]. Удалено строк: {deleted_count}")
|
|
||||||
return deleted_count
|
|
||||||
def init_department_synonyms_db():
|
|
||||||
"""Создает таблицу синонимов отделов в БД SQLite."""
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("""
|
|
||||||
CREATE TABLE IF NOT EXISTS department_synonyms (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
short_name TEXT UNIQUE NOT NULL,
|
|
||||||
full_name TEXT NOT NULL,
|
|
||||||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
|
||||||
""")
|
|
||||||
# По умолчанию добавляем ОВК -> Отдел внутреннего контроля
|
|
||||||
cursor.execute("""
|
|
||||||
INSERT OR IGNORE INTO department_synonyms (short_name, full_name)
|
|
||||||
VALUES ('овк', 'отдел внутреннего контроля')
|
|
||||||
""")
|
|
||||||
conn.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def get_department_synonyms_dict():
|
|
||||||
"""Возвращает словарь всех изученных синонимов {short_name: full_name}."""
|
|
||||||
init_department_synonyms_db()
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("SELECT LOWER(short_name), LOWER(full_name) FROM department_synonyms")
|
|
||||||
return {row[0]: row[1] for row in cursor.fetchall()}
|
|
||||||
|
|
||||||
|
|
||||||
def add_department_synonym_to_db(short_name, full_name):
|
|
||||||
"""Сохраняет новую пару синонимов отдела в базу SQLite."""
|
|
||||||
init_department_synonyms_db()
|
|
||||||
with get_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute(
|
|
||||||
"INSERT OR REPLACE INTO department_synonyms (short_name, full_name) VALUES (?, ?)",
|
|
||||||
(short_name.strip().lower(), full_name.strip().lower())
|
|
||||||
)
|
|
||||||
conn.commit()
|
|
||||||
print(f"[✓] В базу SQLite добавлен новый синоним отдела: '{short_name}' ⟷ '{full_name}'")
|
|
||||||
|
|
||||||
def load_staff_from_db(date_str):
|
|
||||||
"""Загружает Штатное расписание 1С из базы SQLite за указанную дату."""
|
|
||||||
with get_connection() as conn:
|
|
||||||
df = pd.read_sql_query(
|
|
||||||
"SELECT fio as 'ФИО', fio_clean, department as 'Подразделение', position as 'Должность' FROM zup_staff WHERE snapshot_date = ?",
|
|
||||||
conn, params=(date_str,)
|
|
||||||
)
|
|
||||||
return df if not df.empty else None
|
|
||||||
|
|
||||||
def load_absences_from_db(date_str):
|
|
||||||
"""Загружает документальные отсутствия 1С из базы SQLite за указанную дату."""
|
|
||||||
with get_connection() as conn:
|
|
||||||
df = pd.read_sql_query(
|
|
||||||
"SELECT fio as 'ФИО', fio_clean, absence_type as 'Вид_отсутствия' FROM zup_absences WHERE absence_date = ?",
|
|
||||||
conn, params=(date_str,)
|
|
||||||
)
|
|
||||||
return df if not df.empty else None
|
|
||||||
|
|
||||||
@@ -0,0 +1,227 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: core/repositories/scud_repo.py
|
||||||
|
ROLE: Репозиторий логов СКУД, сохранение и загрузка снапшотов.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
import pandas as pd
|
||||||
|
from core.connection import get_connection
|
||||||
|
|
||||||
|
|
||||||
|
def has_scud_logs_for_date(date_str: str) -> bool:
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT 1 FROM scud_logs WHERE log_date = ? LIMIT 1", (date_str,))
|
||||||
|
return cursor.fetchone() is not None
|
||||||
|
|
||||||
|
|
||||||
|
def has_yesterday_final_snapshot(date_str: str) -> bool:
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute(
|
||||||
|
"SELECT 1 FROM scud_logs WHERE log_date = ? AND (snapshot_id LIKE 'Y%' OR snapshot_time LIKE '%23:59:59' OR snapshot_time LIKE '%22:00:00') LIMIT 1",
|
||||||
|
(date_str,)
|
||||||
|
)
|
||||||
|
return cursor.fetchone() is not None
|
||||||
|
|
||||||
|
|
||||||
|
def get_or_create_snapshot_id(snapshot_time: str, date_str: str = None, is_yesterday: bool = False) -> str:
|
||||||
|
try:
|
||||||
|
dt_snap = datetime.strptime(snapshot_time, "%Y-%m-%d %H:%M:%S").date()
|
||||||
|
date_prefix = dt_snap.strftime("%Y%m%d")
|
||||||
|
except (ValueError, TypeError):
|
||||||
|
dt_snap = datetime.now().date()
|
||||||
|
date_prefix = dt_snap.strftime("%Y%m%d")
|
||||||
|
|
||||||
|
if date_str:
|
||||||
|
try:
|
||||||
|
dt_log = datetime.strptime(date_str, "%d.%m.%Y").date()
|
||||||
|
if dt_log < dt_snap:
|
||||||
|
is_yesterday = True
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
prefix = "Y" if is_yesterday else ""
|
||||||
|
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
if date_str:
|
||||||
|
cursor.execute(
|
||||||
|
"SELECT snapshot_id FROM scud_logs WHERE log_date = ? AND snapshot_time = ? AND snapshot_id IS NOT NULL LIMIT 1",
|
||||||
|
(date_str, snapshot_time)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
cursor.execute(
|
||||||
|
"SELECT snapshot_id FROM scud_logs WHERE snapshot_time = ? AND snapshot_id IS NOT NULL LIMIT 1",
|
||||||
|
(snapshot_time,)
|
||||||
|
)
|
||||||
|
|
||||||
|
row = cursor.fetchone()
|
||||||
|
if row and row[0]:
|
||||||
|
return row[0]
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT snapshot_id FROM scud_logs
|
||||||
|
WHERE snapshot_id LIKE ? OR snapshot_id LIKE ?
|
||||||
|
ORDER BY snapshot_id DESC LIMIT 1
|
||||||
|
""", (f"{date_prefix}-%", f"Y{date_prefix}-%"))
|
||||||
|
|
||||||
|
last_row = cursor.fetchone()
|
||||||
|
next_seq = 1
|
||||||
|
if last_row and last_row[0]:
|
||||||
|
parts = last_row[0].replace("Y", "").split("-")
|
||||||
|
if len(parts) > 1 and parts[1].isdigit():
|
||||||
|
next_seq = int(parts[1]) + 1
|
||||||
|
|
||||||
|
return f"{prefix}{date_prefix}-{next_seq:03d}"
|
||||||
|
|
||||||
|
|
||||||
|
def save_scud_to_db(df_scud: pd.DataFrame, date_str: str, snapshot_time: str = None, is_yesterday: bool = False) -> None:
|
||||||
|
if df_scud is None or df_scud.empty:
|
||||||
|
return
|
||||||
|
|
||||||
|
if not snapshot_time:
|
||||||
|
snapshot_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
|
|
||||||
|
snapshot_id = get_or_create_snapshot_id(snapshot_time, date_str=date_str, is_yesterday=is_yesterday)
|
||||||
|
|
||||||
|
data_to_insert = [
|
||||||
|
(
|
||||||
|
date_str,
|
||||||
|
r.get('Сотрудник', r.get('fio_raw', '')),
|
||||||
|
r.get('fio_clean', ''),
|
||||||
|
r.get('Подразделение', ''),
|
||||||
|
r.get('Должность', ''),
|
||||||
|
r.get('Начало_дня', 'Нет входа'),
|
||||||
|
r.get('Первая_активность', '—'),
|
||||||
|
r.get('Конец_дня', 'Нет выхода'),
|
||||||
|
r.get('Находился_в_здании', '00:00'),
|
||||||
|
1 if r.get('Пришел', False) else 0,
|
||||||
|
r.get('anomaly_flag', 'NONE'),
|
||||||
|
snapshot_time,
|
||||||
|
snapshot_id
|
||||||
|
)
|
||||||
|
for _, r in df_scud.iterrows()
|
||||||
|
]
|
||||||
|
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM scud_logs WHERE log_date = ? AND snapshot_time = ?", (date_str, snapshot_time))
|
||||||
|
cursor.executemany("""
|
||||||
|
INSERT INTO scud_logs (
|
||||||
|
log_date, fio, fio_clean, department, position,
|
||||||
|
time_in, first_activity, time_out, time_in_building,
|
||||||
|
is_present, anomaly_flag, snapshot_time, snapshot_id
|
||||||
|
)
|
||||||
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||||
|
""", data_to_insert)
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def get_latest_snapshot_time(date_str: str = None):
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
if date_str:
|
||||||
|
cursor.execute("SELECT snapshot_time FROM scud_logs WHERE log_date = ? AND snapshot_time IS NOT NULL ORDER BY snapshot_time DESC LIMIT 1", (date_str,))
|
||||||
|
else:
|
||||||
|
cursor.execute("SELECT snapshot_time FROM scud_logs WHERE snapshot_time IS NOT NULL ORDER BY snapshot_time DESC LIMIT 1")
|
||||||
|
row = cursor.fetchone()
|
||||||
|
return row[0] if row else None
|
||||||
|
|
||||||
|
|
||||||
|
def load_scud_from_db_by_snapshot(date_str: str, snapshot_param: str = None) -> pd.DataFrame:
|
||||||
|
with get_connection() as conn:
|
||||||
|
df = pd.DataFrame()
|
||||||
|
|
||||||
|
# 1. Если передан конкретный ID снапшота (например 'Y20260820-004')
|
||||||
|
if snapshot_param:
|
||||||
|
df = pd.read_sql_query(
|
||||||
|
"SELECT * FROM scud_logs WHERE snapshot_id = ?",
|
||||||
|
conn, params=(str(snapshot_param),)
|
||||||
|
)
|
||||||
|
|
||||||
|
# 2. Если ищем за дату (для вчерашнего дня строго ищем Y-снапшот)
|
||||||
|
if df.empty and date_str:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
|
||||||
|
# ⭐️ Жесткий приоритет 1: Ищем снапшот с префиксом 'Y'
|
||||||
|
cursor.execute(
|
||||||
|
"SELECT snapshot_id FROM scud_logs WHERE log_date = ? AND snapshot_id LIKE 'Y%' ORDER BY snapshot_time DESC, id DESC LIMIT 1",
|
||||||
|
(date_str,)
|
||||||
|
)
|
||||||
|
row = cursor.fetchone()
|
||||||
|
|
||||||
|
# Приоритет 2: Если Y нет (например, за сегодня), берем самый свежий по времени
|
||||||
|
if not row:
|
||||||
|
cursor.execute(
|
||||||
|
"SELECT snapshot_id FROM scud_logs WHERE log_date = ? ORDER BY snapshot_time DESC, id DESC LIMIT 1",
|
||||||
|
(date_str,)
|
||||||
|
)
|
||||||
|
row = cursor.fetchone()
|
||||||
|
|
||||||
|
if row and row[0]:
|
||||||
|
target_id = row[0]
|
||||||
|
df = pd.read_sql_query(
|
||||||
|
"SELECT * FROM scud_logs WHERE snapshot_id = ?",
|
||||||
|
conn, params=(target_id,)
|
||||||
|
)
|
||||||
|
|
||||||
|
if not df.empty:
|
||||||
|
rename_map = {
|
||||||
|
'department': 'department_scud',
|
||||||
|
'position': 'Должность',
|
||||||
|
'fio': 'Сотрудник',
|
||||||
|
'time_in': 'Начало_дня',
|
||||||
|
'first_activity': 'Первая_активность',
|
||||||
|
'time_out': 'Конец_дня',
|
||||||
|
'time_in_building': 'Находился_в_здании',
|
||||||
|
'is_present': 'Пришел'
|
||||||
|
}
|
||||||
|
df = df.rename(columns={k: v for k, v in rename_map.items() if k in df.columns})
|
||||||
|
if 'department_scud' in df.columns and 'Подразделение' not in df.columns:
|
||||||
|
df['Подразделение'] = df['department_scud']
|
||||||
|
|
||||||
|
for col in ['Пришел', 'Начало_дня', 'Первая_активность', 'Конец_дня', 'Находился_в_здании', 'anomaly_flag']:
|
||||||
|
if col not in df.columns:
|
||||||
|
df[col] = False if col == 'Пришел' else '—'
|
||||||
|
if 'Пришел' in df.columns:
|
||||||
|
df['Пришел'] = df['Пришел'].astype(bool)
|
||||||
|
|
||||||
|
return df
|
||||||
|
|
||||||
|
|
||||||
|
def get_available_snapshots(date_str: str = None):
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
query = """
|
||||||
|
SELECT snapshot_id, log_date, snapshot_time, COUNT(*) as cnt
|
||||||
|
FROM scud_logs
|
||||||
|
WHERE snapshot_time IS NOT NULL
|
||||||
|
"""
|
||||||
|
params = []
|
||||||
|
if date_str:
|
||||||
|
query += " AND log_date = ?"
|
||||||
|
params.append(date_str)
|
||||||
|
query += " GROUP BY snapshot_id, log_date, snapshot_time ORDER BY snapshot_time DESC"
|
||||||
|
cursor.execute(query, params)
|
||||||
|
return cursor.fetchall()
|
||||||
|
|
||||||
|
|
||||||
|
def delete_snapshot_by_id(snapshot_id: str) -> int:
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM scud_logs WHERE snapshot_id = ?", (snapshot_id,))
|
||||||
|
cnt = cursor.rowcount
|
||||||
|
conn.commit()
|
||||||
|
return cnt
|
||||||
|
|
||||||
|
|
||||||
|
def delete_snapshots_by_date(date_str: str) -> int:
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM scud_logs WHERE log_date = ? OR snapshot_id LIKE ?", (date_str, f"%{date_str.replace('.', '')}%"))
|
||||||
|
cnt = cursor.rowcount
|
||||||
|
conn.commit()
|
||||||
|
return cnt
|
||||||
@@ -0,0 +1,97 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: core/repositories/zup_repo.py
|
||||||
|
ROLE: Репозиторий кадровых данных 1С:ЗУП, аномалий и базы знаний.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import pandas as pd
|
||||||
|
from typing import List, Dict, Any
|
||||||
|
from core.connection import get_connection
|
||||||
|
|
||||||
|
|
||||||
|
def save_staff_to_db(df_staff: pd.DataFrame, date_str: str) -> None:
|
||||||
|
if df_staff is None or df_staff.empty:
|
||||||
|
return
|
||||||
|
data = [
|
||||||
|
(date_str, r.get('ФИО', ''), r.get('fio_clean', ''), r.get('Подразделение', ''), r.get('Должность', ''))
|
||||||
|
for _, r in df_staff.iterrows()
|
||||||
|
]
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM zup_staff WHERE snapshot_date = ?", (date_str,))
|
||||||
|
cursor.executemany("INSERT INTO zup_staff (snapshot_date, fio, fio_clean, department, position) VALUES (?, ?, ?, ?, ?)", data)
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def load_staff_from_db(date_str: str) -> pd.DataFrame:
|
||||||
|
with get_connection() as conn:
|
||||||
|
df = pd.read_sql_query(
|
||||||
|
"SELECT fio as 'ФИО', fio_clean, department as 'Подразделение', position as 'Должность' FROM zup_staff WHERE snapshot_date = ?",
|
||||||
|
conn, params=(date_str,)
|
||||||
|
)
|
||||||
|
return df if not df.empty else None
|
||||||
|
|
||||||
|
|
||||||
|
def save_absences_to_db(df_absent: pd.DataFrame, date_str: str) -> None:
|
||||||
|
if df_absent is None or df_absent.empty:
|
||||||
|
return
|
||||||
|
data = [
|
||||||
|
(date_str, r.get('ФИО', r.get('fio_clean', '')), r.get('fio_clean', ''), r.get('Вид_отсутствия', ''))
|
||||||
|
for _, r in df_absent.iterrows()
|
||||||
|
]
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM zup_absences WHERE absence_date = ?", (date_str,))
|
||||||
|
cursor.executemany("INSERT INTO zup_absences (absence_date, fio, fio_clean, absence_type) VALUES (?, ?, ?, ?)", data)
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def load_absences_from_db(date_str: str) -> pd.DataFrame:
|
||||||
|
with get_connection() as conn:
|
||||||
|
df = pd.read_sql_query(
|
||||||
|
"SELECT fio as 'ФИО', fio_clean, absence_type as 'Вид_отсутствия' FROM zup_absences WHERE absence_date = ?",
|
||||||
|
conn, params=(date_str,)
|
||||||
|
)
|
||||||
|
return df if not df.empty else None
|
||||||
|
|
||||||
|
|
||||||
|
def save_anomalies_to_db(anomalies_list: List[Dict[str, Any]], date_str: str) -> None:
|
||||||
|
if not anomalies_list:
|
||||||
|
return
|
||||||
|
data = [(date_str, a.get('fio', ''), a.get('type', ''), a.get('details', '')) for a in anomalies_list]
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM anomalies_history WHERE anomaly_date = ?", (date_str,))
|
||||||
|
cursor.executemany("INSERT INTO anomalies_history (anomaly_date, fio, anomaly_type, details) VALUES (?, ?, ?, ?)", data)
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def get_all_rules_from_db() -> List[str]:
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT rule_text FROM ai_knowledge_base")
|
||||||
|
return [r[0] for r in cursor.fetchall()]
|
||||||
|
|
||||||
|
|
||||||
|
def add_rule_to_db(rule_text: str, added_by: str = "Human") -> None:
|
||||||
|
if not rule_text or not rule_text.strip():
|
||||||
|
return
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("INSERT OR IGNORE INTO ai_knowledge_base (rule_text, added_by) VALUES (?, ?)", (rule_text.strip(), added_by))
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def get_department_synonyms_dict() -> Dict[str, str]:
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT LOWER(short_name), LOWER(full_name) FROM department_synonyms")
|
||||||
|
return {row[0]: row[1] for row in cursor.fetchall()}
|
||||||
|
|
||||||
|
|
||||||
|
def add_department_synonym_to_db(short_name: str, full_name: str) -> None:
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("INSERT OR REPLACE INTO department_synonyms (short_name, full_name) VALUES (?, ?)", (short_name.strip().lower(), full_name.strip().lower()))
|
||||||
|
conn.commit()
|
||||||
+176
@@ -0,0 +1,176 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: core/schema.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
ROLE: DDL-схемы таблиц, создание индексов и инициализация базы данных.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from core.connection import get_connection
|
||||||
|
|
||||||
|
logger = logging.getLogger("DB_SCHEMA")
|
||||||
|
|
||||||
|
|
||||||
|
def init_all_tables() -> None:
|
||||||
|
"""Инициализирует все таблицы и индексы системы."""
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
|
||||||
|
# 1. Логи СКУД
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS scud_logs (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
log_date TEXT NOT NULL,
|
||||||
|
fio TEXT NOT NULL,
|
||||||
|
fio_clean TEXT NOT NULL,
|
||||||
|
department TEXT,
|
||||||
|
position TEXT,
|
||||||
|
time_in TEXT,
|
||||||
|
first_activity TEXT DEFAULT '—',
|
||||||
|
time_out TEXT,
|
||||||
|
time_in_building TEXT,
|
||||||
|
is_present INTEGER NOT NULL,
|
||||||
|
anomaly_flag TEXT DEFAULT 'NONE',
|
||||||
|
snapshot_time TEXT DEFAULT NULL,
|
||||||
|
snapshot_id TEXT DEFAULT NULL,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
# 2. Кадровые реестры 1С
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS zup_staff (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
snapshot_date TEXT NOT NULL,
|
||||||
|
fio TEXT NOT NULL,
|
||||||
|
fio_clean TEXT NOT NULL,
|
||||||
|
department TEXT,
|
||||||
|
position TEXT,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS zup_absences (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
absence_date TEXT NOT NULL,
|
||||||
|
fio TEXT NOT NULL,
|
||||||
|
fio_clean TEXT NOT NULL,
|
||||||
|
absence_type TEXT NOT NULL,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
# 3. Аномалии, база знаний и синонимы
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS anomalies_history (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
anomaly_date TEXT NOT NULL,
|
||||||
|
fio TEXT NOT NULL,
|
||||||
|
anomaly_type TEXT NOT NULL,
|
||||||
|
details TEXT NOT NULL,
|
||||||
|
human_status TEXT DEFAULT 'Pending',
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS ai_knowledge_base (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
rule_text TEXT UNIQUE NOT NULL,
|
||||||
|
added_by TEXT DEFAULT 'Human',
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS department_synonyms (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
short_name TEXT UNIQUE NOT NULL,
|
||||||
|
full_name TEXT NOT NULL,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
# 4. Исключения и Кэш сопоставлений личностей (ИИ / Ручной)
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS exceptions_registry (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
category TEXT NOT NULL,
|
||||||
|
value TEXT NOT NULL,
|
||||||
|
comment TEXT,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
UNIQUE(category, value)
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS person_identity_mapping (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
scud_fio TEXT NOT NULL,
|
||||||
|
zup_fio TEXT NOT NULL,
|
||||||
|
scud_dept TEXT,
|
||||||
|
zup_dept TEXT,
|
||||||
|
match_source TEXT DEFAULT 'AI', -- 'AI', 'MANUAL', 'EXACT'
|
||||||
|
status TEXT DEFAULT 'ACTIVE', -- 'ACTIVE', 'PROPOSED', 'REJECTED'
|
||||||
|
confidence REAL DEFAULT 1.0,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
UNIQUE(scud_fio, zup_fio)
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
# 5. Узлы системного промпта, сессии, сообщения и задачи
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS system_prompt_nodes (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
prompt_name TEXT DEFAULT 'main_agent',
|
||||||
|
section_id INTEGER NOT NULL,
|
||||||
|
item_id INTEGER NOT NULL,
|
||||||
|
content TEXT NOT NULL,
|
||||||
|
is_active INTEGER DEFAULT 1,
|
||||||
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
UNIQUE(prompt_name, section_id, item_id)
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS session_states (
|
||||||
|
session_id TEXT PRIMARY KEY,
|
||||||
|
state_type TEXT NOT NULL,
|
||||||
|
pending_data TEXT,
|
||||||
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS chat_messages (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
session_id TEXT NOT NULL,
|
||||||
|
role TEXT NOT NULL,
|
||||||
|
content TEXT NOT NULL,
|
||||||
|
is_ephemeral INTEGER DEFAULT 0,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS tasks (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
task_id TEXT,
|
||||||
|
module TEXT DEFAULT 'general',
|
||||||
|
title TEXT NOT NULL,
|
||||||
|
priority TEXT DEFAULT 'MEDIUM',
|
||||||
|
status TEXT DEFAULT 'BACKLOG',
|
||||||
|
due_date TEXT,
|
||||||
|
user_id INTEGER DEFAULT 1,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
|
||||||
|
# 6. Индексы
|
||||||
|
cursor.execute("CREATE INDEX IF NOT EXISTS idx_scud_date ON scud_logs(log_date);")
|
||||||
|
cursor.execute("CREATE INDEX IF NOT EXISTS idx_scud_fio ON scud_logs(fio_clean);")
|
||||||
|
cursor.execute("CREATE INDEX IF NOT EXISTS idx_mapping_scud ON person_identity_mapping(scud_fio);")
|
||||||
|
|
||||||
|
conn.commit()
|
||||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -11,4 +11,6 @@ fio,reason,note
|
|||||||
Ковалев Владимир Владимирович,Удаленная работа,Постоянная удаленка
|
Ковалев Владимир Владимирович,Удаленная работа,Постоянная удаленка
|
||||||
Кожокарь Татьяна Юрьевна,Удаленная работа,Постоянная удаленка
|
Кожокарь Татьяна Юрьевна,Удаленная работа,Постоянная удаленка
|
||||||
Субетто Юлия Викторовна,Удаленная работа,Постоянная удаленка
|
Субетто Юлия Викторовна,Удаленная работа,Постоянная удаленка
|
||||||
Ароян Станислав Месропович,Удаленная работа,Временная удаленка
|
Чуркина Елена Геннадьевна,Удаленная работа,Постоянная удаленка
|
||||||
|
Пушков Александр Александрович,Удаленная работа,Временная удаленка с 26.08.2026 по 04.09.2026
|
||||||
|
Сысоев Алексей Валерьевич,Удаленная работа,Временная удаленка с 25.08.2026 по 27.08.2026
|
||||||
|
@@ -0,0 +1,150 @@
|
|||||||
|
# 📋 История изменений (CHANGELOG)
|
||||||
|
|
||||||
|
Все ключевые изменения архитектуры, инструментов и модулей проекта SCUD Orion AI фиксируются в данном файле.
|
||||||
|
|
||||||
|
## [3.2.0] — 2026-08-27
|
||||||
|
### 🧠 Интеллектуальный арбитраж личностей, агрегация мульти-пропусков и автозакрытие смен
|
||||||
|
|
||||||
|
#### ✨ Добавлено:
|
||||||
|
- **Кэш сопоставлений личностей (`person_identity_mapping`)**:
|
||||||
|
- Создана таблица в SQLite (`core/schema.py`) для постоянного хранения подтвержденных связок между сырыми ФИО СКУД и эталонными ФИО 1С:ЗУП.
|
||||||
|
- Поддержка мгновенного извлечения проверенных сопоставлений в `merger.py` (время выборки < 1 мс).
|
||||||
|
- **Агрегация дубликатов пропусков и мульти-учеток СКУД (`merger.py`)**:
|
||||||
|
- Функция `aggregate_scud_by_person` объединяет все карты/записи одного физлица: фиксирует самый ранний вход `min(valid_ins)`, самый поздний выход `max(valid_outs)` и итоговый статус присутствия.
|
||||||
|
- Исключено ложное попадание сотрудников с несколькими пропусками (например, Кондрашова Е.В.) в категорию «неизвестно».
|
||||||
|
- **Интеллектуальный выбор ставки совместителей 1С (`merger.py`)**:
|
||||||
|
- Функция `select_best_zup_position` сопоставляет множественные должности одного сотрудника в 1С со СКУД и подтягивает ставку того подразделения, где человек находится физически.
|
||||||
|
- **Автозакрытие незакрытых смен по Правилу 8.5ч (`excel_exporter.py`)**:
|
||||||
|
- При наличии утреннего входа и отсутствии вечернего выхода у офисных сотрудников в детальном отчете за вчера расчетный выход ставится как `Вход + 8ч 30мин`, норма 8 часов и нулевое отклонение `0:00` (вместо ложного штрафа `-8:00`).
|
||||||
|
- **Теневой ИИ-арбитраж нераспознанных ФИО (`text_reporter.py`)**:
|
||||||
|
- Модуль `find_ai_identity_suggestions` выявляет нечеткие совпадения и опечатки операторов СКУД, выводя рекомендации для кадровой службы в блоке «💡 Предложения ИИ по сопоставлению ФИО».
|
||||||
|
- **Расширение CLI-утилиты (`scripts/db_cli.py`)**:
|
||||||
|
- Добавлены команды управления связками: `python scripts/db_cli.py mapping [list|add|del]`.
|
||||||
|
|
||||||
|
#### 🔧 Изменено:
|
||||||
|
- **`services/scud_etl/anomaly_detector.py`**:
|
||||||
|
- Добавлено авто-детектирование задвоенных карточек СКУД (`DUPLICATE_SCUD_CARD`) с передачей информации в ИИ-сводку.
|
||||||
|
- **`main_etl.py`**:
|
||||||
|
- Восстановлена передача сырых датасетов (`raw_scud_df`, `raw_staff_df`, `raw_absent_df`) в ИИ-аудитор и генерация диагностического дампа `СКУД_Сырые_данные_ДД.ММ.ГГГГ.xlsx`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## [3.1.0] — 2026-08-25
|
||||||
|
### 🛡️ Реляционный реестр исключений, Белый список и отказоустойчивость выгрузок
|
||||||
|
|
||||||
|
#### ✨ Добавлено:
|
||||||
|
- **Реляционный реестр исключений (`exceptions_registry`)**:
|
||||||
|
- Перенос исключений (`departments`, `positions`, `fio`, `position_keywords`) и белого списка (`include_fio`) из статического JSON в базу данных SQLite (`core/schema.py`).
|
||||||
|
- Репозиторий `services/exceptions_repo.py` с функциями CRUD и авто-миграцией из `exceptions.json`.
|
||||||
|
- **Белый список (`include_fio`)**:
|
||||||
|
- Поддержка явного включения сотрудников в расчет контроллинга в обход должностных и отдельских фильтров (Тарасенко А.А., Журиков М.Н.).
|
||||||
|
- **Управление исключениями в CLI (`scripts/db_cli.py`)**:
|
||||||
|
- Добавлены команды `exceptions list`, `exceptions add -c ... -v ...`, `exceptions del`, `exceptions sync`.
|
||||||
|
- **Резервный загрузчик кадровых отсутствий (`data_loader.py`)**:
|
||||||
|
- Реализован каскадный fallback: `MS SQL 1C:ЗУП` (таймаут 30с) $\rightarrow$ резервный парсинг скопированных Excel-файлов `Отсутствия_*.xlsx` $\rightarrow$ `static_reason_workers.csv`.
|
||||||
|
|
||||||
|
#### 🔧 Изменено:
|
||||||
|
- **Структура отчетов Excel (`excel_exporter.py`)**:
|
||||||
|
- Изменен базовый путь сохранения отчетов на `output/reports/ГОД/МЕСЯЦ/`.
|
||||||
|
- В Ежедневной сводке блок «В том числе на удаленной работе» вынесен под «Итого на работе», а блок «Исключения» размещен в самом низу с поддержкой группировки.
|
||||||
|
- Добавлена защита от блокировок Excel (`PermissionError`/`OSError`): при занятости файла создается резервная копия с временной меткой.
|
||||||
|
- **Подключение к СКУД (`services/scud_export.py`)**:
|
||||||
|
- Актуализирован адрес сервера MS SQL Орион (`172.16.200.147\SQL`).
|
||||||
|
- Добавлена защита инициализации файлового логгера от ошибок доступа прав пользователей.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## [3.0.0] — 2026-08-21
|
||||||
|
### 🚀 Глубокий архитектурный рефакторинг (Unified Clean Architecture)
|
||||||
|
|
||||||
|
#### Добавлено:
|
||||||
|
- **Пакет ядра `core/`:**
|
||||||
|
- `core/connection.py`: централизованный менеджер SQLite-подключений с поддержкой WAL-режима и row_factory.
|
||||||
|
- `core/schema.py`: единая точка инициализации схемы данных и индексов.
|
||||||
|
- `core/repositories/`: низкоуровневые репозитории `scud_repo.py` и `zup_repo.py`.
|
||||||
|
- **Доменные сервисы `services/`:**
|
||||||
|
- `services/tasks/`: полная изоляция логики задач (CRUD, Markdown-экспорт, генерация `ROADMAP.md`).
|
||||||
|
- `services/prompts/`: доменный сервис системного промпта, поддержка превью, генерация HTML-Diff и пакетное удаление узлов (`BATCH_DELETE`).
|
||||||
|
- `services/snapshots/`: бизнес-логика срезов СКУД с защитой финишных Y-снапшотов.
|
||||||
|
- `services/knowledge/`: сервис базы знаний правил и синонимов отделов.
|
||||||
|
- `services/scud_etl/`: модульный ETL-конвейер (`pipeline.py`, `anomaly_detector.py`, `merger.py`, `sql_queries.py`).
|
||||||
|
- **ИИ-оркестратор `modules/ai_engine/`:**
|
||||||
|
- Декомпозиция монолитного `agent.py` на узкие обработчики `handlers/` и динамический контекст-билдер `context_builder.py`.
|
||||||
|
- Двухконтурная обработка: нативный Function Calling + семантический страхующий инжектор (`tool_injector.py`).
|
||||||
|
- Интеллектуальный Topic Drift менеджер с вежливым напоминанием на 3-й шаг и автоочисткой эфемерных сообщений на 4-й.
|
||||||
|
- **Интерфейс и UI:**
|
||||||
|
- Кнопка «✏️ Редактировать» прямо в карточке просмотра системного промпта с инлайн-редактором.
|
||||||
|
|
||||||
|
#### Исправлено:
|
||||||
|
- Устранена утечка системного промпта в постоянный контекст (`chat_messages`).
|
||||||
|
- Исправлены ошибки несоответствия сигнатур (`TypeError: create_prompt_preview`, `save_full_prompt_draft`, `get_snapshots_registry`) в Fast-Path.
|
||||||
|
- Устранена галлюцинация слов-паразитов («Почемучка», «Почемучто») на уровне валидации ответа.
|
||||||
|
- Исправлено устаревшее противоречивое правило 2.3 в базе данных, блокировавшее интерактивное удаление задач.
|
||||||
|
|
||||||
|
#### Удалено:
|
||||||
|
- Устаревшие файлы-дубли: `modules/web_api/llm/db/db_tasks.py`, `db_snapshots.py`, `prompt_merger.py`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## [2026-08-19] — Инлайн-редактор задач, интеллектуальный ContextManager и стабилизация Function Calling
|
||||||
|
|
||||||
|
### ✨ Добавлено
|
||||||
|
* **Инлайн-редактирование задач (`task_widget.js` / `tasks.py`)**:
|
||||||
|
- Добавлена кнопка редактирования (`✏️`) на каждой карточке задачи.
|
||||||
|
- Карточка трансформируется в инлайн-форму с тремя полями: текст задачи, HTML5-пикер календаря планируемого срока и выпадающий список важности (`LOW`, `MEDIUM`, `HIGH`, `CRITICAL`).
|
||||||
|
- Добавлен REST API эндпоинт `PATCH /api/v1/tasks/{task_id}` с поддержкой комплексного обновления полей в SQLite.
|
||||||
|
- В карточки выведено отображение даты создания задачи.
|
||||||
|
* **Интеллектуальный ContextManager (`context_manager.py`)**:
|
||||||
|
- Служебные запросы и карточки инструментов сохраняются синхронными парами (`is_ephemeral = 1`) и атомарно удаляются при завершении операций.
|
||||||
|
- Свободные диалоги оператора (включая темы Topic Drift) сохраняются в постоянную историю (`is_ephemeral = 0`) и не стираются при закрытии сессий промпта/задач.
|
||||||
|
* **Поддержка Y-снапшотов (`db_snapshots.py`)**:
|
||||||
|
- Исправлен SQL-запрос `db_get_snapshots` для точного извлечения ночных и фоновых срезов с префиксом `Y` (например, `Y20260817-003`).
|
||||||
|
|
||||||
|
### 🔧 Изменено
|
||||||
|
* **`modules/web_api/llm/agent.py`**:
|
||||||
|
- Полностью удалены эвристические текстовые фильтры и принудительные сортировки инструментов.
|
||||||
|
- Настроен нативный возврат и исполнение инструментов без зацикливания модели.
|
||||||
|
- Устранена ошибка `UnboundLocalError` при вызове справочных инструментов.
|
||||||
|
* **`modules/web_api/llm/schemas.py`**:
|
||||||
|
- Очищены описания инструментов от агрессивных директив и капса, стабилизирована селективность Function Calling для Qwen 2.5.
|
||||||
|
* **`modules/web_api/llm/core/ollama_client.py`**:
|
||||||
|
- Оптимизированы параметры генерации для Function Calling (`repeat_penalty: 1.0`, `presence_penalty: 0.0`, `temperature: 0.0`), что исключило спам несколькими вызовами инструментов на один запрос.
|
||||||
|
* **`modules/web_api/static/js/chat/task_widget.js`**:
|
||||||
|
- Фильтр по умолчанию переведен на вкладку **«В работе»** (`IN_PROGRESS`).
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## [2026-08-18] — Двухфазный жизненный цикл сессий, двусторонний Diff и Topic Drift Guard
|
||||||
|
|
||||||
|
### ✨ Добавлено
|
||||||
|
* **Двусторонний клиентский Diff (`core.js`)**: Реализован алгоритм сравнения черновика с baseline-промптом. Удаленные строки остаются на экране с красным зачеркиванием `[УДАЛЕНИЕ]`, а добавленные/измененные строки выделяются красной рамкой с авто-отступами.
|
||||||
|
* **Topic Drift Guard (`agent.py`)**: Внедрен двухуровневый контроль смены темы:
|
||||||
|
- Мгновенный сброс стейта и зачистка эфемерных сообщений при вызове сторонних инструментов (задачи, снапшоты, база).
|
||||||
|
- Мягкий счетчик отвлечений (`idle_turns` = 3) в свободном диалоге с выводом напоминания и кнопок из `tool_action_registry`.
|
||||||
|
* **Команда `context purge` в CLI (`db_cli.py`)**: Добавлен флаг `--all` для полной очистки таблицы `chat_messages` и сброса стейтов сессий.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## [2026-08-17] — Внедрение CLI-инспекции контекста и переход на нативный Function Calling
|
||||||
|
|
||||||
|
### ✨ Добавлено
|
||||||
|
* **`scripts/db_cli.py`**: Добавлена команда `context` для инспекции истории сообщений `chat_messages` по сессиям с визуальным отображением флагов `[ЭФЕМЕРНОЕ]`.
|
||||||
|
* **`ROADMAP.md`**: Сформирована актуальная дорожная карта с планом отказа от регулярных выражений в пользу нативного Function Calling.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## [2026-08-15] — Реляционная архитектура системного промпта
|
||||||
|
|
||||||
|
### ✨ Добавлено
|
||||||
|
* **`system_prompt_nodes`**: Создана таблица реляционных узлов промпта с индексацией по разделам и пунктам (`section_id`, `item_id`).
|
||||||
|
* **`db_prompt_node_edit`**: Добавлен инструмент точечного добавления, изменения и удаления пунктов системного промпта.
|
||||||
|
* **Инлайн-редактор**: Добавлен визуальный diff-просмотр изменений с подсветкой и кнопкой `[✏️ Редактировать]`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## [2026-08-10] — Механизм эфемерных сообщений и очистки диалога
|
||||||
|
|
||||||
|
### ✨ Добавлено
|
||||||
|
* **Эфемерные сообщения (`is_ephemeral`)**: Маркировка временных сервисных ответов и черновиков.
|
||||||
|
* **`db_purge_ephemeral_messages`**: Механизм физической зачистки временных записей из SQLite при завершении операций.
|
||||||
|
* **Интерактивные виджеты**: Поддержка карточек задач (`TASK_INTERACTIVE_CARD`) с кнопками быстрого действия.
|
||||||
@@ -0,0 +1,185 @@
|
|||||||
|
# **PROJECT BRAIN: SCUD Orion AI & Context API (Master Technical & Architectural Manifesto)**
|
||||||
|
|
||||||
|
**Версия документа:** 5.0 (Ultimate Developer & LLM Context Engine)
|
||||||
|
**Назначение:** Исчерпывающий контекстный дамп проекта для немедленного продолжения разработки в любой передовой LLM без потери исторического контекста, с полным пониманием бизнес-логики, схем баз данных, подводных камней и нерешенных задач.
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**1\. Архитектурный паспорт и инфраструктурный профиль**
|
||||||
|
|
||||||
|
| Компонент | Параметры и стек |
|
||||||
|
| :---- | :---- |
|
||||||
|
| **Среда и ОС** | Linux Docker Home (Debian/Ubuntu), Python 3.11/3.12 в виртуальном окружении venv, Git/Gitea. Разработка ведется через VS Code Remote-SSH. |
|
||||||
|
| **Локальная СУБД проекта** | SQLite (data/scud\_orion\_ai.db), режим PRAGMA journal\_mode \= WAL, PRAGMA synchronous \= NORMAL, PRAGMA foreign\_keys \= ON. |
|
||||||
|
| **СКУД Источник (MSSQL)** | MS SQL Server 172.16.31.221\\SQL (база Orion-14.01.21-1), драйвер ODBC Driver 18 for SQL Server (TrustServerCertificate=yes, Encrypt=no). Доступ через pyodbc. |
|
||||||
|
| **1С:ЗУП 3.1 Источник (MSSQL)** | MS SQL Server ACCOUNT-01 (база ZUP30, пользователь scud\_reader). Прямое чтение таблиц dbo.\_Reference299 (сотрудники) и dbo.\_InfoRg16921 (состояния отсутствий). Резервная Samba-шара: /mnt/scud\_share. |
|
||||||
|
| **Локальный контур LLM** | Ollama на http://192.168.11.3:11434 (GPU NVIDIA RTX 5070 Ti 16GB). Модель qwen2.5:14b (основной агент, Function Calling, анализ аномалий), qwen2.5vl:7b-q8\_0 (OCR документов/картинок). |
|
||||||
|
| **Web API & Фронтенд** | FastAPI (Uvicorn), JWT (HS256, 30 дней), Passlib (bcrypt). Фронтенд: Single Page Application (HTML5, Tailwind CSS, Vanilla JS) с Generative UI виджетами. |
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**2\. Энциклопедия подводных камней, инцидентов и принятых инженерных решений**
|
||||||
|
|
||||||
|
### **2.1. Галлюцинации LLM и переход к парадигме Python-Driven Diff**
|
||||||
|
|
||||||
|
> * **Проблема:** Попытка передать списки в 300 ФИО напрямую в промпт модели с командой «сравни списки» приводила к потере контекста, выдуманным опечаткам (*«ЛЕНМОРНИИПРОЕКТ ОЭС Варламова Мария»*) и плейсхолдерам ('Васильев Л.И.', ...).
|
||||||
|
> * **Решение:** Утверждена строгая гибридная схема:
|
||||||
|
1. **Детерминированный слой (Python / Pandas / SQL):** читает файлы, нормализует строки, агрегирует совмещения и мгновенно находит 95%+ точных соответствий.
|
||||||
|
2. **Интеллектуальный слой (Qwen 2.5:14b):** вызывается *только на спорный несовпавший остаток* (1–5 записей) в виде строгого JSON-интерфейса Entity Resolution.
|
||||||
|
|
||||||
|
### **2.2. Символьная нормализация и кадровые совмещения (Кейс Балаевой О.Г.)**
|
||||||
|
|
||||||
|
> * **Проблема символов:** Латинские омоглифы (буквы клавиатуры a, c, e, o, p, x, y вместо русских), разделители 1С в скобках ((осн.), (совм.), (внешн.)), неразрывные пробелы \\xa0, буква ё и случайный ввод цифры 0 вместо буквы О.
|
||||||
|
> * **Решение (normalize\_fio):** Очистка скобок через регулярное выражение re.sub(r'\\(.\*?\\)', '', fio), замена латиницы через str.maketrans('aceopxyABCEHKMOPTX', 'асеорхуАВСЕНКМОРТХ'), замена ё-\>е и удаление лишних пробелов.
|
||||||
|
> * **Коллизия 270 vs 271:** В СКУД 270 физических лиц (один пропуск на человека), в 1С — 271 строка. Сотрудник **Балаева Ольга Геннадьевна** числится на двух ставках в разных отделах (*Руководитель управления персоналом* и *Руководитель режимно-секретного отдела*).
|
||||||
|
> * **Решение:** Внедрена группировка датафрейма 1С по fio\_clean с агрегацией должностей и подразделений через разделитель " / ".
|
||||||
|
|
||||||
|
### **2.3. Защита от коллизий полных тёзок и ложных связок однофамильцев (Кейс Семеновых)**
|
||||||
|
|
||||||
|
> * **Инцидент:** Неограниченная строгими правилами LLM связала двух разных сотрудниц — *«Семенову Юлию Борисовну»* (работает) и *«Семенову Дину Дмитриевну»* (в отпуске), посчитав разные имена опечаткой из\-за одинаковой фамилии. В результате Юлии Борисовне ошибочно был приписан статус отпуска.
|
||||||
|
> * **Двухконтурный барьер:**
|
||||||
|
1. **В промпте:** Запрет объединения однофамильцев с разными именами/отчествами.
|
||||||
|
2. **В Python:** Проверка совпадения первых букв имени и отчества. При несовпадении сопоставление ИИ аннулируется с логированием ⚠️ \[Блокировка фейка\].
|
||||||
|
|
||||||
|
### **2.4. Вынесение исключений в exceptions.json и синонимы отделов**
|
||||||
|
|
||||||
|
> * Служебный персонал (отдел «ЭТО», уборщики производственных и служебных помещений, клинеры, дворники, гардеробщики) вынесен в exceptions.json.
|
||||||
|
> * Для сопоставления кратких аббревиатур СКУД и полных названий отделов 1С (*«ОВК» ⟷ «Отдел внутреннего контроля»*) создана таблица department\_synonyms с автоматическим самообучением при работе человека (Human-in-the-Loop).
|
||||||
|
|
||||||
|
### **2.5. Разгадка феномена «исчезающих неизвестных» и аппаратный сбой СКУД 04.08.2026**
|
||||||
|
|
||||||
|
> * **Аномалия:** Утром 04.08.2026 на главном КПП зависли контроллеры входа. У 119 сотрудников не записалось утреннее событие (time\_in \= 'Нет входа'). Из них 57 человек были в отпусках/командировках по 1С. Оставшиеся 74 человека были на работе без утренней отметки. Однако вечерняя сводка показала всего 4–7 «неизвестных».
|
||||||
|
> * **Причина:** Скрипт агрегации видел дневные перемещения (обед, внутренние двери, туалет, вечерний выход) и автоматически проставлял Статус \= 'Присутствовал' и is\_present \= 1, несмотря на то, что колонка Начало\_дня оставалась 'Нет входа'. В итоге 70 человек тихо ушли в «Итого на работе», а системный сбой оборудования был замаскирован.
|
||||||
|
> * **Архитектурный фикс:**
|
||||||
|
1. Введена колонка **first\_activity («Первая активность»)** — вычисление MIN(TimeVal) AS FirstRawEvent по таблице pLogData без учета направления двери.
|
||||||
|
2. Если time\_in \== 'Нет входа', но first\_activity \!= '—', записи ставится anomaly\_flag \= 'ANOMALY\_NO\_IN\_HAS\_ACTIVITY', а is\_present принудительно сбрасывается в 0\.
|
||||||
|
3. Внедрен модуль analyze\_scud\_mass\_failure\_ai(): если доля таких аномалий превышает 5% смены (\>10 чел.), ИИ генерирует системный алерт о массовом сбое контроллеров КПП.
|
||||||
|
> * **Кейс Равина В.Э.:** Главный специалист официально находился в основном отпуске по 1С, утреннего входа на КПП не было, но зафиксирована активность в 14:11:06 и вечерний выход в 17:18:38. Система зарегистрировала истинную кадровую аномалию (физическое присутствие при официальном отпуске).
|
||||||
|
|
||||||
|
### **2.6. Система сквозных датированных снапшотов (срезов)**
|
||||||
|
|
||||||
|
> * **Формат ID:** \[Y\]YYYYMMDD-NNN (например, Y20260804-001 для вчерашнего зафиксированного вечернего среза на 22:00 и 20260805-002 для дневного).
|
||||||
|
> * **Оптимизация:** Функция has\_yesterday\_final\_snapshot() проверяет наличие вчерашнего среза в SQLite: если он есть, повторный тяжелый запрос к MS SQL Орион пропускается.
|
||||||
|
> * **CLI-управление:** Поддержка запуска отчетов по любому историческому снапшоту через ключ \--snapshot \[ID\] и пропуск выгрузки через \--skip-export.
|
||||||
|
|
||||||
|
### **2.7. Web API, Generative UI и управление жизненным циклом сессий (Topic Drift & Ephemeral Buffer)**
|
||||||
|
|
||||||
|
> * **Проблема засорения контекста LLM:** Вывод объемных списков задач, системных промптов и таблиц срезов забивал окно диалога, вызывая галлюцинации и «залипание» модели на старых данных.
|
||||||
|
> * **Решение:**
|
||||||
|
* В chat\_messages введен флаг is\_ephemeral \= 1 для временных сообщений.
|
||||||
|
* Двухфазные сессии (PROMPT\_PREVIEW → PROMPT\_FOLLOWUP) и детерминированный счетчик idle\_turns.
|
||||||
|
* **Context Guard:** На 3-м шаге отвлечения оператора на сторонние темы система отвечает на вопрос и выводит интерактивное напоминание с кнопками \[Завершить настройку\] / \[Показать промпт\]. При idle\_turns \> 3 происходит автоочистка сессии вызовом db\_purge\_ephemeral\_messages().
|
||||||
|
* **Fast-Path:** Прямой перехват команд подтверждения («да», «подтверждаю») и выхода («нет, спасибо») без обращения к LLM (ответ за 0.05 сек).
|
||||||
|
* **Декларативный реестр (tool\_action\_registry):** Хранение шаблонов ответов, кнопок и настроек эфемерности в SQLite.
|
||||||
|
> * **Generative UI (Задачи):** Инструмент db\_get\_tasks возвращает TASK\_INTERACTIVE\_CARD, рендерящий на клиенте живой виджет задач с табами (Все / В работе / В планах / Завершенные), чекбоксами и inline-созданием задач через REST API.
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**3\. Полная схема базы данных SQLite (data/scud\_orion\_ai.db)**
|
||||||
|
|
||||||
|
| Таблица | Назначение и бизнес-логика | Структура полей |
|
||||||
|
| :---- | :---- | :---- |
|
||||||
|
| **scud\_logs** | Логи проходов СКУД, расчетное время в здании и снапшоты. | id INTEGER PK AUTOINCREMENT, log\_date TEXT, fio TEXT, fio\_clean TEXT, department TEXT, position TEXT, time\_in TEXT, first\_activity TEXT DEFAULT '—', time\_out TEXT, time\_in\_building TEXT, is\_present INTEGER, anomaly\_flag TEXT DEFAULT 'NONE', snapshot\_time TEXT, snapshot\_id TEXT, created\_at TIMESTAMP |
|
||||||
|
| **zup\_staff** | Штатное расписание 1С:ЗУП на даты срезов. | id INTEGER PK AUTOINCREMENT, snapshot\_date TEXT, fio TEXT, fio\_clean TEXT, department TEXT, position TEXT, created\_at TIMESTAMP |
|
||||||
|
| **zup\_absences** | Документы отклонений 1С (отпуска, больничные, командировки, декреты). | id INTEGER PK AUTOINCREMENT, absence\_date TEXT, fio TEXT, fio\_clean TEXT, absence\_type TEXT, created\_at TIMESTAMP |
|
||||||
|
| **anomalies\_history** | Журнал выявленных кадровых и аппаратных аномалий. | id INTEGER PK AUTOINCREMENT, anomaly\_date TEXT, fio TEXT, anomaly\_type TEXT, details TEXT, human\_status TEXT DEFAULT 'Pending', created\_at TIMESTAMP |
|
||||||
|
| **ai\_knowledge\_base** | Глобальные правила арбитража и приоритеты компании для ИИ. | id INTEGER PK AUTOINCREMENT, rule\_text TEXT UNIQUE, added\_by TEXT DEFAULT 'Human', created\_at TIMESTAMP |
|
||||||
|
| **system\_prompts** | Версионные системные промпты агента (активный по is\_active=1). | id INTEGER PK AUTOINCREMENT, name TEXT, prompt\_text TEXT, is\_active INTEGER DEFAULT 1, updated\_at TIMESTAMP |
|
||||||
|
| **tool\_action\_registry** | Декларативный реестр шаблонов, кнопок и настроек эфемерности инструментов. | id INTEGER PK AUTOINCREMENT, tool\_name TEXT UNIQUE, category TEXT, bypass\_llm INTEGER, success\_template TEXT, follow\_up\_question TEXT, action\_type TEXT, buttons\_json TEXT, is\_active INTEGER, is\_ephemeral INTEGER, updated\_at TIMESTAMP |
|
||||||
|
| **session\_states** | Активные состояния сессий, черновики и счетчик idle\_turns. | session\_id TEXT PRIMARY KEY, state\_type TEXT, pending\_data TEXT, updated\_at TIMESTAMP |
|
||||||
|
| **chat\_messages** | История диалогов с поддержкой разделения на постоянные и эфемерные сообщения. | id INTEGER PK AUTOINCREMENT, session\_id TEXT, role TEXT, content TEXT, is\_ephemeral INTEGER DEFAULT 0, created\_at TIMESTAMP |
|
||||||
|
| **tasks** | Реестр задач операторов с привязкой к user\_id. | id INTEGER PK AUTOINCREMENT, task\_id TEXT, module TEXT, title TEXT, priority TEXT, status TEXT, due\_date TEXT, user\_id INTEGER, created\_at TIMESTAMP |
|
||||||
|
| **users** | Учетные записи операторов (хэши паролей bcrypt, права администратора). | id INTEGER PK AUTOINCREMENT, username TEXT UNIQUE, password\_hash TEXT, is\_admin INTEGER, full\_name TEXT, created\_at TIMESTAMP |
|
||||||
|
| **department\_synonyms** | База соответствия аббревиатур СКУД полным названиям отделов 1С. | id INTEGER PK AUTOINCREMENT, short\_name TEXT UNIQUE, full\_name TEXT, created\_at TIMESTAMP |
|
||||||
|
| **system\_reference** | Справочник команд и примеров промптов для операторов. | id INTEGER PK AUTOINCREMENT, category TEXT, title TEXT, example\_prompt TEXT, description TEXT |
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**4\. Архитектурная структура каталогов**
|
||||||
|
|
||||||
|
scud\_ai/
|
||||||
|
├── config.py \# Конфигурация: пути, расчет дат, URL Ollama, параметры MS SQL
|
||||||
|
├── exceptions.json \# Декларативные фильтры исключений (ОВК, уборщики, подрядчики)
|
||||||
|
├── main\_etl.py \# Оркестратор ETL: экспорт СКУД/1С \-\> SQLite \-\> Excel/MD отчеты
|
||||||
|
├── project\_code\_snapshot.md \# Полный монолитный исходный код всех файлов репозитория
|
||||||
|
├── core/
|
||||||
|
│ └── database.py \# Драйвер SQLite (WAL), функции сохранения/загрузки срезов
|
||||||
|
├── services/
|
||||||
|
│ ├── scud\_export.py \# Прямой SQL-экспорт из MS SQL «Орион Pro»
|
||||||
|
│ ├── zup\_extractor.py \# Прямой SQL-экспорт из MS SQL 1С:ЗУП 3.1
|
||||||
|
│ ├── data\_loader.py \# Загрузка данных, нормализация ФИО, расчет первой активности
|
||||||
|
│ ├── data\_validator.py \# Проверка свежести и наличия файлов за Вчера и Сегодня
|
||||||
|
│ ├── excel\_exporter.py \# Генерация сводки и детального отчета в Excel (openpyxl)
|
||||||
|
│ ├── text\_reporter.py \# Построение итогового Markdown-отчета через Qwen 2.5
|
||||||
|
│ ├── ai\_verifier.py \# Модули ИИ-аудита, Fuzzy Matching и детекции сбоев СКУД
|
||||||
|
│ ├── feedback\_loop.py \# Интерактивный консольный модуль Human-in-the-Loop
|
||||||
|
│ ├── knowledge\_base.py \# Интерфейс работы с базой знаний правил
|
||||||
|
│ └── share\_copier.py \# Резервное скачивание файлов 1С с шары /mnt/scud\_share
|
||||||
|
├── scripts/
|
||||||
|
│ ├── db\_cli.py \# CLI-утилита управления БД, снапшотами, дампами и сессиями
|
||||||
|
│ ├── fix\_snapshots.py \# Скрипт переиндексации и нормализации снапшотов
|
||||||
|
│ ├── init\_tool\_registry.py \# Инициализация декларативного реестра инструментов
|
||||||
|
│ └── diagnostics/
|
||||||
|
│ ├── inspect\_db.py \# Диагностика структуры таблиц и количества записей
|
||||||
|
│ ├── inspect\_ephemeral.py \# Анализ распределения постоянных и эфемерных сообщений
|
||||||
|
│ ├── inspect\_files.py \# Аудит файлов и размеров на диске
|
||||||
|
│ └── show\_tree.py \# Вывод дерева каталогов
|
||||||
|
├── modules/web\_api/
|
||||||
|
│ ├── main.py \# Точка входа FastAPI приложения, роутеры и раздача статики
|
||||||
|
│ ├── routers/ (admin, auth, chat, tasks) \# REST эндпоинты
|
||||||
|
│ ├── llm/
|
||||||
|
│ │ ├── agent.py \# Координатор диалога, Tool Calls и Topic Drift Guard
|
||||||
|
│ │ ├── schemas.py \# Описание JSON-схем инструментов для Ollama
|
||||||
|
│ │ ├── file\_parser.py \# Мультимодальный парсер файлов (PDF, PNG, XLSX, TXT)
|
||||||
|
│ │ ├── db\_tools.py \# Фасадные вызовы функций БД для агента
|
||||||
|
│ │ ├── core/ (calendar\_utils, fast\_path, ollama\_client, prompt\_merger, tool\_injector)
|
||||||
|
│ │ └── db/ (connection, db\_chat, db\_prompts, db\_snapshots, db\_tasks)
|
||||||
|
│ └── static/ (index.html, css/styles.css, js/app.js, js/auth.js, js/chat.js, js/tasks.js)
|
||||||
|
└── docs/
|
||||||
|
└── development\_roadmap.md \# Стратегическая дорожная карта и архитектурный бэклог
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**5\. Справочник команд CLI (HOWTO)**
|
||||||
|
|
||||||
|
### **ETL Пайплайн (main\_etl.py):**
|
||||||
|
|
||||||
|
> * python main\_etl.py — стандартный дневной запуск (экспорт из MS SQL, сохранение снапшота, генерация отчетов).
|
||||||
|
> * python main\_etl.py \--skip-export — построение отчетов по последнему имеющемуся снапшоту из SQLite без повторного запроса к MS SQL.
|
||||||
|
> * python main\_etl.py \--snapshot 20260805-001 — точный расчет отчетов по указанному составному ID снапшота.
|
||||||
|
> * python main\_etl.py \-d — запуск в режиме расширенной отладки (DEBUG).
|
||||||
|
|
||||||
|
### **CLI базы данных (scripts/db\_cli.py):**
|
||||||
|
|
||||||
|
> * python scripts/db\_cli.py stats — статистика по количеству строк во всех таблицах БД.
|
||||||
|
> * python scripts/db\_cli.py snapshots \[ДД.ММ.ГГГГ\] — просмотр реестра снапшотов с сортировкой по времени создания (свежие сверху).
|
||||||
|
> * python scripts/db\_cli.py scud \[ДД.ММ.ГГГГ\] \[--snapshot ID\] \[--export-xlsx NAME\] — инспекция логов СКУД и экспорт в Excel.
|
||||||
|
> * python scripts/db\_cli.py absences \[ДД.ММ.ГГГГ\] — список кадровых отклонений из 1С:ЗУП.
|
||||||
|
> * python scripts/db\_cli.py anomalies — история выявленных аномалий.
|
||||||
|
> * python scripts/db\_cli.py rules — правила базы знаний компании.
|
||||||
|
> * python scripts/db\_cli.py prompts — системные промпты из system\_prompts.
|
||||||
|
> * python scripts/db\_cli.py tools — декларативный реестр действий инструментов и кнопок.
|
||||||
|
> * python scripts/db\_cli.py sessions — активные сессионные стейты и висящие черновики.
|
||||||
|
> * python scripts/db\_cli.py snapshot del \[ID\] — удаление конкретного снапшота по ID.
|
||||||
|
> * python scripts/db\_cli.py snapshot del \--day \[ДД.ММ.ГГГГ\] — удаление всех снапшотов за день.
|
||||||
|
> * python scripts/db\_cli.py dump \[backup.xlsx\] — полный дамп всех таблиц базы данных в многостраничный Excel.
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**6\. Стратегическая дорожная карта и нерешенные задачи (Roadmap & Backlog)**
|
||||||
|
|
||||||
|
> 1. **Масштабирование механизма эфемерного контекста на все инструменты (Rollout Roadmap):**
|
||||||
|
* **Снапшоты (snapshots):** Перевод вывода объемных таблиц срезов в эфемерный режим (is\_ephemeral \= 1), внедрение интерактивного подтверждения при удалении снапшотов по ID/дате через состояния SNAPSHOTS\_PREVIEW / SNAPSHOTS\_FOLLOWUP.
|
||||||
|
* **База знаний и аномалии (kb & anomalies):** Вывод справочников команд, правил арбитража и списков аномалий исключительно во временный слой для исключения «залипания» модели на устаревших правилах из истории сообщений.
|
||||||
|
* **Декларативная унификация в tool\_action\_registry:** Расширение схемы таблицы полями followup\_state\_type, auto\_purge\_on\_exit, guard\_question\_template для устранения хардкода имен функций в agent.py.
|
||||||
|
> 2. **Модуль анализа внутридневных перемещений («Контроль перекуров»):**
|
||||||
|
* Анализ интервалов между событиями OUT (выход) и IN (возврат) по журналу pLogData.
|
||||||
|
* Подсчет суммарного времени нахождения вне рабочего места в течение смены с отсечением технологической нормы (30–40 минут).
|
||||||
|
* Вывод аналитического блока и злостных нарушителей в отдельный лист Excel-отчета и итоговую Markdown-сводку.
|
||||||
|
> 3. **Интерактивный Task Planner & Roadmap Engine:**
|
||||||
|
* Специализированный инструмент для ведения динамических проектных планов и чек-листов с чекбоксами (\[ \] / \[x\]) прямо в диалоге с синхронизацией в таблицу tasks.
|
||||||
|
> 4. **Изолированная среда исполнения кода (Code Execution Sandbox):**
|
||||||
|
* Безопасный изолированный Docker/gVisor контейнер для динамического выполнения генерируемого моделью Python/Pandas кода на лету с возвратом stdout/stderr и графиков в контекст.
|
||||||
|
> 5. **Автоматизация расписания срезов и Telegram-алерты:**
|
||||||
|
* Автозапуск выгрузок в 12:00, 17:00 и 19:00 через systemd-таймеры с фиксацией snapshot\_time.
|
||||||
|
* Отправка мгновенных алертов о массовых сбоях турникетов (\>5%) и критических кадровых конфликтах дежурному системному администратору в Telegram.
|
||||||
+223
@@ -0,0 +1,223 @@
|
|||||||
|
# **SCUD Orion AI — Полная энциклопедическая хроника, архитектурный паспорт, разбор всех инцидентов и технический контекст проекта**
|
||||||
|
|
||||||
|
**Версия документа:** 4.0 (Исчерпывающий монолитный контекстный паспорт для новой сессии)
|
||||||
|
**Стек технологий и инфраструктура:** Python 3.11+, FastAPI (Uvicorn), SQLite (WAL mode, PRAGMA synchronous=NORMAL, внешние ключи ON), Ollama REST API (Qwen 2.5:14b — основной текстовый агент и классификатор, Qwen 2.5-VL:7b-q8\_0 / Qwen 3-VL:8b — OCR и зрение), MS SQL Server 2019/2022 (СКУД «Орион Pro» ЗАО НВП «Болид», 1С:ЗУП 3.1 через pyodbc / ODBC Driver 18), сетевая Samba-шара (/mnt/scud\_share), Git/Gitea, HTML5 / Vanilla JS / Tailwind CSS.
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**1\. Исходный бизнес-контекст, роли систем и фундаментальные задачи**
|
||||||
|
|
||||||
|
Проект решает комплексную проблему контроллинга трудовой дисциплины, автоматического сопоставления кадровых фактов и защиты от рассинхронизации мастер-данных на предприятии со штатом \~270–276 сотрудников.
|
||||||
|
|
||||||
|
### **Источники первичных данных:**
|
||||||
|
|
||||||
|
> * **СКУД «Орион Pro» (Болид):** База данных MS SQL Server (SERVER: 172.16.31.221\\SQL, DB: Orion-14.01.21-1). Основные таблицы:
|
||||||
|
* pLogData — журнал всех сырых физических транзакций турникетов, калиток и внутренних дверей. Ключевые поля: TimeVal (время сервера), DeviceTime (время контроллера), HozOrgan / Hrk (ID сотрудника), Event (код события: 26, 28, 32 — вход; 27, 29, 33 — выход; 54, 55, 64, 65 — внутренние считыватели), Mode (направление: 1 — IN, 2 — OUT), ReaderIndex, DoorIndex.
|
||||||
|
* pList — реестр физических лиц и пропусков (ФИО, табельный номер, статус архива).
|
||||||
|
* PDivision, PPost — справочники подразделений и должностей СКУД.
|
||||||
|
> * **1С:ЗУП 3.1 (Кадры):** База данных MS SQL Server (SERVER: ACCOUNT-01, DB: ZUP30).
|
||||||
|
* dbo.\_Reference299 — эталонный справочник сотрудников.
|
||||||
|
* dbo.\_InfoRg16921 — периодический регистр сведений состояний сотрудников (отпуска основные, больничные, командировки, декретные отпуска, отпуска без сохранения оплаты).
|
||||||
|
* Файловые выгрузки с сетевой шары /mnt/scud\_share: файлы Штат\_ДД\_ММ\_ГГГГ.xlsx (шапка 7–8 строк) и Отсутствия\_ДД\_ММ\_ГГГГ.xlsx (шапка 3–4 строки).
|
||||||
|
> * **Реестры статических исключений:** static\_reason\_workers.csv (реестр удаленщиков и постоянных разъездных сотрудников), exceptions.json (декларативный фильтр обслуживающего персонала и спец-отделов).
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**2\. Подробнейшая пошаговая хронология разработки, эволюция идей и разбор ошибок**
|
||||||
|
|
||||||
|
### **Фаза 1: Первые эксперименты с LLM, выявление галлюцинаций и гибридная парадигма**
|
||||||
|
|
||||||
|
> * **Попытка передачи сырых списков в чат:** Первоначальная попытка скормить модели Qwen 2.5:14b список из сотен ФИО с инструкцией *«запомни список 1С»* показала, что у LLM нет постоянной памяти между сессиями. Модель теряла контекст, путала фамилии и выдавала ответы с троеточиями ('Васильев Л.И.', ...) или придумывала несуществующие строки (*«ЛЕНМОРНИИПРОЕКТ ОЭС Варламова Мария»*).
|
||||||
|
> * **Ошибки openpyxl vs pandas:** Первая генерация кода моделью использовала библиотеку openpyxl с построчным перебором ячеек, жестко зашитыми опечатками (поиск слова 'фиио') и падением по FileNotFoundError.
|
||||||
|
> * **Переход к детерминированному пайплайну:** Принято ключевое архитектурное решение — **LLM не считает математику и не фильтрует 300 строк текста**. Всю механическую работу (ETL, слияние, фильтрацию) выполняет быстрый детерминированный Python (pandas, SQL), а локальная LLM выступает исключительно в роли «умного ревизора» на этапе сопоставления сомнительного несовпавшего остатка (Diff).
|
||||||
|
|
||||||
|
### **Фаза 2: Нормализация ФИО, кодировки и коллизия совместителей (Кейс Балаевой О.Г.)**
|
||||||
|
|
||||||
|
> * **Символьные ловушки:**
|
||||||
|
* Латинские буквы-двойники (омоглифы a, c, e, o, p, x, y, случайно набранные в английской раскладке).
|
||||||
|
* Служебные суффиксы в скобках, добавляемые 1С ((осн.), (совм.), (внешн.)).
|
||||||
|
* Буквы ё/е, двойные пробелы, неразрывные пробелы \\xa0 и случайный ввод цифры 0 вместо буквы О.
|
||||||
|
> * **Функция normalize\_fio():** Разработан строгий конвейер очистки:
|
||||||
|
> `def normalize_fio(fio):`
|
||||||
|
> `if not fio or not isinstance(fio, str): return ""`
|
||||||
|
> `fio_clean = re.sub(r'\(.*?\)', '', fio) # Удаление скобок`
|
||||||
|
> `fio_clean = fio_clean.replace('\xa0', ' ')`
|
||||||
|
> `trans = str.maketrans('aceopxyABCEHKMOPTX', 'асеорхуАВСЕНКМОРТХ')`
|
||||||
|
> `fio_clean = fio_clean.translate(trans)`
|
||||||
|
> `fio_clean = fio_clean.replace('ё', 'е').replace('Ё', 'Е')`
|
||||||
|
> `return " ".join(fio_clean.strip().split()).title()`
|
||||||
|
> * **Коллизия 270 vs 271:** В СКУД числилось 270 уникальных человек, а в 1С — 271 строка. Ручной разбор показал, что **Балаева Ольга Геннадьевна** занимает две ставки в 1С (*Руководитель управления персоналом* и *Руководитель режимно-секретного отдела*) при одном физическом пропуске в СКУД.
|
||||||
|
> * **Решение:** Внедрена обязательная группировка штата 1С по fio\_clean с объединением должностей и подразделений через " / ".
|
||||||
|
|
||||||
|
### **Фаза 3: Защита Entity Resolution от однофамильцев (Инцидент с Семеновыми)**
|
||||||
|
|
||||||
|
> * **Сбой модели:** При передаче несовпавших ФИО в Qwen 2.5 модель ошибочно сопоставила *«Семенову Юлию Борисовну»* (которая была на работе) с *«Семеновой Диной Дмитриевной»* (находившейся в отпуске), посчитав это опечаткой в имени из\-за одинаковой фамилии. В результате Юлия Борисовна получила статус нарушителя («в отпуске, но пришла»).
|
||||||
|
> * **Двухконтурная защита:**
|
||||||
|
1. В системный промпт ИИ зашито категорическое правило: однофамильцы с разными именами/отчествами — строго разные люди, объединение запрещено.
|
||||||
|
2. В Python встроен дублирующий барьер: если фамилии совпадают, но инициалы/первые буквы имен различаются, связь ИИ принудительно аннулируется с логированием ⚠️ \[Блокировка фейка\].
|
||||||
|
|
||||||
|
### **Фаза 4: Декларативные исключения и синонимы отделов**
|
||||||
|
|
||||||
|
> * **Переход к exceptions.json:** Исключение клининга, дворников, гардеробщиков и хозяйственного персонала отдела «ЭТО» вынесено из кода в конфигурационный файл exceptions.json (поддержка списков departments, positions, fio, position\_keywords).
|
||||||
|
> * **База синонимов отделов (department\_synonyms):** Для сопоставления сокращений СКУД с длинными названиями 1С (например, *«ОВК» ⟷ «Отдел внутреннего контроля»*) создана таблица SQLite, автоматически обучаемая при подтверждениях оператора.
|
||||||
|
|
||||||
|
### **Фаза 5: Аппаратный сбой СКУД 04.08.2026, разгадка «исчезающих неизвестных» и Первая активность**
|
||||||
|
|
||||||
|
> * **Анатомия сбоя:** Утром 04.08.2026 зависли регистраторы входа на главном КПП. У 119 сотрудников не записалось событие входа (time\_in \= 'Нет входа'). Из них 57 человек были официально закрыты документами 1С (отпуска, командировки). Оставшиеся \~74 человека реально пришли на работу, но утренней отметки не имели. Однако вечером сводка выдала всего 4–7 «неизвестных».
|
||||||
|
> * **Вскрытие механизма ошибки:** Было обнаружено, что скрипт агрегации видел дневные перемещения сотрудников (выходы на обед, считыватели туалетов, внутренние двери) и автоматически выставлял Статус \= 'Присутствовал' и is\_present \= 1\. Из-за этого люди попадали в категорию «Итого на работе», а массовый аппаратный сбой скрывался.
|
||||||
|
> * **Внедрение «Первой активности»:**
|
||||||
|
1. В базу данных и выгрузки Excel добавлена колонка **first\_activity («Первая активность»)** — минимальное время любого события дня в здании (MIN(TimeVal) AS FirstRawEvent без учета направления двери).
|
||||||
|
2. Если time\_in \== 'Нет входа', но first\_activity \!= '—', записи присваивается anomaly\_flag \= 'ANOMALY\_NO\_IN\_HAS\_ACTIVITY', а is\_present принудительно сбрасывается в 0\.
|
||||||
|
3. Внедрен модуль analyze\_scud\_mass\_failure\_ai(): при превышении порога в 5% смены (\>10 человек) ИИ поднимает тревогу о массовом сбое контроллеров КПП.
|
||||||
|
> * **Кейс Равина В.Э.:** Сотрудник находился в официальном отпуске, утреннего входа не имел, но пришел после обеда (первая активность 14:11:06, выход 17:18:38). Система корректно зафиксировала истинную кадровую аномалию (физическое присутствие при официальном отпуске).
|
||||||
|
|
||||||
|
### **Фаза 6: Архитектурная миграция под Gitea и система срезов (снапшотов)**
|
||||||
|
|
||||||
|
> * **Безопасный рефакторинг:** Монолит разделен на слои (core/, services/, scripts/, modules/web\_api/) через git mv с сохранением истории коммитов в ветке refactor/modular-structure.
|
||||||
|
> * **Формат ID снапшота:** Введена схема \[Y\]YYYYMMDD-NNN (например, Y20260804-001 для вчерашнего зафиксированного вечернего среза на 22:00 и 20260805-002 для дневного).
|
||||||
|
> * **Оптимизация экспорта:** Если в SQLite уже зафиксирован вчерашний срез с префиксом Y, повторный тяжелый SQL-запрос к СКУД Орион не выполняется (функция has\_yesterday\_final\_snapshot).
|
||||||
|
> * **Инспекция и CLI:** Написан инструмент scripts/db\_cli.py с командами stats, snapshots, scud, absences, anomalies, rules, prompts, tools, sessions, dump, snapshot del.
|
||||||
|
|
||||||
|
### **Фаза 7: Web API, Generative UI, управление памятью LLM и Topic Drift Guard**
|
||||||
|
|
||||||
|
> * **FastAPI & Безопасность:** Развернут Web API с JWT-авторизацией (bcrypt), изолированными задачами пользователей (tasks с привязкой к user\_id), гостевым режимом и ролевой моделью.
|
||||||
|
> * **Проблема деградации контекста LLM:** Вывод объемных списков задач, текстов системных промптов и таблиц снапшотов засорял контекстное окно чата, приводя к галлюцинациям и «залипанию» модели на старых правилах.
|
||||||
|
> * **Решение проблемы памяти (Topic Drift & Ephemeral Context):**
|
||||||
|
* В таблицу chat\_messages добавлен флаг is\_ephemeral \= 1 для временных сообщений.
|
||||||
|
* Введены двухфазные сессии (PROMPT\_PREVIEW → PROMPT\_FOLLOWUP) и детерминированный счетчик отвлечений idle\_turns.
|
||||||
|
* **Context Guard:** На 3-м шаге отвлечения оператора на сторонние темы система отвечает на вопрос и выводит интерактивное напоминание с кнопками \[Завершить настройку\] / \[Показать промпт\]. При idle\_turns \> 3 происходит автоочистка сессии вызовом db\_purge\_ephemeral\_messages().
|
||||||
|
* **Fast-Path:** Прямой перехват команд подтверждения («да», «подтверждаю», «сохранить») и выхода («нет, спасибо», «закончить настройку») без обращения к LLM (ответ за 0.05 сек).
|
||||||
|
* **Декларативный реестр действий (tool\_action\_registry):** Хранение шаблонов ответов, типов кнопок и признаков эфемерности в SQLite.
|
||||||
|
> * **Generative UI (Интерактивные карточки задач):** Инструмент db\_get\_tasks возвращает полезную нагрузку TASK\_INTERACTIVE\_CARD. Фронтенд рендерит живой виджет с табами (Все / В работе / В планах / Завершенные), чекбоксами и inline-созданием задач через REST API без вызова LLM.
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**3\. Полная схема базы данных SQLite (data/scud\_orion\_ai.db)**
|
||||||
|
|
||||||
|
| Таблица | Назначение и бизнес-логика | Полная структура колонок |
|
||||||
|
| :---- | :---- | :---- |
|
||||||
|
| **scud\_logs** | Хранение логов проходов СКУД, расчетного времени и снапшотов. | id (PK), log\_date, fio, fio\_clean, department, position, time\_in, first\_activity, time\_out, time\_in\_building, is\_present, anomaly\_flag, snapshot\_time, snapshot\_id, created\_at |
|
||||||
|
| **zup\_staff** | Штатное расписание 1С:ЗУП на даты срезов. | id (PK), snapshot\_date, fio, fio\_clean, department, position, created\_at |
|
||||||
|
| **zup\_absences** | Официальные кадровые отклонения 1С (отпуска, больничные, командировки, декреты). | id (PK), absence\_date, fio, fio\_clean, absence\_type, created\_at |
|
||||||
|
| **anomalies\_history** | Журнал выявленных кадровых и аппаратных аномалий. | id (PK), anomaly\_date, fio, anomaly\_type, details, human\_status, created\_at |
|
||||||
|
| **ai\_knowledge\_base** | Глобальные правила арбитража и приоритеты компании для ИИ. | id (PK), rule\_text, added\_by, created\_at |
|
||||||
|
| **system\_prompts** | Версионные системные промпты агента (активный промпт по is\_active=1). | id (PK), name, prompt\_text, is\_active, updated\_at |
|
||||||
|
| **tool\_action\_registry** | Декларативный реестр шаблонов, кнопок и настроек эфемерности инструментов. | id (PK), tool\_name, category, bypass\_llm, success\_template, follow\_up\_question, action\_type, buttons\_json, is\_active, is\_ephemeral, updated\_at |
|
||||||
|
| **session\_states** | Активные состояния сессий, черновики и счетчик idle\_turns. | session\_id (PK), state\_type, pending\_data, updated\_at |
|
||||||
|
| **chat\_messages** | История диалогов с поддержкой разделения на постоянные и эфемерные сообщения. | id (PK), session\_id, role, content, is\_ephemeral, created\_at |
|
||||||
|
| **tasks** | Реестр задач пользователей с привязкой к user\_id. | id (PK), task\_id, module, title, priority, status, due\_date, user\_id, created\_at |
|
||||||
|
| **users** | Учетные записи операторов (хэши паролей bcrypt, права администратора). | id (PK), username, password\_hash, is\_admin, full\_name, created\_at |
|
||||||
|
| **department\_synonyms** | База знаний соответствия аббревиатур СКУД полным названиям отделов 1С. | id (PK), short\_name, full\_name, created\_at |
|
||||||
|
| **system\_reference** | Справочник команд и примеров промптов для операторов. | id (PK), category, title, example\_prompt, description |
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**4\. Актуальная архитектура каталогов и назначение файлов**
|
||||||
|
|
||||||
|
`scud_ai/`
|
||||||
|
`├── config.py # Конфигурация: пути, расчет дат, URL Ollama, параметры MS SQL`
|
||||||
|
`├── exceptions.json # Декларативные фильтры исключений (ОВК, уборщики, подрядчики)`
|
||||||
|
`├── main_etl.py # Оркестратор ETL: экспорт СКУД/1С -> SQLite -> Excel/MD отчеты`
|
||||||
|
`├── project_code_snapshot.md # Полный исходный код всех файлов репозитория`
|
||||||
|
`├── core/`
|
||||||
|
`│ └── database.py # Драйвер SQLite (WAL), функции сохранения/загрузки срезов`
|
||||||
|
`├── services/`
|
||||||
|
`│ ├── scud_export.py # Прямой SQL-экспорт из MS SQL «Орион Pro»`
|
||||||
|
`│ ├── zup_extractor.py # Прямой SQL-экспорт из MS SQL 1С:ЗУП 3.1`
|
||||||
|
`│ ├── data_loader.py # Загрузка данных, нормализация ФИО, расчет первой активности`
|
||||||
|
`│ ├── data_validator.py # Проверка свежести и наличия файлов за Вчера и Сегодня`
|
||||||
|
`│ ├── excel_exporter.py # Генерация сводки и детального отчета в Excel (openpyxl)`
|
||||||
|
`│ ├── text_reporter.py # Построение итогового Markdown-отчета через Qwen 2.5`
|
||||||
|
`│ ├── ai_verifier.py # Модули ИИ-аудита, Fuzzy Matching и детекции сбоев СКУД`
|
||||||
|
`│ ├── feedback_loop.py # Интерактивный консольный модуль Human-in-the-Loop`
|
||||||
|
`│ ├── knowledge_base.py # Интерфейс работы с базой знаний правил`
|
||||||
|
`│ └── share_copier.py # Резервное скачивание файлов 1С с шары /mnt/scud_share`
|
||||||
|
`├── scripts/`
|
||||||
|
`│ ├── db_cli.py # CLI-утилита управления БД, снапшотами, дампами и сессиями`
|
||||||
|
`│ ├── fix_snapshots.py # Скрипт переиндексации и нормализации снапшотов`
|
||||||
|
`│ ├── init_tool_registry.py # Инициализация декларативного реестра инструментов`
|
||||||
|
`│ └── diagnostics/`
|
||||||
|
`│ ├── inspect_db.py # Диагностика структуры таблиц и количества записей`
|
||||||
|
`│ ├── inspect_ephemeral.py # Анализ распределения постоянных и эфемерных сообщений`
|
||||||
|
`│ ├── inspect_files.py # Аудит файлов и размеров на диске`
|
||||||
|
`│ └── show_tree.py # Вывод дерева каталогов`
|
||||||
|
`├── modules/web_api/`
|
||||||
|
`│ ├── main.py # Точка входа FastAPI приложения, роутеры и раздача статики`
|
||||||
|
`│ ├── routers/`
|
||||||
|
`│ │ ├── admin.py # Управление пользователями`
|
||||||
|
`│ │ ├── auth.py # JWT аутентификация и смена паролей`
|
||||||
|
`│ │ ├── chat.py # Эндпоинты диалогов (авторизованный и гостевой)`
|
||||||
|
`│ │ └── tasks.py # REST API управления задачами`
|
||||||
|
`│ ├── llm/`
|
||||||
|
`│ │ ├── agent.py # Координатор диалога, Tool Calls и Topic Drift Guard`
|
||||||
|
`│ │ ├── schemas.py # Описание JSON-схем инструментов для Ollama`
|
||||||
|
`│ │ ├── file_parser.py # Мультимодальный парсер файлов (PDF, PNG, XLSX, TXT)`
|
||||||
|
`│ │ ├── db_tools.py # Фасадные вызовы функций БД для агента`
|
||||||
|
`│ │ ├── core/`
|
||||||
|
`│ │ │ ├── calendar_utils.py # Формирование системного контекста дат и дней недели`
|
||||||
|
`│ │ │ ├── fast_path.py # Быстрый перехват команд подтверждения/отмены (0.05 с)`
|
||||||
|
`│ │ │ ├── ollama_client.py # HTTP-клиент взаимодействия с Ollama API`
|
||||||
|
`│ │ │ ├── prompt_merger.py # Парсинг и слияние правок системного промпта`
|
||||||
|
`│ │ │ └── tool_injector.py # Очистка сырых тегов и артефактов Tool Calls`
|
||||||
|
`│ │ └── db/`
|
||||||
|
`│ │ ├── connection.py # Подключение к scud_orion_ai.db`
|
||||||
|
`│ │ ├── db_chat.py # Сохранение истории и очистка эфемерных сообщений`
|
||||||
|
`│ │ ├── db_prompts.py # Системные промпты, стейты и реестр действий`
|
||||||
|
`│ │ ├── db_snapshots.py # Выборка и удаление снапшотов`
|
||||||
|
`│ │ └── db_tasks.py # CRUD операции с задачами пользователей`
|
||||||
|
`│ └── static/`
|
||||||
|
`│ ├── index.html # SPA разметка (модалки, чат, боковая панель задач)`
|
||||||
|
`│ ├── css/styles.css # Стили и мобильные оптимизации`
|
||||||
|
`│ └── js/`
|
||||||
|
`│ ├── app.js # Инициализация приложения`
|
||||||
|
`│ ├── auth.js # Логика входа, смены паролей и администрирования`
|
||||||
|
`│ ├── chat.js # Обработка сообщений, Drag&Drop, виджет задач и кнопки`
|
||||||
|
`│ └── tasks.js # Управление боковой панелью задач (Drawer)`
|
||||||
|
`└── docs/`
|
||||||
|
`└── development_roadmap.md # Стратегическая дорожная карта и архитектурный бэклог`
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**5\. Справочник команд командной строки (CLI Reference)**
|
||||||
|
|
||||||
|
### **Главный конвейер контроллинга (main\_etl.py)**
|
||||||
|
|
||||||
|
> * python main\_etl.py — стандартный дневной запуск: экспорт свежих данных СКУД и 1С, сохранение снапшота в SQLite, генерация Excel и Markdown отчетов.
|
||||||
|
> * python main\_etl.py \--skip-export — построение отчетов по последнему имеющемуся снапшоту из SQLite без повторного запроса к MS SQL.
|
||||||
|
> * python main\_etl.py \--snapshot 20260805-001 — точный расчет отчетов по указанному составному ID снапшота.
|
||||||
|
> * python main\_etl.py \-d — запуск в режиме расширенной отладки (DEBUG).
|
||||||
|
|
||||||
|
### **CLI-утилита управления SQLite базой данных (scripts/db\_cli.py)**
|
||||||
|
|
||||||
|
> * python scripts/db\_cli.py stats — общая статистика количества записей по всем таблицам БД.
|
||||||
|
> * python scripts/db\_cli.py snapshots \[ДД.ММ.ГГГГ\] — просмотр реестра снапшотов с сортировкой по времени создания (самые свежие сверху).
|
||||||
|
> * python scripts/db\_cli.py scud \[ДД.ММ.ГГГГ\] \[--snapshot ID\] \[--export-xlsx NAME\] — просмотр логов СКУД и опциональный экспорт в Excel.
|
||||||
|
> * python scripts/db\_cli.py absences \[ДД.ММ.ГГГГ\] — список кадровых отклонений из 1С:ЗУП.
|
||||||
|
> * python scripts/db\_cli.py anomalies — история выявленных аномалий.
|
||||||
|
> * python scripts/db\_cli.py rules — правила базы знаний компании.
|
||||||
|
> * python scripts/db\_cli.py prompts — просмотр системных промптов из таблицы system\_prompts.
|
||||||
|
> * python scripts/db\_cli.py tools — вывод декларативного реестра действий инструментов и шаблонов кнопок.
|
||||||
|
> * python scripts/db\_cli.py sessions — просмотр активных сессионных стейтов и висящих черновиков.
|
||||||
|
> * python scripts/db\_cli.py snapshot del \[ID\] — удаление конкретного снапшота по ID.
|
||||||
|
> * python scripts/db\_cli.py snapshot del \--day \[ДД.ММ.ГГГГ\] — удаление всех снапшотов за выбранный день.
|
||||||
|
> * python scripts/db\_cli.py dump \[backup.xlsx\] — полный дамп всех таблиц базы данных в многостраничный Excel-файл.
|
||||||
|
|
||||||
|
## ---
|
||||||
|
|
||||||
|
**6\. Стратегическая дорожная карта развития (Roadmap & Backlog)**
|
||||||
|
|
||||||
|
> 1. **Масштабирование механизма эфемерного контекста на все инструменты (Rollout Roadmap):**
|
||||||
|
* **Снапшоты (snapshots):** Перевод вывода объемных таблиц срезов в эфемерный режим (is\_ephemeral \= 1), внедрение интерактивного подтверждения при удалении снапшотов по ID/дате через состояния SNAPSHOTS\_PREVIEW / SNAPSHOTS\_FOLLOWUP.
|
||||||
|
* **База знаний и аномалии (kb & anomalies):** Вывод справочников команд, правил арбитража и списков аномалий исключительно во временный слой для исключения «залипания» модели на устаревших правилах из истории сообщений.
|
||||||
|
* **Декларативная унификация в tool\_action\_registry:** Расширение схемы таблицы полями followup\_state\_type, auto\_purge\_on\_exit, guard\_question\_template для устранения хардкода имен функций в agent.py.
|
||||||
|
> 2. **Модуль анализа внутридневных перемещений («Контроль перекуров»):**
|
||||||
|
* Анализ интервалов между событиями OUT (выход) и IN (возврат) по журналу pLogData.
|
||||||
|
* Подсчет суммарного времени нахождения вне рабочего места в течение смены с отсечением технологической нормы (30–40 минут).
|
||||||
|
* Вывод аналитического блока и злостных нарушителей в отдельный лист Excel-отчета и итоговую Markdown-сводку.
|
||||||
|
> 3. **Интерактивный Task Planner & Roadmap Engine:**
|
||||||
|
* Специализированный инструмент для ведения динамических проектных планов и чек-листов с чекбоксами (\[ \] / \[x\]) прямо в диалоге с синхронизацией в таблицу tasks.
|
||||||
|
> 4. **Изолированная среда исполнения кода (Code Execution Sandbox):**
|
||||||
|
* Безопасный изолированный Docker/gVisor контейнер для динамического выполнения генерируемого моделью Python/Pandas кода на лету с возвратом stdout/stderr и графиков в контекст.
|
||||||
|
> 5. **Автоматизация расписания срезов и Telegram-алерты:**
|
||||||
|
* Автозапуск выгрузок в 12:00, 17:00 и 19:00 через systemd-таймеры с фиксацией snapshot\_time.
|
||||||
|
* Отправка мгновенных алертов о массовых сбоях турникетов (\>5%) и критических кадровых конфликтах дежурному системному администратору в Telegram.
|
||||||
@@ -1,41 +0,0 @@
|
|||||||
# 🗺️ ДОРОЖНАЯ КАРТА И АРХИТЕКТУРНЫЙ РОАДМАП ПРОЕКТА SCUD ORION AI
|
|
||||||
|
|
||||||
## 📌 Зафиксированные вехи и концептуальные архитектурные решения
|
|
||||||
|
|
||||||
Схема веток Gitea для ближайшего развития проекта:
|
|
||||||
|
|
||||||
main (100% стабильный релиз)
|
|
||||||
│
|
|
||||||
├──► feature/tool-registry-db ──────┐ (Создание таблицы tool_action_registry в SQLite)
|
|
||||||
│ ▼
|
|
||||||
├───────────────────────────────► merge to main
|
|
||||||
│
|
|
||||||
├──► feature/interactive-buttons ───┐ (Подключение Да/Нет чипсов на фронтенде)
|
|
||||||
│ ▼
|
|
||||||
├───────────────────────────────► merge to main
|
|
||||||
│
|
|
||||||
└──► feature/ephemeral-context ─────┐ (Внедрение флагов is_ephemeral для очистки памяти LLM)
|
|
||||||
▼
|
|
||||||
merge to main
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
### 1. Архитектура скользящего контекста и отслеживания смены темы (Intent / Topic Drift Tracking)
|
|
||||||
* **Проблема:** При работе с длинными операциями (редактирование промптов, мастера табелей, снапшотов) оператор может отвлекаться на сторонние вопросы или уточнения. Жесткий сброс сессии уничтожает черновики, а вечное хранение засоряет память LLM.
|
|
||||||
* **Решение (Сферическое / Фундаментальное):**
|
|
||||||
1. **Семантическая оценка моделью:** Модель через системный контекст определяет, относится ли вопрос оператора к активному черновику/инструменту или разговор ушел в сторону (`topic_shift = true`).
|
|
||||||
2. **Детерминированный счетчик в бэкенде:** Бэкенд фиксирует шаги отвлечения (`idle_turns`). Пока $N < 3$, оператор может свободно общаться, не теряя висящий контекст.
|
|
||||||
3. **Вежливый перехват и выбор (Guard):** На 3-м шаге отвлечения система отвечает на текущий вопрос оператора и мягко напоминает о незавершенной транзакции кнопками `[Применить / Сохранить]` или `[Отменить и сбросить]`.
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## 🚀 Будущие модули и запланированный функционал (Backlog)
|
|
||||||
|
|
||||||
### 2. Интерактивный модуль планирования и дорожных карт (Checklist & Task Planner Engine)
|
|
||||||
* **Концепция:** Специализированный инструмент для LLM-агента, позволяющий вести динамические проектные чек-листы и планы с интерактивными чекбоксами прямо в диалоге и базе данных.
|
|
||||||
* **Ключевые возможности:**
|
|
||||||
* Составление многоуровневых планов с чекбоксами (`[ ]` / `[x]`).
|
|
||||||
* Фиксация промежуточных комментариев и статусов выполнения по каждому пункту.
|
|
||||||
* Синхронизация задач плана с SQLite-таблицей `tasks`.
|
|
||||||
* Экспорт планов в Markdown/Excel и отображение на дашборде.
|
|
||||||
@@ -0,0 +1,147 @@
|
|||||||
|
# План реализации (ROADMAP)
|
||||||
|
|
||||||
|
## 1. Очистка от регулярок и костылей (`tool_injector.py`) `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] Полностью удалить принудительные перехваты текста регулярными выражениями для команд добавления, редактирования и удаления пунктов[cite: 3].
|
||||||
|
- [x] Оставить в модуле только базовую санитарную очистку сырых тегов (`<tool_call>`)[cite: 3].
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 2. Настройка контекста и инструкций сессии (`agent.py`) `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] Передать управление диалогом языковой модели через системный блок `role: "system"`[cite: 3].
|
||||||
|
- [x] При активном состоянии `PROMPT_PREVIEW` передавать модели инструкцию:
|
||||||
|
- **Подтверждение / отмена / корректировка:** продолжать работу с превью и вызывать соответствующие инструменты[cite: 3].
|
||||||
|
- **Смена темы:** вежливо напомнить об открытом изменении и запросить решение[cite: 3].
|
||||||
|
- [x] Обеспечить видимость эфемерных сообщений (`is_ephemeral = 1`) для модели во время активной работы с превью[cite: 3].
|
||||||
|
- [x] Внедрить семантический Topic Drift Guard (`idle_turns` = 3) с вопросами и кнопками из `tool_action_registry`[cite: 3].
|
||||||
|
- [x] Реализовать детерминированный мгновенный сброс сессии и зачистку эфемерного контекста при вызове сторонних инструментов[cite: 3].
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 3. Очистка эфемерных сообщений при завершении (`fast_path.py` / `context_manager.py`) `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] Настроить удаление временных сообщений превью (`db_purge_ephemeral_messages`) строго в момент нажатия кнопок **«Подтвердить»** или **«Отменить»**[cite: 3].
|
||||||
|
- [x] Сбрасывать состояние сессии в базе данных после фиксации решения[cite: 3].
|
||||||
|
- [x] Внедрить прямое точечное применение изменений через `db_apply_prompt_node_action`[cite: 3].
|
||||||
|
- [x] Добавить обработку фазы `PROMPT_FOLLOWUP` с кнопками завершения и очистки контекста[cite: 3].
|
||||||
|
- [x] Создать `context_manager.py` для строгого разграничения служебных tool-пар (`is_ephemeral=1`) и содержательного диалога (`is_ephemeral=0`), сохраняя беседы Topic Drift в истории[cite: 3].
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 4. Инлайн-редактор в окне диалога (`core.js`) `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] Проверить работу блока ручного редактирования (`inline-prompt-editor-container`) с кнопками **«Сохранить правки»** и **«Свернуть»**[cite: 3].
|
||||||
|
- [x] Обеспечить сохранение черновика через API (`/api/v1/chat/draft`) и отображение обновленного текста перед подтверждением[cite: 3].
|
||||||
|
- [x] Реализовать двусторонний клиентский Diff-рендерер (одновременная подсветка добавленных строк и зачеркивание удаленных `[УДАЛЕНИЕ]`)[cite: 3].
|
||||||
|
- [x] Добавить авто-форматирование и отступы подпунктов (`X.Y.`) при ручном сохранении черновика[cite: 3].
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 5. Тестирование и валидация системного промпта `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] **Нативные вызовы:** проверить добавление, редактирование и удаление пунктов через нативные вызовы модели (`db_prompt_node_edit`)[cite: 3].
|
||||||
|
- [x] **Контекстные сценарии:** проверить поведение модели при смене темы диалога оператором (Guardrail)[cite: 3].
|
||||||
|
- [x] **UI и очистка:** проверить ручное редактирование через кнопку в окне чата и последующую очистку контекста (`db_purge_ephemeral_messages`)[cite: 3].
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 6. Распространение архитектурного паттерна на модуль задач (`tasks`) `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] **Масштабирование UI задач:**
|
||||||
|
- Увеличена ширина карточки до `max-w-4xl` и динамическая высота скролла до `70vh`[cite: 3].
|
||||||
|
- Реализованы переключатели фильтрации: «В работе» (по умолчанию), «В планах», «Готово», «Все»[cite: 3].
|
||||||
|
- Разделены кнопки действия: «В работу» (для плановых) и «Готово» (для текущих)[cite: 3].
|
||||||
|
- [x] **Инлайн-редактирование карточки задачи:**
|
||||||
|
- Плавная трансформация карточки в 3-строчную форму (текст, дата со встроенным пикером, выпадающий список важности)[cite: 3].
|
||||||
|
- Отображение даты создания задачи[cite: 3].
|
||||||
|
- Сохранение через REST API эндпоинт `PATCH /api/v1/tasks/{id}`[cite: 3].
|
||||||
|
- [x] **Детерминированный Fast-Path и двухфазное удаление:**
|
||||||
|
- Мгновенная смена статусов без задержек LLM[cite: 3].
|
||||||
|
- Карточка подтверждения удаления с автоочисткой контекста[cite: 3].
|
||||||
|
- [x] **Генерация отчетов задач в Markdown:**
|
||||||
|
- Добавлен инструмент `db_export_tasks_markdown` для формирования Markdown-файла с группировкой по модулям и чекбоксами `[x]` / `[ ]`[cite: 3].
|
||||||
|
- Реализован роутер безопасной отдачи файлов с сохранением имени (`/api/v1/files/download/...`)[cite: 3].
|
||||||
|
- [x] **Доменная консолидация задач:**
|
||||||
|
- Реализован консолидированный инструмент `db_tasks_edit(action: ["ADD", "UPDATE", "DELETE", "EXPORT"], ...)`[cite: 3].
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 7. Распространение на модуль снапшотов (`snapshots`) `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] **Поддержка Y-снапшотов:** универсальный парсинг и поиск по `snapshot_id LIKE 'Y%'` и срезам за 22:00:00[cite: 3].
|
||||||
|
- [x] **Single Active Snapshot View:** сохранение активного среза в контексте для аналитики с автоматической ротацией и зачисткой при запросе новых дат[cite: 3].
|
||||||
|
- [x] **Интерактивный UI с чекбоксами и защитой срезов:**
|
||||||
|
- Чекбоксы в строках дневных срезов и кнопка «Выбрать все» в шапке[cite: 3].
|
||||||
|
- Защита итогового вечернего среза Y (иконка замочка `🔒`, блокировка выбора)[cite: 3].
|
||||||
|
- Динамическая кнопка «Удалить выбранные (N)» в подвале карточки[cite: 3].
|
||||||
|
- [x] **Детерминированный Fast-Path удаления срезов:**
|
||||||
|
- Двухфазное подтверждение удаления (одиночное и пакетное) с корректным счётчиком количества удаляемых элементов[cite: 3].
|
||||||
|
- Защита от сброса фильтра даты (`query_date`) при обновлении карточки после удаления[cite: 3].
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 8. Глубокий архитектурный рефакторинг ядра и сервисов `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] **Выделение общего слоя ядра (`core/`):**
|
||||||
|
- Изолированный пул соединений SQLite (`core/connection.py`) с поддержкой WAL и таймаутов.
|
||||||
|
- DDL-схемы таблиц и индексов (`core/schema.py`).
|
||||||
|
- Репозитории СКУД и 1С (`core/repositories/scud_repo.py`, `core/repositories/zup_repo.py`).
|
||||||
|
- Фасад обратной совместимости (`core/database.py`).
|
||||||
|
- [x] **Декомпозиция предметных доменов (`services/`):**
|
||||||
|
- Сервис задач (`services/tasks/`), системного промпта (`services/prompts/`), срезов СКУД (`services/snapshots/`), базы знаний (`services/knowledge/`).
|
||||||
|
- [x] **Рефакторинг ETL-конвейера (`services/scud_etl/`):**
|
||||||
|
- Оркестратор `pipeline.py`, детектор аномалий `anomaly_detector.py`, слияние `merger.py`.
|
||||||
|
- [x] **Модульный ИИ-оркестратор (`modules/ai_engine/`):**
|
||||||
|
- Динамический строитель системного контекста (`context_builder.py`).
|
||||||
|
- Изолированные обработчики инструментов в `handlers/`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 9. Реляционный реестр исключений, схлопывание мульти-пропусков и автозакрытие 8.5ч `[ЗАВЕРШЕНО]`
|
||||||
|
- [x] **Реестр исключений в SQLite (`exceptions_registry`):**
|
||||||
|
- Хранение категорий `departments`, `positions`, `fio`, `position_keywords`, `include_fio` в БД.
|
||||||
|
- Доменный сервис `exceptions_repo.py` и полная поддержка в CLI `scripts/db_cli.py exceptions`.
|
||||||
|
- [x] **Агрегация мульти-пропусков физлиц в СКУД:**
|
||||||
|
- Функция `aggregate_scud_by_person` в `merger.py`: объединение событий всех пропусков одного человека (ранний вход, поздний выход, присутствие).
|
||||||
|
- Автоматическая фиксация аномалий дублирования пропусков (`DUPLICATE_SCUD_CARD`).
|
||||||
|
- [x] **Умный выбор ставки совместителей 1С:ЗУП:**
|
||||||
|
- Функция `select_best_zup_position` в `merger.py`: привязка ставки 1С по подразделению физического нахождения в СКУД.
|
||||||
|
- [x] **Автозакрытие смен по Правилу 8.5ч:**
|
||||||
|
- Для забывших отметиться на выходе офисных сотрудников: автоматический расчет `Вход + 8ч 30мин`, норма 8 часов и отклонение `0:00` в `excel_exporter.py`.
|
||||||
|
- [x] **Кэш сопоставлений личностей (`person_identity_mapping`):**
|
||||||
|
- Таблица в SQLite и команды `db_cli.py mapping [list|add|del]`.
|
||||||
|
- Теневой режим ИИ-подсказок по нечетким ФИО в `text_reporter.py`.
|
||||||
|
- [x] **Каскадный fallback кадровых отсутствий:**
|
||||||
|
- `MS SQL ЗУП` $\rightarrow$ резервный парсинг `Отсутствия_*.xlsx` $\rightarrow$ `static_reason_workers.csv`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 10. Разграничение Сводки и Отчета + Почасовые срезы и Y-23:59:59 `[В РАБОТЕ]`
|
||||||
|
- [ ] **Корректировка финишного среза Y на 23:59:59:**
|
||||||
|
- [ ] Обновить время вечернего среза с `22:00:00` на `23:59:59` в `services/scud_export.py`, `scud_repo.py` и `pipeline.py`.
|
||||||
|
- [ ] **Разделение генераторов отчетов:**
|
||||||
|
- [ ] Создать `services/scud_etl/svodka_generator.py` (оперативная сводка, последний доступный срез или выбранный дневной).
|
||||||
|
- [ ] Создать `services/scud_etl/otchet_generator.py` (детальный отчет за прошлую смену, строго по единственному итоговому `Y`-снапшоту за 23:59:59).
|
||||||
|
- [ ] Исключить устаревший `report_generator.py`.
|
||||||
|
- [ ] **Параметризация `main_etl.py`:**
|
||||||
|
- [ ] Добавить поддержку аргументов `--date ДД.ММ.ГГГГ` и `--snapshot ID` для свободного запуска расчета за любой прошлый день.
|
||||||
|
- [ ] **Политика почасовых срезов (Hourly Snapshots) и retention-очистка:**
|
||||||
|
- [ ] Реализовать фоновое снятие часовых срезов в течение рабочего дня.
|
||||||
|
- [ ] Настроить ночную очистку промежуточных дневных срезов с сохранением полуденного (13:00) и финишного (`Y` 23:59:59).
|
||||||
|
- [ ] **Толерантный поиск срезов (Smart Snap-to-Grid, ±15–30 мин):**
|
||||||
|
- [ ] Поиск готового снапшота в SQLite при запросе оператора с возможностью мгновенного построения сводки без лишних запросов к MS SQL Орион.
|
||||||
|
- [ ] Формирование On-Demand запроса в MS SQL Орион при отсутствии подходящего среза.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 11. Интеллектуальный кадровый арбитраж ДО генерации отчетов `[В ПЛАНАХ]`
|
||||||
|
- [ ] **Двухконтурный запуск ИИ:**
|
||||||
|
- [ ] Перенос арбитража не сопоставившихся персон на этап [2.5] ДО сохранения Excel-файлов `сводка.xlsx` и `отчет.xlsx`.
|
||||||
|
- [ ] **Якорный табельный номер (TabNo Matching):**
|
||||||
|
- [ ] Извлечение `TabNo` из MS SQL Орион (`pList.TabNo`) и MS SQL 1С:ЗУП.
|
||||||
|
- [ ] Добавление колонок `scud_tab_no` и `zup_tab_no` в SQLite (`scud_logs`, `zup_staff`, `person_identity_mapping`).
|
||||||
|
- [ ] Защита от смены фамилий и опечаток через неизменяемый табельный номер.
|
||||||
|
- [ ] **Динамическая инвалидация кэша связок:**
|
||||||
|
- [ ] Автоматическая отбраковка записей кэша при увольнении сотрудника или несовпадении подразделения.
|
||||||
|
---
|
||||||
|
|
||||||
|
## 12. Изолированная песочница кода (Code Execution Sandbox Engine) `[В ПЛАНАХ]`
|
||||||
|
- [ ] **Docker/gVisor контур:**
|
||||||
|
- Создание изолированного контейнера без доступа к внешней сети (network: none) с ограниченными лимитами по памяти и CPU (cgroups)[cite: 3].
|
||||||
|
- Настройка безопасного монтирования только необходимых CSV/Parquet-файлов данных в режиме Read-Only[cite: 3].
|
||||||
|
- [ ] **Динамические Python/Pandas вычисления:**
|
||||||
|
- Инструмент генерации и безопасного выполнения скриптов агрегации и аналитики данных СКУД / 1С на лету[cite: 3].
|
||||||
|
- Перехват stdout/stderr, сбор результатов расчетов и графиков с передачей в UI-чата[cite: 3].
|
||||||
File diff suppressed because it is too large
Load Diff
+7
-4
@@ -7,10 +7,9 @@
|
|||||||
"Уборщик служебных помещений"
|
"Уборщик служебных помещений"
|
||||||
],
|
],
|
||||||
"fio": [
|
"fio": [
|
||||||
"Таткало Валерий Валерьевич",
|
"Таткало Валерий Валерьевич",
|
||||||
"Петренюк Андрей Германович",
|
"Петренюк Андрей Германович",
|
||||||
"Чуркина Елена Геннадьевна",
|
"Михалев Сергей Геннадьевич"
|
||||||
"Михалев Сергей Геннадьевич"
|
|
||||||
],
|
],
|
||||||
"position_keywords": [
|
"position_keywords": [
|
||||||
"уборщик",
|
"уборщик",
|
||||||
@@ -18,5 +17,9 @@
|
|||||||
"дворник",
|
"дворник",
|
||||||
"гардероб",
|
"гардероб",
|
||||||
"рабочий по обслуживанию"
|
"рабочий по обслуживанию"
|
||||||
|
],
|
||||||
|
"include_fio": [
|
||||||
|
"Тарасенко Александр Александрович",
|
||||||
|
"Журиков Михаил Николаевич"
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
+102
-533
@@ -1,571 +1,140 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: main_etl.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
ROLE: Главная точка входа ETL-конвейера СКУД ⟷ 1С:ЗУП.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import json
|
|
||||||
import argparse
|
import argparse
|
||||||
|
import logging
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
from datetime import datetime, timedelta
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
from services.scud_etl.pipeline import load_best_snapshot_for_date, load_1c_files_for_date
|
||||||
|
from services.scud_etl.merger import merge_scud_and_1c, calculate_summary_metrics
|
||||||
|
from services.scud_etl.anomaly_detector import detect_registry_anomalies
|
||||||
|
from services.scud_etl.svodka_generator import generate_svodka_service
|
||||||
|
from services.scud_etl.otchet_generator import generate_otchet_service
|
||||||
|
from services.text_reporter import generate_markdown_report
|
||||||
|
|
||||||
from config import DATE_TODAY, DATE_YESTERDAY, OUTPUT_DIR, DATA_DIR, normalize_fio
|
|
||||||
from services.scud_export import run_export
|
from services.scud_export import run_export
|
||||||
from services.share_copier import copy_1c_files_from_share
|
from services.share_copier import copy_1c_files_from_share
|
||||||
from services.data_validator import check_file_freshness
|
from services.excel_exporter import export_raw_scud
|
||||||
from services.data_loader import load_1c_data_smart, load_scud_data
|
|
||||||
from services.excel_exporter import generate_summary_excel, generate_detailed_excel
|
|
||||||
from services.ai_verifier import ai_verify_scud_against_staff, analyze_scud_mass_failure_ai
|
|
||||||
from services.text_reporter import generate_markdown_report
|
|
||||||
from services.feedback_loop import review_ai_decisions
|
|
||||||
from services.knowledge_base import load_knowledge_base
|
|
||||||
from core.database import (
|
|
||||||
init_db,
|
|
||||||
save_scud_to_db,
|
|
||||||
save_staff_to_db,
|
|
||||||
save_absences_to_db,
|
|
||||||
save_anomalies_to_db,
|
|
||||||
load_scud_from_db_by_snapshot,
|
|
||||||
load_staff_from_db,
|
|
||||||
load_absences_from_db,
|
|
||||||
get_latest_snapshot_time,
|
|
||||||
has_scud_logs_for_date
|
|
||||||
)
|
|
||||||
|
|
||||||
|
logging.basicConfig(level=logging.INFO, format="[%(asctime)s] [%(levelname)s] %(message)s")
|
||||||
def load_exceptions_config():
|
|
||||||
"""Загружает файл exceptions.json из корня проекта."""
|
|
||||||
root_dir = os.path.dirname(os.path.abspath(__file__))
|
|
||||||
json_path = os.path.join(root_dir, "exceptions.json")
|
|
||||||
if not os.path.exists(json_path):
|
|
||||||
return {}
|
|
||||||
try:
|
|
||||||
with open(json_path, 'r', encoding='utf-8') as f:
|
|
||||||
return json.load(f)
|
|
||||||
except Exception as e:
|
|
||||||
print(f"[⚠️] Ошибка чтения exceptions.json: {e}")
|
|
||||||
return {}
|
|
||||||
|
|
||||||
|
|
||||||
def apply_exceptions_from_json(df, exceptions_cfg):
|
|
||||||
"""
|
|
||||||
Быстрая и эффективная разметка флага is_excluded=True на основе exceptions.json.
|
|
||||||
Выполняется мгновенно без цикличных HTTP-запросов к ИИ.
|
|
||||||
"""
|
|
||||||
if df is None or df.empty or not exceptions_cfg:
|
|
||||||
if df is not None:
|
|
||||||
df['is_excluded'] = False
|
|
||||||
return df
|
|
||||||
|
|
||||||
deps = [d.strip().lower() for d in exceptions_cfg.get("departments", []) if d]
|
|
||||||
exact_pos = [p.strip().lower() for p in exceptions_cfg.get("positions", []) if p]
|
|
||||||
pos_kw = [k.strip().lower() for k in exceptions_cfg.get("position_keywords", []) if k]
|
|
||||||
exc_fios = [normalize_fio(f) for f in exceptions_cfg.get("fio", []) if f]
|
|
||||||
|
|
||||||
df['is_excluded'] = False
|
|
||||||
|
|
||||||
for idx, row in df.iterrows():
|
|
||||||
fio_clean = row.get('fio_clean', '')
|
|
||||||
dep_1c = str(row.get('Подразделение', '')).lower()
|
|
||||||
dep_scud = str(row.get('department_scud', row.get('department', ''))).lower()
|
|
||||||
pos = str(row.get('Должность', '')).lower()
|
|
||||||
|
|
||||||
is_fio_exc = fio_clean in exc_fios
|
|
||||||
is_pos_exc = (pos in exact_pos) or any(k in pos for k in pos_kw if k) if pos else False
|
|
||||||
|
|
||||||
# Быстрая проверка отделов по подстрокам
|
|
||||||
is_dep_exc = False
|
|
||||||
if deps:
|
|
||||||
is_dep_exc = any(d in dep_1c or d in dep_scud for d in deps)
|
|
||||||
|
|
||||||
if is_fio_exc or is_dep_exc or is_pos_exc:
|
|
||||||
df.at[idx, 'is_excluded'] = True
|
|
||||||
|
|
||||||
return df
|
|
||||||
|
|
||||||
|
|
||||||
def filter_report_dataframe(merged_df):
|
|
||||||
"""
|
|
||||||
Исключает сотрудников из списка исключений и сотрудников без пропуска из детального отчета,
|
|
||||||
ЕСЛИ у них нет официального документа отсутствия из 1С:ЗУП.
|
|
||||||
"""
|
|
||||||
if merged_df is None or merged_df.empty:
|
|
||||||
return merged_df
|
|
||||||
|
|
||||||
has_1c_reason = (
|
|
||||||
merged_df['Вид_отсутствия'].notna() &
|
|
||||||
(merged_df['Вид_отсутствия'].astype(str).str.strip() != '') &
|
|
||||||
(~merged_df['Вид_отсутствия'].astype(str).str.startswith('Исключение'))
|
|
||||||
)
|
|
||||||
is_not_excluded = merged_df.get('is_excluded', False) == False
|
|
||||||
is_not_no_pass = merged_df.get('no_scud_pass', False) == False
|
|
||||||
|
|
||||||
filtered_df = merged_df[(is_not_excluded & is_not_no_pass) | has_1c_reason].copy()
|
|
||||||
return filtered_df
|
|
||||||
|
|
||||||
|
|
||||||
def load_static_reason_workers():
|
|
||||||
"""Загружает реестр удалёнщиков и статических причин из CSV."""
|
|
||||||
static_path = os.path.join(DATA_DIR, "static_reason_workers.csv")
|
|
||||||
if not os.path.exists(static_path):
|
|
||||||
return {}
|
|
||||||
try:
|
|
||||||
df_static = pd.read_csv(static_path, encoding='utf-8')
|
|
||||||
if 'fio' in df_static.columns and 'reason' in df_static.columns:
|
|
||||||
df_static['fio_clean'] = df_static['fio'].apply(normalize_fio)
|
|
||||||
return dict(zip(df_static['fio_clean'], df_static['reason']))
|
|
||||||
except Exception as e:
|
|
||||||
print(f"[⚠️] Ошибка чтения static_reason_workers.csv: {e}")
|
|
||||||
return {}
|
|
||||||
|
|
||||||
|
|
||||||
def detect_all_anomalies(merged_df, static_reasons_dict, kb_rules, scud_fios_set=None):
|
|
||||||
"""
|
|
||||||
Автоматически выявляет истинные аномалии СКУД ⟷ 1С.
|
|
||||||
"""
|
|
||||||
anomalies = []
|
|
||||||
kb_rules_text = " ".join(kb_rules).lower() if kb_rules else ""
|
|
||||||
|
|
||||||
ALLOWED_WORK_TRIP_KEYWORDS = ['командировк', 'разъездн', 'поездк']
|
|
||||||
|
|
||||||
for idx, row in merged_df.iterrows():
|
|
||||||
fio = row.get('Сотрудник', row.get('fio_clean', ''))
|
|
||||||
fio_clean = row.get('fio_clean', '')
|
|
||||||
is_present = row.get('Пришел', False)
|
|
||||||
is_exc = row.get('is_excluded', False)
|
|
||||||
reason_1c = str(row.get('Вид_отсутствия', '')).strip()
|
|
||||||
has_1c_reason = pd.notna(row.get('Вид_отсутствия')) and reason_1c != '' and not reason_1c.startswith('Исключение')
|
|
||||||
anom_flag = row.get('anomaly_flag', 'NONE')
|
|
||||||
|
|
||||||
is_fio_whitelisted_in_kb = fio_clean.lower() in kb_rules_text
|
|
||||||
|
|
||||||
if is_present and has_1c_reason:
|
|
||||||
reason_lower = reason_1c.lower()
|
|
||||||
is_allowed_trip = any(kw in reason_lower for kw in ALLOWED_WORK_TRIP_KEYWORDS)
|
|
||||||
|
|
||||||
if not is_allowed_trip and not is_fio_whitelisted_in_kb:
|
|
||||||
anomalies.append({
|
|
||||||
"type": "ФИЗИЧЕСКОЕ ПРИСУТСТВИЕ ПРИ ОФИЦИАЛЬНОМ ОТСУТСТВИИ",
|
|
||||||
"fio": fio,
|
|
||||||
"details": f"Сотрудник пришел по СКУД, но в 1С оформлен документ: '{reason_1c}'"
|
|
||||||
})
|
|
||||||
|
|
||||||
if is_exc and not has_1c_reason:
|
|
||||||
continue
|
|
||||||
|
|
||||||
if anom_flag == 'ANOMALY_NO_IN_HAS_ACTIVITY':
|
|
||||||
first_act = row.get('Первая_активность', '—')
|
|
||||||
anomalies.append({
|
|
||||||
"type": "АНОМАЛИЯ СКУД: ПЕРЕМЕЩЕНИЕ БЕЗ ВХОДА",
|
|
||||||
"fio": fio,
|
|
||||||
"details": f"Отсутствует регистрация входа на КПП при зафиксированной первой активности в {first_act}"
|
|
||||||
})
|
|
||||||
|
|
||||||
if scud_fios_set is not None:
|
|
||||||
if fio_clean not in scud_fios_set and not has_1c_reason:
|
|
||||||
anomalies.append({
|
|
||||||
"type": "АНОМАЛИЯ УЧЕТА: СОТРУДНИК ОТСУТСТВУЕТ В СКУД ОРИОН PRO",
|
|
||||||
"fio": fio,
|
|
||||||
"details": f"Сотрудник числится в Штатном расписании 1С ({row.get('Подразделение', '—')}), но полностью отсутствует в базе СКУД Орион Pro (профиль не создан или карта не выдана)"
|
|
||||||
})
|
|
||||||
|
|
||||||
return anomalies
|
|
||||||
|
|
||||||
|
|
||||||
def aggregate_scud_by_employee(df, debug=False):
|
|
||||||
"""Агрегирует проходы СКУД по уникальным сотрудникам."""
|
|
||||||
if df is None or df.empty or 'fio_clean' not in df.columns:
|
|
||||||
return df
|
|
||||||
|
|
||||||
aggregated = []
|
|
||||||
for fio_clean, group in df.groupby('fio_clean', sort=False):
|
|
||||||
is_present = group['Пришел'].any() if 'Пришел' in group.columns else False
|
|
||||||
|
|
||||||
if is_present and 'Пришел' in group.columns:
|
|
||||||
present_rows = group[group['Пришел'] == True]
|
|
||||||
best_row = present_rows.iloc[0].to_dict() if not present_rows.empty else group.iloc[0].to_dict()
|
|
||||||
else:
|
|
||||||
best_row = group.iloc[0].to_dict()
|
|
||||||
|
|
||||||
best_row['Пришел'] = is_present
|
|
||||||
aggregated.append(best_row)
|
|
||||||
|
|
||||||
return pd.DataFrame(aggregated)
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
help_text = """
|
parser = argparse.ArgumentParser(description="Модульный контроллинг СКУД ⟷ 1С")
|
||||||
Система автоматизированного контроллинга СКУД ⟷ 1С:ЗУП (scud_orion_ai_v2)
|
parser.add_argument("-d", "--debug", action="store_true", help="Режим отладки")
|
||||||
|
parser.add_argument("--skip-export", action="store_true", help="Пропустить выгрузку СКУД из MS SQL")
|
||||||
ПРИМЕРЫ ЗАПУСКА:
|
parser.add_argument("--date", type=str, default=None, help="Дата расчета в формате ДД.ММ.ГГГГ")
|
||||||
python main.py -- Обычный дневной запуск
|
parser.add_argument("--time", type=str, default=None, help="Время среза для сводки (например, 14:30)")
|
||||||
python main.py --skip-export -- Расчет отчета по ПОСЛЕДНЕМУ имеющемуся снапшоту из SQLite
|
parser.add_argument("--snapshot", type=str, default=None, help="Точный ID снапшота для расчета")
|
||||||
python main.py --snapshot 20260805-001 -- Расчет отчета строго по ID снапшота
|
|
||||||
python main.py -d -- Запуск в режиме расширенной отладки (DEBUG)
|
|
||||||
"""
|
|
||||||
parser = argparse.ArgumentParser(
|
|
||||||
description=help_text,
|
|
||||||
formatter_class=argparse.RawDescriptionHelpFormatter
|
|
||||||
)
|
|
||||||
parser.add_argument('-d', '--debug', action='store_true', help="Запуск в режиме отладки с выводом подробных логов")
|
|
||||||
parser.add_argument('--skip-export', action='store_true', help="Пропустить выгрузку СКУД из MS SQL и построить отчет по последнему снапшоту")
|
|
||||||
parser.add_argument('--snapshot', type=str, default=None, help="Составной ID снапшота или время создания")
|
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
DEBUG = args.debug
|
|
||||||
|
|
||||||
print("=" * 60)
|
print("=" * 60)
|
||||||
print(f"ЗАПУСК СИСТЕМЫ МОДУЛЬНОГО КОНТРОЛЛИНГА СКУД ⟷ 1С {'[DEBUG MODE]' if DEBUG else ''}")
|
print(f"ЗАПУСК СИСТЕМЫ МОДУЛЬНОГО КОНТРОЛЛИНГА СКУД ⟷ 1С {'[DEBUG]' if args.debug else ''}")
|
||||||
print("=" * 60)
|
print("=" * 60)
|
||||||
|
|
||||||
init_db()
|
now = datetime.now()
|
||||||
kb_data = load_knowledge_base()
|
if args.date:
|
||||||
kb_rules = kb_data.get("rules", [])
|
today_str = args.date.replace('_', '.')
|
||||||
exceptions_cfg = load_exceptions_config()
|
dt_target = datetime.strptime(today_str, "%d.%m.%Y")
|
||||||
|
days_back = 3 if dt_target.weekday() == 0 else 1
|
||||||
if args.snapshot:
|
yesterday_str = (dt_target - timedelta(days=days_back)).strftime("%d.%m.%Y")
|
||||||
snapshot_param = args.snapshot
|
|
||||||
print(f"[📸] РЕЖИМ СНАПШОТА: Расчет отчета строго по срезу '{snapshot_param}'")
|
|
||||||
elif args.skip_export:
|
|
||||||
snapshot_param = get_latest_snapshot_time()
|
|
||||||
print(f"[📸] РЕЖИМ --skip-export: Используем последний снапшот из SQLite ('{snapshot_param}')")
|
|
||||||
else:
|
else:
|
||||||
snapshot_param = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
today_str = now.strftime("%d.%m.%Y")
|
||||||
print(f"[📸] СФОРМИРОВАН НОВЫЙ СНАПШОТ: '{snapshot_param}'")
|
if now.weekday() == 0:
|
||||||
|
yesterday_str = (now - timedelta(days=3)).strftime("%d.%m.%Y")
|
||||||
|
else:
|
||||||
|
yesterday_str = (now - timedelta(days=1)).strftime("%d.%m.%Y")
|
||||||
|
|
||||||
|
# [Этап 0] Выгрузка свежих данных СКУД
|
||||||
if not args.skip_export and not args.snapshot:
|
if not args.skip_export and not args.snapshot:
|
||||||
try:
|
print(f"\n[0/5] Экспорт данных СКУД за {today_str} и {yesterday_str}...")
|
||||||
run_export(save_xlsx=True, debug=DEBUG)
|
run_export(input_date=args.date, debug=args.debug, save_xlsx=True)
|
||||||
except Exception as e:
|
|
||||||
print(f"[⚠️] Ошибка автоэкспорта из БД: {e}. Переходим к записям в SQLite.")
|
|
||||||
else:
|
else:
|
||||||
print("[0/5] Пропуск прямого экспорта из MS SQL (чтение из базы SQLite)...")
|
print("\n[0/5] Пропуск прямого экспорта СКУД из MS SQL (--skip-export)...")
|
||||||
|
|
||||||
# ⚡️ ПРОВЕРКА ШАРЫ ТОЛЬКО ДЛЯ ОБЫЧНОГО ДНЕВНОГО ЗАПУСКА
|
# [Этап 0.5] Синхронизация файлов с шары 1С
|
||||||
if not args.snapshot and not args.skip_export:
|
if not args.snapshot and not args.skip_export:
|
||||||
|
print(f"\n[0.5/5] Проверка и копирование файлов 1С с шары...")
|
||||||
copy_1c_files_from_share()
|
copy_1c_files_from_share()
|
||||||
|
|
||||||
print("[1/5] Проверка актуальности и свежести входных данных...")
|
|
||||||
is_valid, warnings, errors, has_today_1c = check_file_freshness()
|
|
||||||
|
|
||||||
if warnings:
|
|
||||||
print("\n--- ⚠️ ПРЕДУПРЕЖДЕНИЯ ОБ АКТУАЛЬНОСТИ ---")
|
|
||||||
for w in warnings:
|
|
||||||
print(f" • {w}")
|
|
||||||
print("-" * 45)
|
|
||||||
|
|
||||||
if not is_valid:
|
|
||||||
print("\n" + "!" * 60)
|
|
||||||
print("🛑 ОСТАНОВКА ВЫПОЛНЕНИЯ: Отсутствуют критически важные файлы за вчерашний день!")
|
|
||||||
for e in errors:
|
|
||||||
print(f" {e}")
|
|
||||||
print("!" * 60)
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
print("[✓] Проверка доступности данных успешно пройдена!\n")
|
|
||||||
else:
|
else:
|
||||||
print("[1/5] Пропуск проверки сетевой шары (все данные читаются из SQLite)...")
|
print("\n[0.5/5] Пропуск синхронизации с шары (чтение локальных данных)...")
|
||||||
has_today_1c = True
|
|
||||||
|
|
||||||
print("[2/5] Загрузка данных из СКУД, 1С:ЗУП, реестра причин и исключений...")
|
# [Этап 1-2] Загрузка данных
|
||||||
|
print(f"\n[1-2/5] Загрузка срезов: Сегодня = {today_str}, Накануне = {yesterday_str}...")
|
||||||
# 🎯 ВЫЧИСЛЕНИЕ ДАТ СНАПШОТА И ДАТ НАКАНУНЕ
|
df_scud_yesterday = load_best_snapshot_for_date(yesterday_str, prefer_final_y=True)
|
||||||
if args.snapshot or args.skip_export:
|
df_scud_today = load_best_snapshot_for_date(today_str, prefer_final_y=False)
|
||||||
raw_scud_today_df = load_scud_from_db_by_snapshot(None, snapshot_param=snapshot_param)
|
|
||||||
|
|
||||||
if not raw_scud_today_df.empty and 'log_date' in raw_scud_today_df.columns:
|
|
||||||
target_date_str = str(raw_scud_today_df['log_date'].iloc[0])
|
|
||||||
else:
|
|
||||||
target_date_str = DATE_TODAY
|
|
||||||
|
|
||||||
dt_target = datetime.strptime(target_date_str, "%d.%m.%Y")
|
df_staff_yesterday, df_abs_yesterday = load_1c_files_for_date(yesterday_str)
|
||||||
if dt_target.weekday() == 0: # Понедельник -> Пятница
|
df_staff_today, df_abs_today = load_1c_files_for_date(today_str)
|
||||||
dt_yesterday = dt_target - timedelta(days=3)
|
|
||||||
else:
|
|
||||||
dt_yesterday = dt_target - timedelta(days=1)
|
|
||||||
yesterday_date_str = dt_yesterday.strftime("%d.%m.%Y")
|
|
||||||
|
|
||||||
print(f"[📸] СНАПШОТ ОПРЕДЕЛЕН: Целевая дата = {target_date_str}, Накануне = {yesterday_date_str}")
|
if df_scud_today is not None and not df_scud_today.empty:
|
||||||
|
export_raw_scud(df_scud_today, filename=f"СКУД_Сырые_данные_{today_str}.xlsx")
|
||||||
|
|
||||||
raw_scud_yesterday_df = load_scud_from_db_by_snapshot(yesterday_date_str, snapshot_param=None)
|
# [Этап 3] Детальный отчет за вчера через otchet_generator
|
||||||
|
print(f"\n[3/5] Обработка и построение детального отчета за ВЧЕРА ({yesterday_str})...")
|
||||||
|
res_otchet = generate_otchet_service(target_date=yesterday_str)
|
||||||
|
if res_otchet.get("status") == "success":
|
||||||
|
print(f"[✓] {res_otchet.get('message')}: {res_otchet.get('filepath')}")
|
||||||
|
|
||||||
# Гарантированная загрузка кадров 1С за обе даты (БД + локальный фолбэк)
|
# [Этап 4] Сводка за сегодня через svodka_generator
|
||||||
df_staff_today, df_absent_today = load_1c_data_smart(target_date_str, use_db=True)
|
print(f"\n[4/5] Обработка и построение Ежедневной сводки за {today_str} {args.time or ''}...")
|
||||||
df_staff_yesterday, df_absent_yesterday = load_1c_data_smart(yesterday_date_str, use_db=True)
|
res_svodka = generate_svodka_service(
|
||||||
else:
|
target_date=today_str,
|
||||||
target_date_str = DATE_TODAY
|
target_time=args.time,
|
||||||
yesterday_date_str = DATE_YESTERDAY
|
snapshot_id=args.snapshot
|
||||||
|
|
||||||
raw_scud_today_df = load_scud_from_db_by_snapshot(target_date_str, snapshot_param=snapshot_param)
|
|
||||||
raw_scud_yesterday_df = load_scud_from_db_by_snapshot(yesterday_date_str, snapshot_param=None)
|
|
||||||
|
|
||||||
df_staff_today, df_absent_today = load_1c_data_smart(DATE_TODAY, use_db=False)
|
|
||||||
df_staff_yesterday, df_absent_yesterday = load_1c_data_smart(DATE_YESTERDAY, use_db=False)
|
|
||||||
|
|
||||||
if df_staff_yesterday is not None:
|
|
||||||
save_staff_to_db(df_staff_yesterday, yesterday_date_str)
|
|
||||||
if df_absent_yesterday is not None:
|
|
||||||
save_absences_to_db(df_absent_yesterday, yesterday_date_str)
|
|
||||||
|
|
||||||
if df_staff_today is not None:
|
|
||||||
save_staff_to_db(df_staff_today, target_date_str)
|
|
||||||
if df_absent_today is not None:
|
|
||||||
save_absences_to_db(df_absent_today, target_date_str)
|
|
||||||
|
|
||||||
staff_fios_yesterday_clean = df_staff_yesterday['fio_clean'].dropna().tolist() if df_staff_yesterday is not None else []
|
|
||||||
static_reasons_dict = load_static_reason_workers()
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# 🎯 ЧАСТЬ 1: ДЕТАЛЬНЫЙ ОТЧЕТ ЗА ВЧЕРА (ДЕНЬ НАКАНУНЕ)
|
|
||||||
# ============================================================
|
|
||||||
print(f"\n[3/5] Обработка и построение детального отчета за ВЧЕРА ({yesterday_date_str})...")
|
|
||||||
|
|
||||||
if not raw_scud_yesterday_df.empty and 'Пришел' not in raw_scud_yesterday_df.columns:
|
|
||||||
raw_scud_yesterday_df['Пришел'] = raw_scud_yesterday_df['is_present'].astype(int) == 1 if 'is_present' in raw_scud_yesterday_df.columns else False
|
|
||||||
|
|
||||||
scud_yesterday_unrecognized = raw_scud_yesterday_df[~raw_scud_yesterday_df['fio_clean'].isin(staff_fios_yesterday_clean)]['fio_clean'].tolist() if not raw_scud_yesterday_df.empty else []
|
|
||||||
|
|
||||||
fio_mapping_scud_y = ai_verify_scud_against_staff(scud_yesterday_unrecognized, staff_fios_yesterday_clean)
|
|
||||||
if fio_mapping_scud_y:
|
|
||||||
raw_scud_yesterday_df['fio_clean'] = raw_scud_yesterday_df['fio_clean'].apply(
|
|
||||||
lambda x: fio_mapping_scud_y[x]['staff_fio'] if x in fio_mapping_scud_y else x
|
|
||||||
)
|
|
||||||
|
|
||||||
if df_absent_yesterday is not None and not df_absent_yesterday.empty:
|
|
||||||
absent_unrecognized_yesterday = df_absent_yesterday[~df_absent_yesterday['fio_clean'].isin(staff_fios_yesterday_clean)]['fio_clean'].tolist()
|
|
||||||
if absent_unrecognized_yesterday:
|
|
||||||
fio_mapping_absent_y = ai_verify_scud_against_staff(absent_unrecognized_yesterday, staff_fios_yesterday_clean)
|
|
||||||
if fio_mapping_absent_y:
|
|
||||||
df_absent_yesterday['fio_clean'] = df_absent_yesterday['fio_clean'].apply(
|
|
||||||
lambda x: fio_mapping_absent_y[x]['staff_fio'] if x in fio_mapping_absent_y else x
|
|
||||||
)
|
|
||||||
|
|
||||||
raw_scud_yesterday_df = aggregate_scud_by_employee(raw_scud_yesterday_df, debug=DEBUG)
|
|
||||||
|
|
||||||
# ⭐️ ФОРМИРУЕМ МЕРДЖ ИСКЛЮЧИТЕЛЬНО НА БАЗЕ ШТАТА ЗА ПРОШЛЫЙ ДЕНЬ
|
|
||||||
merged_yesterday = df_staff_yesterday.copy() if (df_staff_yesterday is not None and not df_staff_yesterday.empty) else (df_staff_today.copy() if df_staff_today is not None else pd.DataFrame())
|
|
||||||
|
|
||||||
if not merged_yesterday.empty:
|
|
||||||
if not raw_scud_yesterday_df.empty:
|
|
||||||
if 'Подразделение' in raw_scud_yesterday_df.columns:
|
|
||||||
raw_scud_yesterday_df['department_scud'] = raw_scud_yesterday_df['Подразделение']
|
|
||||||
elif 'department' in raw_scud_yesterday_df.columns:
|
|
||||||
raw_scud_yesterday_df['department_scud'] = raw_scud_yesterday_df['department']
|
|
||||||
else:
|
|
||||||
raw_scud_yesterday_df['department_scud'] = ''
|
|
||||||
|
|
||||||
merged_yesterday = merged_yesterday.merge(
|
|
||||||
raw_scud_yesterday_df[['fio_clean', 'Пришел', 'Начало_дня', 'Первая_активность', 'Конец_дня', 'Находился_в_здании', 'anomaly_flag', 'department_scud']],
|
|
||||||
on='fio_clean', how='left'
|
|
||||||
)
|
|
||||||
|
|
||||||
if 'Вид_отсутствия' in merged_yesterday.columns:
|
|
||||||
merged_yesterday = merged_yesterday.drop(columns=['Вид_отсутствия'])
|
|
||||||
|
|
||||||
# Присоединяем отсутствия СТРОГО за дату yesterday_date_str
|
|
||||||
if df_absent_yesterday is not None and not df_absent_yesterday.empty:
|
|
||||||
merged_yesterday = merged_yesterday.merge(
|
|
||||||
df_absent_yesterday[['fio_clean', 'Вид_отсутствия']],
|
|
||||||
on='fio_clean',
|
|
||||||
how='left'
|
|
||||||
)
|
|
||||||
|
|
||||||
if 'Пришел' not in merged_yesterday.columns:
|
|
||||||
merged_yesterday['Пришел'] = False
|
|
||||||
else:
|
|
||||||
merged_yesterday['Пришел'] = merged_yesterday['Пришел'].fillna(False)
|
|
||||||
|
|
||||||
if 'Сотрудник' not in merged_yesterday.columns:
|
|
||||||
merged_yesterday['Сотрудник'] = merged_yesterday.get('ФИО', merged_yesterday['fio_clean'])
|
|
||||||
|
|
||||||
if static_reasons_dict:
|
|
||||||
for fio_clean, reason_val in static_reasons_dict.items():
|
|
||||||
mask_yesterday = (
|
|
||||||
(merged_yesterday['Пришел'] == False) &
|
|
||||||
(merged_yesterday['Вид_отсутствия'].isna() | (merged_yesterday['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
|
||||||
(merged_yesterday['fio_clean'] == fio_clean)
|
|
||||||
)
|
|
||||||
merged_yesterday.loc[mask_yesterday, 'Вид_отсутствия'] = reason_val
|
|
||||||
|
|
||||||
merged_yesterday = apply_exceptions_from_json(merged_yesterday, exceptions_cfg)
|
|
||||||
mask_exc_yesterday = (
|
|
||||||
(merged_yesterday['Пришел'] == False) &
|
|
||||||
(merged_yesterday['Вид_отсутствия'].isna() | (merged_yesterday['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
|
||||||
(merged_yesterday.get('is_excluded', False) == True)
|
|
||||||
)
|
)
|
||||||
merged_yesterday.loc[mask_exc_yesterday, 'Вид_отсутствия'] = 'Исключение (ОВК/Подрядчики)'
|
if res_svodka.get("status") == "success":
|
||||||
|
print(f"[✓] {res_svodka.get('message')}: {res_svodka.get('filepath')}")
|
||||||
|
if res_svodka.get("note"):
|
||||||
|
print(f" ℹ️ {res_svodka.get('note')}")
|
||||||
|
|
||||||
scud_fios_yesterday_set = set(raw_scud_yesterday_df['fio_clean'].dropna().tolist()) if not raw_scud_yesterday_df.empty else set()
|
# [Этап 5] Формирование Markdown-сводки через ИИ-аудитора (Ollama)
|
||||||
merged_yesterday['no_scud_pass'] = (
|
print(f"\n[5/5] Формирование Markdown-сводки через ИИ-аудитора (Ollama)...")
|
||||||
(~merged_yesterday['fio_clean'].isin(scud_fios_yesterday_set)) &
|
|
||||||
(merged_yesterday['Вид_отсутствия'].isna() | (merged_yesterday['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
df_merged_today = merge_scud_and_1c(df_scud_today, df_staff_today, df_abs_today)
|
||||||
(merged_yesterday.get('is_excluded', False) == False)
|
anomalies_today = detect_registry_anomalies(df_merged_today, df_raw_scud=df_scud_today)
|
||||||
|
|
||||||
|
absent_explained = df_merged_today[
|
||||||
|
(df_merged_today['Пришел'] == False) &
|
||||||
|
(df_merged_today['Вид_отсутствия'].notna()) &
|
||||||
|
(~df_merged_today['Вид_отсутствия'].astype(str).str.startswith('Исключение'))
|
||||||
|
]
|
||||||
|
absent_unexplained = df_merged_today[
|
||||||
|
(df_merged_today['Пришел'] == False) &
|
||||||
|
(df_merged_today['Вид_отсутствия'].isna() | (df_merged_today['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
||||||
|
(df_merged_today.get('is_excluded', False) == False)
|
||||||
|
]
|
||||||
|
|
||||||
|
summary_md = generate_markdown_report(
|
||||||
|
merged_df=df_merged_today[df_merged_today.get('is_excluded', False) == False],
|
||||||
|
absent_explained=absent_explained,
|
||||||
|
absent_unexplained=absent_unexplained,
|
||||||
|
scud_present_but_absent_in_1c=pd.DataFrame(),
|
||||||
|
anomalies_list=anomalies_today,
|
||||||
|
raw_scud_df=df_scud_today,
|
||||||
|
raw_staff_df=df_staff_today,
|
||||||
|
raw_absent_df=df_abs_today,
|
||||||
|
date_str=today_str
|
||||||
)
|
)
|
||||||
|
|
||||||
|
os.makedirs("output", exist_ok=True)
|
||||||
|
md_file_path = f"output/Сводка_контроллинга_{today_str}.md"
|
||||||
|
with open(md_file_path, "w", encoding="utf-8") as f:
|
||||||
|
f.write(summary_md)
|
||||||
|
print(f"[✓] Текстовый отчет сохранен в: {md_file_path}")
|
||||||
|
|
||||||
anomalies_yesterday_list = detect_all_anomalies(merged_yesterday, static_reasons_dict, kb_rules, scud_fios_set=scud_fios_yesterday_set)
|
print("\n" + "=" * 60)
|
||||||
save_anomalies_to_db(anomalies_yesterday_list, yesterday_date_str)
|
print("ГОТОВАЯ ТЕКСТОВАЯ СВОДКА ИИ-АУДИТОРА:")
|
||||||
|
print("=" * 60)
|
||||||
filtered_yesterday = filter_report_dataframe(merged_yesterday)
|
print(summary_md)
|
||||||
generate_detailed_excel(merged_df=filtered_yesterday, date_str=yesterday_date_str)
|
|
||||||
|
|
||||||
yesterday_dt_obj = datetime.strptime(yesterday_date_str, "%d.%m.%Y")
|
|
||||||
yesterday_22_str = yesterday_dt_obj.strftime("%Y-%m-%d 22:00:00")
|
|
||||||
save_scud_to_db(merged_yesterday, yesterday_date_str, snapshot_time=yesterday_22_str, is_yesterday=True)
|
|
||||||
print(f"[✓] Детальный отчет за вчера сформирован и зафиксирован в SQLite за {yesterday_date_str}")
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# 🎯 ЧАСТЬ 2: ЕЖЕДНЕВНАЯ СВОДКА (ЗА ЦЕЛЕВОЙ ДЕНЬ СНАПШОТА)
|
|
||||||
# ============================================================
|
|
||||||
if not args.snapshot and not args.skip_export:
|
|
||||||
save_scud_to_db(raw_scud_today_df, target_date_str, snapshot_time=snapshot_param, is_yesterday=False)
|
|
||||||
|
|
||||||
if has_today_1c and df_staff_today is not None and df_absent_today is not None:
|
|
||||||
print(f"\n[4/5] Обработка и построение Ежедневной сводки за {target_date_str}...")
|
|
||||||
staff_fios_today_clean = df_staff_today['fio_clean'].dropna().tolist()
|
|
||||||
|
|
||||||
if not raw_scud_today_df.empty and 'Пришел' not in raw_scud_today_df.columns:
|
|
||||||
raw_scud_today_df['Пришел'] = raw_scud_today_df['is_present'].astype(int) == 1 if 'is_present' in raw_scud_today_df.columns else False
|
|
||||||
|
|
||||||
scud_unrecognized = raw_scud_today_df[~raw_scud_today_df['fio_clean'].isin(staff_fios_today_clean)]['fio_clean'].tolist() if not raw_scud_today_df.empty else []
|
|
||||||
fio_mapping_scud = ai_verify_scud_against_staff(scud_unrecognized, staff_fios_today_clean)
|
|
||||||
if fio_mapping_scud:
|
|
||||||
raw_scud_today_df['fio_clean'] = raw_scud_today_df['fio_clean'].apply(
|
|
||||||
lambda x: fio_mapping_scud[x]['staff_fio'] if x in fio_mapping_scud else x
|
|
||||||
)
|
|
||||||
|
|
||||||
absent_unrecognized_today = df_absent_today[~df_absent_today['fio_clean'].isin(staff_fios_today_clean)]['fio_clean'].tolist()
|
|
||||||
if absent_unrecognized_today:
|
|
||||||
fio_mapping_absent = ai_verify_scud_against_staff(absent_unrecognized_today, staff_fios_today_clean)
|
|
||||||
if fio_mapping_absent:
|
|
||||||
df_absent_today['fio_clean'] = df_absent_today['fio_clean'].apply(
|
|
||||||
lambda x: fio_mapping_absent[x]['staff_fio'] if x in fio_mapping_absent else x
|
|
||||||
)
|
|
||||||
|
|
||||||
raw_scud_today_df = aggregate_scud_by_employee(raw_scud_today_df, debug=DEBUG)
|
|
||||||
|
|
||||||
merged_today = df_staff_today.copy()
|
|
||||||
if not raw_scud_today_df.empty:
|
|
||||||
if 'Подразделение' in raw_scud_today_df.columns:
|
|
||||||
raw_scud_today_df['department_scud'] = raw_scud_today_df['Подразделение']
|
|
||||||
elif 'department' in raw_scud_today_df.columns:
|
|
||||||
raw_scud_today_df['department_scud'] = raw_scud_today_df['department']
|
|
||||||
else:
|
|
||||||
raw_scud_today_df['department_scud'] = ''
|
|
||||||
|
|
||||||
merged_today = merged_today.merge(
|
|
||||||
raw_scud_today_df[['fio_clean', 'Пришел', 'Начало_дня', 'Первая_активность', 'Конец_дня', 'Находился_в_здании', 'anomaly_flag', 'department_scud']],
|
|
||||||
on='fio_clean', how='left'
|
|
||||||
)
|
|
||||||
|
|
||||||
if df_absent_today is not None and not df_absent_today.empty:
|
|
||||||
merged_today = merged_today.merge(df_absent_today[['fio_clean', 'Вид_отсутствия']], on='fio_clean', how='left')
|
|
||||||
|
|
||||||
if 'Пришел' not in merged_today.columns:
|
|
||||||
merged_today['Пришел'] = False
|
|
||||||
else:
|
|
||||||
merged_today['Пришел'] = merged_today['Пришел'].fillna(False)
|
|
||||||
|
|
||||||
if 'Сотрудник' not in merged_today.columns:
|
|
||||||
merged_today['Сотрудник'] = merged_today.get('ФИО', merged_today['fio_clean'])
|
|
||||||
|
|
||||||
if static_reasons_dict:
|
|
||||||
for fio_clean, reason_val in static_reasons_dict.items():
|
|
||||||
mask_today = (
|
|
||||||
(merged_today['Пришел'] == False) &
|
|
||||||
(merged_today['Вид_отсутствия'].isna() | (merged_today['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
|
||||||
(merged_today['fio_clean'] == fio_clean)
|
|
||||||
)
|
|
||||||
merged_today.loc[mask_today, 'Вид_отсутствия'] = reason_val
|
|
||||||
|
|
||||||
merged_today = apply_exceptions_from_json(merged_today, exceptions_cfg)
|
|
||||||
mask_exc_today = (
|
|
||||||
(merged_today['Пришел'] == False) &
|
|
||||||
(merged_today['Вид_отсутствия'].isna() | (merged_today['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
|
||||||
(merged_today.get('is_excluded', False) == True)
|
|
||||||
)
|
|
||||||
merged_today.loc[mask_exc_today, 'Вид_отсутствия'] = 'Исключение (ОВК/Подрядчики)'
|
|
||||||
|
|
||||||
scud_fios_today_set = set(raw_scud_today_df['fio_clean'].dropna().tolist()) if not raw_scud_today_df.empty else set()
|
|
||||||
merged_today['no_scud_pass'] = (
|
|
||||||
(~merged_today['fio_clean'].isin(scud_fios_today_set)) &
|
|
||||||
(merged_today['Вид_отсутствия'].isna() | (merged_today['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
|
||||||
(merged_today.get('is_excluded', False) == False)
|
|
||||||
)
|
|
||||||
|
|
||||||
scud_fios_set = set(raw_scud_today_df['fio_clean'].dropna().tolist()) if not raw_scud_today_df.empty else set()
|
|
||||||
anomalies_list = detect_all_anomalies(merged_today, static_reasons_dict, kb_rules, scud_fios_set=scud_fios_set)
|
|
||||||
|
|
||||||
mass_failure_today = analyze_scud_mass_failure_ai(raw_scud_today_df)
|
|
||||||
if mass_failure_today and mass_failure_today.get("is_mass_failure"):
|
|
||||||
print("\n" + "!" * 60)
|
|
||||||
print(f"🚨 ВНИМАНИЕ! ИИ ОБНАРУЖИЛ ОПЕРАТИВНЫЙ СБОЙ ТУРНИКЕТОВ ВХОДА СЕГОДНЯ ({mass_failure_today['anomaly_percent']}% СМЕНЫ)")
|
|
||||||
print(mass_failure_today["alert_text"])
|
|
||||||
print("!" * 60 + "\n")
|
|
||||||
|
|
||||||
save_anomalies_to_db(anomalies_list, target_date_str)
|
|
||||||
|
|
||||||
is_no_pass_today = merged_today['no_scud_pass'] == True if 'no_scud_pass' in merged_today.columns else False
|
|
||||||
is_exc_today = merged_today.get('is_excluded', False) == True
|
|
||||||
|
|
||||||
absent_explained = merged_today[
|
|
||||||
(merged_today['Пришел'] == False) &
|
|
||||||
(merged_today['Вид_отсутствия'].notna()) &
|
|
||||||
(~merged_today['Вид_отсутствия'].astype(str).str.startswith('Исключение'))
|
|
||||||
]
|
|
||||||
|
|
||||||
absent_unexplained = merged_today[
|
|
||||||
(merged_today['Пришел'] == False) &
|
|
||||||
(merged_today['Вид_отсутствия'].isna() | (merged_today['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
|
||||||
(~is_no_pass_today) &
|
|
||||||
(~is_exc_today)
|
|
||||||
]
|
|
||||||
|
|
||||||
scud_present_but_absent_in_1c = merged_today[
|
|
||||||
(~is_exc_today) &
|
|
||||||
(merged_today['Пришел'] == True) &
|
|
||||||
(merged_today['Вид_отсутствия'].notna()) &
|
|
||||||
(~merged_today['Вид_отсутствия'].astype(str).str.startswith('Исключение'))
|
|
||||||
]
|
|
||||||
|
|
||||||
print("[5/5] Запуск ИИ-аудитора и построение Ежедневной сводки...")
|
|
||||||
filtered_anomalies_list = [
|
|
||||||
a for a in anomalies_list
|
|
||||||
if "ОТСУТСТВУЕТ В СКУД" not in a.get('type', '')
|
|
||||||
]
|
|
||||||
|
|
||||||
report_text = generate_markdown_report(
|
|
||||||
merged_df=merged_today,
|
|
||||||
absent_explained=absent_explained,
|
|
||||||
absent_unexplained=absent_unexplained,
|
|
||||||
scud_present_but_absent_in_1c=scud_present_but_absent_in_1c,
|
|
||||||
anomalies_list=filtered_anomalies_list,
|
|
||||||
raw_scud_df=raw_scud_today_df,
|
|
||||||
raw_absent_df=df_absent_today,
|
|
||||||
date_str=target_date_str
|
|
||||||
)
|
|
||||||
|
|
||||||
generate_summary_excel(merged_df=merged_today, date_str=target_date_str)
|
|
||||||
|
|
||||||
md_report_path = os.path.join(OUTPUT_DIR, f"Сводка_контроллинга_{target_date_str}.md")
|
|
||||||
with open(md_report_path, "w", encoding="utf-8") as f:
|
|
||||||
f.write(report_text)
|
|
||||||
print(f"\n[✓] Текстовый отчет сохранен в: {md_report_path}")
|
|
||||||
|
|
||||||
suspicious_cases = []
|
|
||||||
for a in anomalies_list:
|
|
||||||
if "ОФИЦИАЛЬНОМ ОТСУТСТВИИ" not in a.get('type', ''):
|
|
||||||
suspicious_cases.append({
|
|
||||||
'fio_target': a['fio'],
|
|
||||||
'reason': f"{a['type']}: {a['details']}"
|
|
||||||
})
|
|
||||||
|
|
||||||
if suspicious_cases:
|
|
||||||
review_ai_decisions(report_text, suspicious_cases)
|
|
||||||
|
|
||||||
print("\n" + "=" * 60)
|
|
||||||
print("ГОТОВАЯ ТЕКСТОВАЯ СВОДКА ИИ-АУДИТОРА:")
|
|
||||||
print("=" * 60)
|
|
||||||
print(report_text)
|
|
||||||
else:
|
|
||||||
print(f"\n[ℹ️] Формирование Ежедневной сводки за {target_date_str} ПРОПУЩЕНО.")
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
@@ -0,0 +1,56 @@
|
|||||||
|
# 🗺️ Дорожная карта задач проекта (ROADMAP)
|
||||||
|
|
||||||
|
> **Сформировано:** 2026-08-19 18:40 | **Всего задач:** 16
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Модуль `api`
|
||||||
|
|
||||||
|
- [x] **#35** [HIGH] `[ЗАВЕРШЕНО]` Создание REST API на FastAPI (/api/v1/snapshots, /api/v1/scud/logs, /api/v1/anomalies)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Модуль `core/database`
|
||||||
|
|
||||||
|
- [x] **#22** [HIGH] `[ЗАВЕРШЕНО]` Перевод подключения 1С:ЗУП на продакшн MS SQL сервер ACCOUNT-01 + гибридный фоллбэк
|
||||||
|
- [x] **#37** [MEDIUM] `[ЗАВЕРШЕНО]` Создание модуля персистентной памяти проекта в SQLite и CLI-инструмента для фиксации архитектурных решений
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Модуль `general`
|
||||||
|
|
||||||
|
- [x] **#38** [HIGH] `[ЗАВЕРШЕНО]` По запросу, вывести возможности системы, которые уже настроены. Например на запрос о возможностях по СКУД. И изменить стартовую плашку.
|
||||||
|
- [ ] **#41** [MEDIUM] Полностью отказаться от tool_injector.py и перевести все вызовы на нативный Function Calling языковой модели. LLM будет основным арбитром *(срок: 2026-08-21)*
|
||||||
|
- [ ] **#44** [LOW] Тестовая-3 *(срок: 2026-08-20)*
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Модуль `scripts`
|
||||||
|
|
||||||
|
- [x] **#26** [MEDIUM] `[ЗАВЕРШЕНО]` Создание и выполнение скрипта сквозной переиндексации снапшотов с префиксом Y
|
||||||
|
- [x] **#27** [MEDIUM] `[ЗАВЕРШЕНО]` Реализация CLI-инспектора БД и выравнивание номеров срезов для идеальной верстки в терминале
|
||||||
|
- [ ] **#33** [MEDIUM] Автоматизация сбора срезов по расписанию (Cron / Systemd Timers на 12:00, 17:00, 19:00)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Модуль `services/ai_verifier`
|
||||||
|
|
||||||
|
- [x] **#28** [HIGH] `[ЗАВЕРШЕНО]` Двойная проверка исключений по department_scud и 1С, удержание 100% приоритета кадровых документов
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Модуль `services/scud_export`
|
||||||
|
|
||||||
|
- [x] **#23** [HIGH] `[ЗАВЕРШЕНО]` Инцидент Равина В.Э.: Фиксация 100% приоритета сырых логов pLogData СКУД над расчетом УРВ Ориона
|
||||||
|
- [x] **#25** [HIGH] `[ЗАВЕРШЕНО]` Исправление бага дублирования номеров снапшотов (переход на вычисление MAX() по суффиксу)
|
||||||
|
- [x] **#29** [LOW] `[ЗАВЕРШЕНО]` Лаконичный вывод неразмеченных сотрудников с короткими аббревиатурами отделов СКУД (РУК, ОИЗ, ПУ и др.)
|
||||||
|
- [ ] **#30** [HIGH] `[В РАБОТЕ]` Внутридневной контроль: Извлечение полных цепочек проходов pLogData (входы/выходы за смену)
|
||||||
|
- [ ] **#31** [HIGH] `[В РАБОТЕ]` Внутридневной контроль: Расчет количества и суммарной длительности отлучек/перекуров сверх обеда
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Модуль `services/text_reporter`
|
||||||
|
|
||||||
|
- [ ] **#32** [MEDIUM] `[В РАБОТЕ]` Внутридневной контроль: Вывод списка системных нарушителей внутреннего распорядка в Markdown-отчет
|
||||||
|
|
||||||
|
---
|
||||||
@@ -0,0 +1,178 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/ai_engine/agent.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: modules / ai_engine
|
||||||
|
ROLE: Лаконичный нативный оркестратор Function Calling, диспетчер handlers
|
||||||
|
и менеджер свободных диалогов (Topic Drift).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[AGENT_PIPELINE_ENTRY]: Главная точка входа process_chat_message.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
from typing import List, Dict, Any, Tuple, Optional
|
||||||
|
|
||||||
|
from modules.web_api.llm.schemas import TOOLS_SCHEMA
|
||||||
|
from modules.web_api.llm.core.ollama_client import call_ollama_chat
|
||||||
|
from modules.web_api.llm.core.fast_path import handle_fast_path_intercept
|
||||||
|
from modules.web_api.llm.core.tool_injector import clean_raw_tool_tags, clean_output, inject_tools_if_needed
|
||||||
|
from modules.web_api.llm.core.context_manager import mark_last_user_message_ephemeral, close_tool_session_and_cleanup
|
||||||
|
from modules.web_api.llm.db.db_chat import db_save_chat_message, db_get_chat_history, db_purge_ephemeral_messages
|
||||||
|
from modules.web_api.llm.db.db_prompts import (
|
||||||
|
db_get_session_state, db_clear_session_state, db_set_session_state,
|
||||||
|
db_get_stats, db_get_anomalies, db_get_reference
|
||||||
|
)
|
||||||
|
from services.knowledge.service import get_rules
|
||||||
|
|
||||||
|
from .context_builder import build_agent_system_context
|
||||||
|
from .handlers.task_handler import handle_tasks_call
|
||||||
|
from .handlers.prompt_handler import handle_prompt_call
|
||||||
|
from .handlers.snapshot_handler import handle_snapshots_call
|
||||||
|
|
||||||
|
logger = logging.getLogger("AI_AGENT")
|
||||||
|
logger.setLevel(logging.INFO)
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[AGENT_PIPELINE_ENTRY]
|
||||||
|
def process_chat_message(
|
||||||
|
user_id: int,
|
||||||
|
user_message: str,
|
||||||
|
file_context: str = "",
|
||||||
|
image_b64: Optional[str] = None,
|
||||||
|
chat_history: List[Dict[str, Any]] = None,
|
||||||
|
session_id: str = "web_session_main"
|
||||||
|
) -> Tuple[str, List[Dict[str, Any]], Optional[Dict[str, Any]]]:
|
||||||
|
"""Главный конвейер обработки сообщений чата."""
|
||||||
|
logger.info(f"Сообщение от user_id={user_id}, session_id={session_id}: {user_message}")
|
||||||
|
|
||||||
|
session_state = db_get_session_state(session_id)
|
||||||
|
if not session_state:
|
||||||
|
db_purge_ephemeral_messages(session_id)
|
||||||
|
|
||||||
|
full_user_content = f"{user_message}\n\n[СОДЕРЖИМОЕ ПРИКРЕПЛЕННОГО ФАЙЛА]:\n{file_context}" if file_context else user_message
|
||||||
|
|
||||||
|
# 1. Быстрый Fast-Path перехват кнопок подтверждения
|
||||||
|
fast_path_res = handle_fast_path_intercept(session_id, user_message, full_user_content, session_state)
|
||||||
|
if fast_path_res:
|
||||||
|
return fast_path_res
|
||||||
|
|
||||||
|
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=0)
|
||||||
|
db_history = db_get_chat_history(session_id, limit=20)
|
||||||
|
system_prompt = build_agent_system_context(user_id, session_state)
|
||||||
|
|
||||||
|
user_msg_obj = {"role": "user", "content": full_user_content}
|
||||||
|
|
||||||
|
try:
|
||||||
|
if image_b64:
|
||||||
|
user_msg_obj["images"] = [image_b64]
|
||||||
|
messages = [{"role": "system", "content": "Строгий модуль OCR. Перепиши весь текст буква в букву."}, user_msg_obj]
|
||||||
|
msg = call_ollama_chat(messages, is_vision=True)
|
||||||
|
else:
|
||||||
|
clean_history = [dict(m) for m in db_history]
|
||||||
|
for m in clean_history: m.pop("images", None)
|
||||||
|
messages = [{"role": "system", "content": system_prompt}] + clean_history + [user_msg_obj]
|
||||||
|
msg = call_ollama_chat(messages, tools=TOOLS_SCHEMA, is_vision=False)
|
||||||
|
|
||||||
|
raw_reply = msg.get("content", "")
|
||||||
|
tool_calls = msg.get("tool_calls", [])
|
||||||
|
|
||||||
|
# 2. Гибридный семантический классификатор намерений (Fallback Safety Net)
|
||||||
|
tool_calls = inject_tools_if_needed(user_message, raw_reply, tool_calls)
|
||||||
|
|
||||||
|
# 3. Исполнение инструментов через изолированные handlers
|
||||||
|
if tool_calls:
|
||||||
|
tool = tool_calls[0]
|
||||||
|
fn_name = tool["function"]["name"]
|
||||||
|
fn_args = tool["function"].get("arguments", {})
|
||||||
|
if isinstance(fn_args, str):
|
||||||
|
try: fn_args = json.loads(fn_args)
|
||||||
|
except Exception: fn_args = {}
|
||||||
|
|
||||||
|
logger.info(f"Вызов инструмента: {fn_name} с аргументами: {fn_args}")
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason=f"ACTIVATE_{fn_name}")
|
||||||
|
mark_last_user_message_ephemeral(session_id)
|
||||||
|
|
||||||
|
state_data = session_state.get("data_json") or {} if session_state else {}
|
||||||
|
|
||||||
|
if fn_name in ["db_get_tasks", "db_tasks_edit", "db_add_task", "db_update_task_status", "db_delete_task"]:
|
||||||
|
return handle_tasks_call(fn_name, fn_args, user_id, session_id)
|
||||||
|
|
||||||
|
elif fn_name in ["db_get_system_prompt", "db_prompt_node_edit"]:
|
||||||
|
return handle_prompt_call(fn_name, fn_args, session_id)
|
||||||
|
|
||||||
|
elif fn_name in ["db_get_snapshots", "db_delete_snapshots"]:
|
||||||
|
return handle_snapshots_call(fn_name, fn_args, session_id, user_message, state_data)
|
||||||
|
|
||||||
|
elif fn_name == "db_get_rules":
|
||||||
|
res_str = json.dumps(get_rules(), ensure_ascii=False)
|
||||||
|
elif fn_name == "db_get_stats":
|
||||||
|
res_str = json.dumps(db_get_stats(), ensure_ascii=False)
|
||||||
|
elif fn_name == "db_get_anomalies":
|
||||||
|
res_str = json.dumps(db_get_anomalies(limit=fn_args.get("limit", 100), date_str=fn_args.get("date_str")), ensure_ascii=False)
|
||||||
|
elif fn_name == "db_get_reference":
|
||||||
|
res_str = json.dumps(db_get_reference(category=fn_args.get("category")), ensure_ascii=False)
|
||||||
|
else:
|
||||||
|
res_str = "{}"
|
||||||
|
|
||||||
|
messages.append(msg)
|
||||||
|
messages.append({"role": "tool", "content": res_str})
|
||||||
|
sec_msg = call_ollama_chat(messages, is_vision=False)
|
||||||
|
final_content = clean_raw_tool_tags(clean_output(sec_msg.get("content", ""))) or "Запрос выполнен."
|
||||||
|
db_save_chat_message(session_id, "assistant", final_content, is_ephemeral=0)
|
||||||
|
return final_content, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
# 4. Обычный содержательный диалог и управление Topic Drift
|
||||||
|
final_reply = clean_raw_tool_tags(clean_output(raw_reply)) or "Запрос обработан."
|
||||||
|
for artifact in ["почемучка,", "почемучка!", "почемучка?", "почемучка", "почемучто,", "почемучто", "почему-то"]:
|
||||||
|
if final_reply.lower().startswith(artifact):
|
||||||
|
final_reply = final_reply[len(artifact):].lstrip(",.!?:; -")
|
||||||
|
|
||||||
|
action_payload = None
|
||||||
|
|
||||||
|
# Обработка ответа "нет / спасибо" в режиме открытого инструмента
|
||||||
|
if session_state and any(kw in user_message.lower() for kw in ["нет", "спасибо", "не надо", "готово", "хватит"]):
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason="USER_DISMISSED_TOOL")
|
||||||
|
db_save_chat_message(session_id, "assistant", final_reply, is_ephemeral=0)
|
||||||
|
return final_reply, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
# Инкремент счётчика шагов в сторону от инструмента (Topic Drift)
|
||||||
|
if session_state and session_state.get("state_type") in ["PROMPT_FOLLOWUP", "PROMPT_PREVIEW", "SNAPSHOTS_VIEW"]:
|
||||||
|
state_type = session_state.get("state_type")
|
||||||
|
state_data = session_state.get("data_json") or {}
|
||||||
|
if not isinstance(state_data, dict):
|
||||||
|
state_data = {}
|
||||||
|
|
||||||
|
idle_turns = state_data.get("idle_turns", 0) + 1
|
||||||
|
state_data["idle_turns"] = idle_turns
|
||||||
|
|
||||||
|
if idle_turns >= 4:
|
||||||
|
# 4-й шаг не по теме: бесшумно закрываем сессию и вычищаем эфемерные карточки
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason="TOPIC_DRIFT_TIMEOUT")
|
||||||
|
elif idle_turns == 3:
|
||||||
|
# 3-й шаг: выводим вежливое напоминание с кнопками
|
||||||
|
tool_label = "системным промптом" if "PROMPT" in state_type else "снапшотами СКУД"
|
||||||
|
guard_question = f"Желаете продолжить работу с {tool_label}?"
|
||||||
|
final_reply += f"\n\n💡 *Напоминание:* {guard_question}"
|
||||||
|
action_payload = {
|
||||||
|
"type": "FOLLOW_UP_ACTION",
|
||||||
|
"buttons": [
|
||||||
|
{"label": "Показать снова", "value": "покажи системный промпт" if "PROMPT" in state_type else "покажи снапшоты", "style": "primary"},
|
||||||
|
{"label": "Завершить", "value": "нет, спасибо", "style": "secondary"}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
db_set_session_state(session_id, state_type, state_data)
|
||||||
|
else:
|
||||||
|
# 1-й и 2-й шаг: фиксируем обновленный счётчик
|
||||||
|
db_set_session_state(session_id, state_type, state_data)
|
||||||
|
|
||||||
|
is_ephem_reply = 1 if "актуальный системный промпт:" in final_reply.lower() else 0
|
||||||
|
db_save_chat_message(session_id, "assistant", final_reply, is_ephemeral=is_ephem_reply)
|
||||||
|
return final_reply, db_get_chat_history(session_id), action_payload
|
||||||
|
|
||||||
|
except Exception as ex:
|
||||||
|
logger.exception(f"Ошибка в агенте: {ex}")
|
||||||
|
return f"Внутренняя ошибка сервера: {ex}", db_get_chat_history(session_id), None
|
||||||
@@ -0,0 +1,84 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/ai_engine/context_builder.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: modules / ai_engine
|
||||||
|
ROLE: Динамическая сборка системного промпта из БД, календаря и контекста сессий.
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[CONTEXT_BUILDER_MAIN]: Формирование полного системного контекста для LLM.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Dict, Any, Optional
|
||||||
|
from services.prompts.service import get_active_system_prompt
|
||||||
|
from modules.web_api.llm.core.calendar_utils import get_dynamic_calendar_context
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[CONTEXT_BUILDER_MAIN]
|
||||||
|
def build_agent_system_context(
|
||||||
|
user_id: int,
|
||||||
|
session_state: Optional[Dict[str, Any]]
|
||||||
|
) -> str:
|
||||||
|
"""Формирует полный динамический системный контекст агента с правилами из БД."""
|
||||||
|
|
||||||
|
# 1. Загружаем актуальный базовый системный промпт из SQLite БД
|
||||||
|
base_system_prompt = get_active_system_prompt()
|
||||||
|
|
||||||
|
# 2. Серверный календарь и контекст дат
|
||||||
|
calendar_context = get_dynamic_calendar_context()
|
||||||
|
user_info = f"Пользователь ID={user_id}" if user_id != 0 else "Гость"
|
||||||
|
|
||||||
|
# 3. Контекст текущего активного стейта сессии
|
||||||
|
current_state_type = session_state.get("state_type") if session_state else None
|
||||||
|
state_data = session_state.get("data_json") or {} if session_state else {}
|
||||||
|
if not isinstance(state_data, dict):
|
||||||
|
state_data = {}
|
||||||
|
|
||||||
|
active_state_context = ""
|
||||||
|
if current_state_type == "PROMPT_PREVIEW":
|
||||||
|
active_state_context = (
|
||||||
|
"\n[ТЕКУЩИЙ РЕЖИМ: ПРЕДПРОСМОТР СИСТЕМНОГО ПРОМПТА]\n"
|
||||||
|
"- Открыт предпросмотр изменений промпта. Для любых правок вызывай db_prompt_node_edit.\n"
|
||||||
|
)
|
||||||
|
elif current_state_type == "PROMPT_FOLLOWUP":
|
||||||
|
active_state_context = (
|
||||||
|
"\n[ТЕКУЩИЙ РЕЖИМ: СЕССИЯ РЕДАКТИРОВАНИЯ ПРОМПТА]\n"
|
||||||
|
"- Оператор просматривает или редактирует системный промпт.\n"
|
||||||
|
"- На любые команды вида 'удали пункт X.Y' или 'удали X.Y' ТЫ ОБЯЗАН ВЫЗВАТЬ db_prompt_node_edit с action='DELETE', section_id=X, item_id=Y.\n"
|
||||||
|
"- На любые команды 'добавь пункт X.Y ...' вызывай action='ADD'.\n"
|
||||||
|
"- Запрещено путать ADD и DELETE.\n"
|
||||||
|
)
|
||||||
|
elif current_state_type == "SNAPSHOTS_VIEW":
|
||||||
|
active_date = state_data.get("query_date", "выбранную дату")
|
||||||
|
active_state_context = (
|
||||||
|
f"\n[ТЕКУЩИЙ РЕЖИМ: ПРОСМОТР СНАПШОТОВ СКУД]\n"
|
||||||
|
f"- Сейчас на экране отображаются снапшоты за {active_date}.\n"
|
||||||
|
f"- Ты можешь форматировать, фильтровать или анализировать этот текущий срез.\n"
|
||||||
|
f"- Если оператор запрашивает ДРУГУЮ дату или день недели (например: 'за вчера', 'а за 13.08') — "
|
||||||
|
f"ТЫ ОБЯЗАН СРАЗУ ВЫЗВАТЬ db_get_snapshots(date_str=...).\n"
|
||||||
|
f"- Запрещено генерировать текст за другую дату по памяти.\n"
|
||||||
|
)
|
||||||
|
elif current_state_type == "SNAPSHOT_DELETE_CONFIRM":
|
||||||
|
active_state_context = "\n[ВНИМАНИЕ: ОЖИДАЕТСЯ ПОДТВЕРЖДЕНИЕ УДАЛЕНИЯ СНАПШОТОВ]\n"
|
||||||
|
elif current_state_type == "TASK_DELETE_CONFIRM":
|
||||||
|
active_state_context = "\n[ВНИМАНИЕ: ОЖИДАЕТСЯ ПОДТВЕРЖДЕНИЕ УДАЛЕНИЯ ЗАДАЧИ]\n"
|
||||||
|
|
||||||
|
# 4. Сборка полного системного сообщения
|
||||||
|
return (
|
||||||
|
f"{base_system_prompt}\n\n"
|
||||||
|
f"СТРОГИЕ ПРАВИЛА ВЫЗОВА ИНСТРУМЕНТОВ:\n"
|
||||||
|
f"1. Для системного промпта:\n"
|
||||||
|
f" - 'добавь X.Y Текст' -> СРАЗУ вызывай db_prompt_node_edit(action='ADD', section_id=X, item_id=Y, content='Текст').\n"
|
||||||
|
f" - 'удали X.Y' -> СРАЗУ вызывай db_prompt_node_edit(action='DELETE', section_id=X, item_id=Y, content='').\n"
|
||||||
|
f" - 'покажи системный промпт' -> СРАЗУ вызывай db_get_system_prompt().\n"
|
||||||
|
f"2. Для задач:\n"
|
||||||
|
f" - 'удали задачу N' -> СРАЗУ вызывай db_tasks_edit(action='DELETE', task_id='N').\n"
|
||||||
|
f" - 'возьми в работу N' -> СРАЗУ вызывай db_tasks_edit(action='UPDATE', task_id='N', status='IN_PROGRESS').\n"
|
||||||
|
f" - 'заверши N' / 'готово N' -> СРАЗУ вызывай db_tasks_edit(action='UPDATE', task_id='N', status='COMPLETED').\n"
|
||||||
|
f"3. Запрещено задавать вопросы и писать подтверждения текстом — СРАЗУ вызывай соответствующий инструмент!\n\n"
|
||||||
|
f"[СИСТЕМНЫЙ КАЛЕНДАРЬ СЕРВЕРА]\n"
|
||||||
|
f"- Пользователь: {user_info}\n"
|
||||||
|
f"- {calendar_context}\n"
|
||||||
|
f"{active_state_context}"
|
||||||
|
)
|
||||||
@@ -0,0 +1,98 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/ai_engine/handlers/prompt_handler.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: modules / ai_engine / handlers
|
||||||
|
ROLE: Изолированная обработка команд управления системным промптом.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Dict, Any, Tuple, Optional
|
||||||
|
from services.prompts.service import get_active_system_prompt, create_prompt_preview
|
||||||
|
from modules.web_api.llm.db.db_chat import db_save_chat_message, db_get_chat_history
|
||||||
|
from modules.web_api.llm.db.db_prompts import db_set_session_state
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[PROMPT_HANDLER_DISPATCH]
|
||||||
|
def handle_prompt_call(
|
||||||
|
fn_name: str,
|
||||||
|
fn_args: Dict[str, Any],
|
||||||
|
session_id: str
|
||||||
|
) -> Tuple[str, list, Optional[Dict[str, Any]]]:
|
||||||
|
"""Обрабатывает вызовы просмотра и изменения системного промпта."""
|
||||||
|
|
||||||
|
# 1. Просмотр промпта с кнопкой быстрого перехода в редактор
|
||||||
|
if fn_name == "db_get_system_prompt":
|
||||||
|
active_prompt = get_active_system_prompt()
|
||||||
|
reply_text = f"Актуальный системный промпт:\n\n{active_prompt}"
|
||||||
|
|
||||||
|
# Сохраняем состояние сессии для возможности мгновенного редактирования и подтверждения
|
||||||
|
db_set_session_state(session_id, "PROMPT_PREVIEW", {
|
||||||
|
"draft_text": active_prompt,
|
||||||
|
"action": "MANUAL_EDIT",
|
||||||
|
"idle_turns": 0
|
||||||
|
})
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "PROMPT_PREVIEW",
|
||||||
|
"raw_draft": active_prompt,
|
||||||
|
"baseline_prompt": active_prompt,
|
||||||
|
"buttons": [
|
||||||
|
{"label": "✏️ Редактировать", "value": "action:open_editor", "style": "secondary"},
|
||||||
|
{"label": "Готово", "value": "нет, спасибо", "style": "secondary"}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
# 2. Предпросмотр точечных или пакетных изменений (ADD / EDIT / DELETE / BATCH_DELETE)
|
||||||
|
action = str(fn_args.get("action", "ADD")).upper()
|
||||||
|
nodes_list = fn_args.get("nodes_list", [])
|
||||||
|
delete_nodes_tuples = []
|
||||||
|
|
||||||
|
if nodes_list:
|
||||||
|
for n_str in nodes_list:
|
||||||
|
parts = str(n_str).strip().split(".")
|
||||||
|
if len(parts) == 2 and parts[0].isdigit() and parts[1].isdigit():
|
||||||
|
delete_nodes_tuples.append((int(parts[0]), int(parts[1])))
|
||||||
|
|
||||||
|
sec_id = None
|
||||||
|
itm_id = None
|
||||||
|
try:
|
||||||
|
if fn_args.get("section_id") is not None:
|
||||||
|
sec_id = int(str(fn_args.get("section_id")).strip())
|
||||||
|
if fn_args.get("item_id") is not None:
|
||||||
|
itm_id = int(str(fn_args.get("item_id")).strip())
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
content = str(fn_args.get("content", "")).strip()
|
||||||
|
|
||||||
|
merged_prompt, diff_html, baseline_prompt = create_prompt_preview(
|
||||||
|
action=action,
|
||||||
|
section_id=sec_id,
|
||||||
|
item_id=itm_id,
|
||||||
|
content=content,
|
||||||
|
delete_nodes=delete_nodes_tuples if delete_nodes_tuples else None
|
||||||
|
)
|
||||||
|
|
||||||
|
db_set_session_state(session_id, "PROMPT_PREVIEW", {
|
||||||
|
"draft_text": merged_prompt,
|
||||||
|
"action": "MANUAL_EDIT",
|
||||||
|
"section_id": sec_id,
|
||||||
|
"item_id": itm_id,
|
||||||
|
"content": content,
|
||||||
|
"idle_turns": 0
|
||||||
|
})
|
||||||
|
|
||||||
|
preview_reply = f"Предпросмотр изменений системного промпта:\n\n{diff_html}\n\nДля применения подтвердите действие, отредактируйте или отмените."
|
||||||
|
db_save_chat_message(session_id, "assistant", preview_reply, is_ephemeral=1)
|
||||||
|
return preview_reply, db_get_chat_history(session_id), {
|
||||||
|
"type": "PROMPT_PREVIEW",
|
||||||
|
"raw_draft": merged_prompt,
|
||||||
|
"baseline_prompt": baseline_prompt,
|
||||||
|
"buttons": [
|
||||||
|
{"label": "Подтвердить", "value": "подтверждаю", "style": "primary"},
|
||||||
|
{"label": "Отменить", "value": "отмена", "style": "danger"},
|
||||||
|
{"label": "✏️ Редактировать", "value": "action:open_editor", "style": "secondary"}
|
||||||
|
]
|
||||||
|
}
|
||||||
@@ -0,0 +1,91 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/ai_engine/handlers/snapshot_handler.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: modules / ai_engine / handlers
|
||||||
|
ROLE: Изолированная обработка запросов просмотра и удаления срезов СКУД.
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[SNAPSHOT_HANDLER_DISPATCH]: Обработка db_get_snapshots и db_delete_snapshots.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Dict, Any, Tuple, Optional, List
|
||||||
|
from services.snapshots.service import get_snapshots_registry, delete_snapshots_safely
|
||||||
|
from modules.web_api.llm.core.calendar_utils import parse_relative_date_ru
|
||||||
|
from modules.web_api.llm.db.db_chat import db_save_chat_message, db_get_chat_history
|
||||||
|
from modules.web_api.llm.db.db_prompts import db_set_session_state
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[SNAPSHOT_HANDLER_DISPATCH]
|
||||||
|
def handle_snapshots_call(
|
||||||
|
fn_name: str,
|
||||||
|
fn_args: Dict[str, Any],
|
||||||
|
session_id: str,
|
||||||
|
user_message: str,
|
||||||
|
state_data: Dict[str, Any]
|
||||||
|
) -> Tuple[str, list, Optional[Dict[str, Any]]]:
|
||||||
|
"""Обрабатывает запросы реестра и безопасного удаления срезов."""
|
||||||
|
|
||||||
|
# 1. Получение срезов
|
||||||
|
if fn_name == "db_get_snapshots":
|
||||||
|
date_param = fn_args.get("date_str")
|
||||||
|
if not date_param and user_message:
|
||||||
|
date_param = parse_relative_date_ru(user_message)
|
||||||
|
|
||||||
|
snapshots_res = get_snapshots_registry(date_str=date_param)
|
||||||
|
query_date = snapshots_res.get("query_date", "выбранную дату")
|
||||||
|
|
||||||
|
db_set_session_state(session_id, "SNAPSHOTS_VIEW", snapshots_res)
|
||||||
|
reply_text = f"Реестр срезов СКУД за {query_date}:"
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "SNAPSHOTS_CARD",
|
||||||
|
"data": snapshots_res
|
||||||
|
}
|
||||||
|
|
||||||
|
# 2. Удаление срезов (двухфазное подтверждение)
|
||||||
|
raw_id = fn_args.get("snapshot_id") or fn_args.get("day_str")
|
||||||
|
raw_ids = fn_args.get("snapshot_ids") or []
|
||||||
|
is_confirmed = fn_args.get("confirmed", False)
|
||||||
|
|
||||||
|
if raw_id and not raw_ids:
|
||||||
|
if isinstance(raw_id, str) and "," in raw_id:
|
||||||
|
raw_ids = [s.strip() for s in raw_id.split(",")]
|
||||||
|
else:
|
||||||
|
raw_ids = [raw_id]
|
||||||
|
|
||||||
|
safe_ids = [s.strip() for s in raw_ids if s and not str(s).strip().startswith("Y")]
|
||||||
|
if not safe_ids:
|
||||||
|
reply_text = "⚠️ Итоговый срез Y защищен от удаления. Выберите дневные снапшоты."
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
if not is_confirmed:
|
||||||
|
query_date = state_data.get("query_date", "")
|
||||||
|
db_set_session_state(session_id, "SNAPSHOT_DELETE_CONFIRM", {
|
||||||
|
"snapshot_ids": safe_ids,
|
||||||
|
"query_date": query_date,
|
||||||
|
"idle_turns": 0
|
||||||
|
})
|
||||||
|
ids_str = ", ".join(safe_ids)
|
||||||
|
reply_text = f"Вы действительно хотите удалить дневные снапшоты: {ids_str}?"
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "SNAPSHOT_DELETE_CONFIRM",
|
||||||
|
"buttons": [
|
||||||
|
{"label": f"Удалить ({len(safe_ids)} шт.)", "value": f"подтверждаю удаление снапшотов {ids_str}", "style": "danger"},
|
||||||
|
{"label": "Отмена", "value": "отмена", "style": "secondary"}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
else:
|
||||||
|
delete_snapshots_safely(snapshot_ids=safe_ids)
|
||||||
|
query_date = state_data.get("query_date", "")
|
||||||
|
updated_data = get_snapshots_registry(date_str=query_date)
|
||||||
|
|
||||||
|
reply_text = f"✅ Успешно удалено снапшотов: {len(safe_ids)} шт."
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "SNAPSHOTS_CARD",
|
||||||
|
"data": updated_data
|
||||||
|
}
|
||||||
@@ -0,0 +1,97 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/ai_engine/handlers/task_handler.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: modules / ai_engine / handlers
|
||||||
|
ROLE: Изолированная обработка вызовов инструментов задач (get, edit, delete, export).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[TASK_HANDLER_DISPATCH]: Обработка вызовов db_get_tasks и db_tasks_edit.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Dict, Any, Tuple, Optional
|
||||||
|
from services.tasks.service import get_tasks, execute_task_action
|
||||||
|
from modules.web_api.llm.db.db_chat import db_save_chat_message, db_get_chat_history
|
||||||
|
from modules.web_api.llm.db.db_prompts import db_set_session_state
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[TASK_HANDLER_DISPATCH]
|
||||||
|
def handle_tasks_call(
|
||||||
|
fn_name: str,
|
||||||
|
fn_args: Dict[str, Any],
|
||||||
|
user_id: int,
|
||||||
|
session_id: str
|
||||||
|
) -> Tuple[str, list, Optional[Dict[str, Any]]]:
|
||||||
|
"""Обрабатывает нативные tool-вызовы по задачам."""
|
||||||
|
|
||||||
|
# 1. Просмотр задач
|
||||||
|
if fn_name == "db_get_tasks":
|
||||||
|
raw_tasks = get_tasks(user_id, status=fn_args.get("status"))
|
||||||
|
reply_text = "Вот интерактивный список ваших текущих задач:"
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "TASK_INTERACTIVE_CARD",
|
||||||
|
"tasks": raw_tasks
|
||||||
|
}
|
||||||
|
|
||||||
|
# 2. Модификация / Удаление / Экспорт
|
||||||
|
action = (fn_args.get("action") or "UPDATE").upper()
|
||||||
|
if fn_name == "db_add_task": action = "ADD"
|
||||||
|
elif fn_name == "db_delete_task": action = "DELETE"
|
||||||
|
elif fn_name == "db_update_task_status": action = "UPDATE"
|
||||||
|
|
||||||
|
# Двухфазное подтверждение удаления
|
||||||
|
if action == "DELETE":
|
||||||
|
task_id_raw = str(fn_args.get("task_id", "")).replace("#", "").replace("TASK-", "").strip()
|
||||||
|
db_set_session_state(session_id, "TASK_DELETE_CONFIRM", {"task_id": task_id_raw, "idle_turns": 0})
|
||||||
|
reply_text = f"Вы действительно хотите удалить задачу #{task_id_raw}?"
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "TASK_DELETE_CONFIRM",
|
||||||
|
"buttons": [
|
||||||
|
{"label": f"Удалить #{task_id_raw}", "value": f"подтверждаю удаление задачи {task_id_raw}", "style": "danger"},
|
||||||
|
{"label": "Отмена", "value": "отмена", "style": "secondary"}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
# Экспорт в Markdown
|
||||||
|
elif action == "EXPORT":
|
||||||
|
export_res = execute_task_action(
|
||||||
|
user_id=user_id,
|
||||||
|
action="EXPORT",
|
||||||
|
filename=fn_args.get("filename"),
|
||||||
|
status=fn_args.get("status")
|
||||||
|
)
|
||||||
|
reply_text = export_res.get("message", "Отчет по задачам успешно экспортирован.")
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=0)
|
||||||
|
|
||||||
|
action_payload = None
|
||||||
|
if export_res.get("status") == "success":
|
||||||
|
action_payload = {
|
||||||
|
"type": "FILE_DOWNLOAD_CARD",
|
||||||
|
"filename": export_res.get("filename"),
|
||||||
|
"download_url": export_res.get("download_url"),
|
||||||
|
"tasks_count": export_res.get("tasks_count")
|
||||||
|
}
|
||||||
|
return reply_text, db_get_chat_history(session_id), action_payload
|
||||||
|
|
||||||
|
# Создание и обновление
|
||||||
|
else:
|
||||||
|
res = execute_task_action(
|
||||||
|
user_id=user_id,
|
||||||
|
action=action,
|
||||||
|
task_id=fn_args.get("task_id"),
|
||||||
|
title=fn_args.get("title"),
|
||||||
|
priority=fn_args.get("priority", "MEDIUM"),
|
||||||
|
status=fn_args.get("status"),
|
||||||
|
module=fn_args.get("module", "general"),
|
||||||
|
due_date=fn_args.get("due_date")
|
||||||
|
)
|
||||||
|
raw_tasks = get_tasks(user_id)
|
||||||
|
reply_text = res.get("message", "Операция над задачами выполнена.")
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "TASK_INTERACTIVE_CARD",
|
||||||
|
"tasks": raw_tasks
|
||||||
|
}
|
||||||
+423
-312
@@ -2,54 +2,63 @@
|
|||||||
===============================================================================
|
===============================================================================
|
||||||
FILE: modules/web_api/llm/agent.py
|
FILE: modules/web_api/llm/agent.py
|
||||||
PROJECT: SCUD Orion AI (Unified Architecture)
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
MODULE: web_api / llm (Core Agent & Function Calling Dispatcher)
|
MODULE: web_api / llm (Core Agent Coordinator)
|
||||||
ROLE: Главный оркестратор взаимодействия с Ollama LLM (Qwen 2.5), разбор вызовов
|
ROLE: Нативный оркестратор диалога, диспетчер инструментов (Function Calling)
|
||||||
инструментов (Function Calling), интеграция с декларативным реестром
|
и управление сессионными стейтами через ContextManager.
|
||||||
действий SQLite (tool_action_registry), детерминированный Fast-Path
|
|
||||||
для подтверждений, динамические кнопки и автоочистка эфемерных сообщений.
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[AGENT_IMPORTS]: Системные и доменные импорты.
|
||||||
|
- ANCHOR[AGENT_MAIN_PIPELINE]: Основная точка входа process_chat_message.
|
||||||
|
- ANCHOR[AGENT_SYSTEM_PROMPT]: Формирование динамического контекста.
|
||||||
|
- ANCHOR[AGENT_TOOL_DISPATCHER]: Исполнение нативных вызовов инструментов.
|
||||||
|
- ANCHOR[AGENT_TOPIC_DRIFT]: Защита контекста и обработка свободных тем.
|
||||||
===============================================================================
|
===============================================================================
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# --- [SECTION 1: SYSTEM PATHS & IMPORTS] --- # ANCHOR[SYS_IMPORTS]
|
# ANCHOR[AGENT_IMPORTS]
|
||||||
import os
|
|
||||||
import sys
|
import sys
|
||||||
import json
|
import json
|
||||||
import urllib.request
|
|
||||||
import urllib.error
|
|
||||||
import logging
|
import logging
|
||||||
import re
|
|
||||||
from typing import List, Dict, Any, Tuple, Optional
|
from typing import List, Dict, Any, Tuple, Optional
|
||||||
|
|
||||||
# Импорт фасада базы данных
|
from .db.connection import get_db_connection
|
||||||
from .db_tools import (
|
from .db_tools import (
|
||||||
db_get_active_system_prompt,
|
db_get_active_system_prompt,
|
||||||
db_add_system_prompt,
|
db_apply_prompt_node_action,
|
||||||
db_get_tool_action,
|
db_get_tool_action,
|
||||||
db_get_tasks,
|
db_get_tasks,
|
||||||
db_update_task_status,
|
db_update_task_status,
|
||||||
db_delete_task,
|
db_delete_task,
|
||||||
db_add_task,
|
db_add_task,
|
||||||
|
db_tasks_edit,
|
||||||
|
db_export_tasks_markdown,
|
||||||
db_get_rules,
|
db_get_rules,
|
||||||
db_set_session_state,
|
db_set_session_state,
|
||||||
db_get_session_state,
|
db_get_session_state,
|
||||||
|
db_clear_session_state,
|
||||||
db_get_snapshots,
|
db_get_snapshots,
|
||||||
db_delete_snapshots,
|
db_delete_snapshots,
|
||||||
db_clear_session_state,
|
|
||||||
db_get_current_server_time,
|
db_get_current_server_time,
|
||||||
db_save_chat_message,
|
db_save_chat_message,
|
||||||
db_get_chat_history,
|
db_get_chat_history,
|
||||||
db_purge_ephemeral_messages,
|
db_purge_ephemeral_messages,
|
||||||
db_get_stats,
|
db_get_stats,
|
||||||
db_get_anomalies,
|
db_get_anomalies,
|
||||||
db_get_session_states,
|
|
||||||
db_get_reference
|
db_get_reference
|
||||||
)
|
)
|
||||||
|
|
||||||
from .schemas import TOOLS_SCHEMA
|
from .schemas import TOOLS_SCHEMA
|
||||||
from .core.calendar_utils import get_dynamic_calendar_context, parse_relative_date_ru
|
from .core.calendar_utils import get_dynamic_calendar_context
|
||||||
from .core.tool_injector import clean_raw_tool_tags, clean_output, inject_tools_if_needed
|
from .core.tool_injector import clean_raw_tool_tags, clean_output, inject_tools_if_needed
|
||||||
|
from .core.ollama_client import call_ollama_chat
|
||||||
|
from .core.fast_path import handle_fast_path_intercept
|
||||||
|
from .core.context_manager import (
|
||||||
|
save_tool_interaction,
|
||||||
|
save_dialog_interaction,
|
||||||
|
mark_last_user_message_ephemeral,
|
||||||
|
close_tool_session_and_cleanup
|
||||||
|
)
|
||||||
|
|
||||||
# --- [SECTION 2: LOGGING CONFIGURATION] --- # ANCHOR[LOGGING_CONFIG]
|
|
||||||
logger = logging.getLogger("SCUD_AGENT")
|
logger = logging.getLogger("SCUD_AGENT")
|
||||||
logger.setLevel(logging.INFO)
|
logger.setLevel(logging.INFO)
|
||||||
logger.propagate = False
|
logger.propagate = False
|
||||||
@@ -60,12 +69,8 @@ if not logger.handlers:
|
|||||||
handler.setFormatter(formatter)
|
handler.setFormatter(formatter)
|
||||||
logger.addHandler(handler)
|
logger.addHandler(handler)
|
||||||
|
|
||||||
OLLAMA_URL = "http://192.168.11.3:11434/api/chat"
|
|
||||||
TEXT_MODEL = "qwen2.5:14b"
|
|
||||||
VISION_MODEL = "qwen2.5vl:7b-q8_0"
|
|
||||||
|
|
||||||
|
# ANCHOR[AGENT_MAIN_PIPELINE]
|
||||||
# --- [SECTION 3: MAIN CHAT PROCESSING PIPELINE] --- # ANCHOR[CHAT_PROCESSOR]
|
|
||||||
def process_chat_message(
|
def process_chat_message(
|
||||||
user_id: int,
|
user_id: int,
|
||||||
user_message: str,
|
user_message: str,
|
||||||
@@ -75,327 +80,433 @@ def process_chat_message(
|
|||||||
session_id: str = "web_session_main"
|
session_id: str = "web_session_main"
|
||||||
) -> Tuple[str, List[Dict[str, Any]], Optional[Dict[str, Any]]]:
|
) -> Tuple[str, List[Dict[str, Any]], Optional[Dict[str, Any]]]:
|
||||||
"""
|
"""
|
||||||
Главный конвейер обработки входящего сообщения:
|
Главный конвейер обработки входящего сообщения чата на базе нативного Function Calling.
|
||||||
1. Fast-Path перехват подтверждений/отмен при активном session_state.
|
|
||||||
2. Перехват завершения работы ('нет, закончить настройку') с автоочисткой эфемерных сообщений.
|
|
||||||
3. Формирование системного контекста и вызов Ollama.
|
|
||||||
4. Выполнение вызванного Tool и опрос Data-Driven реестра действий.
|
|
||||||
5. Возврат кортежа: (reply_text, chat_history, action_metadata).
|
|
||||||
"""
|
"""
|
||||||
logger.info(f"Получено сообщение от user_id={user_id}, session_id={session_id}: {user_message}")
|
logger.info(f"Получено сообщение от user_id={user_id}, session_id={session_id}: {user_message}")
|
||||||
|
|
||||||
# 3.1. Обогащение текста вложением (при наличии)
|
|
||||||
full_user_content = user_message
|
|
||||||
if file_context:
|
|
||||||
full_user_content = f"{user_message}\n\n[СОДЕРЖИМОЕ ПРИКРЕПЛЕННОГО ФАЙЛА]:\n{file_context}"
|
|
||||||
|
|
||||||
session_state = db_get_session_state(session_id)
|
session_state = db_get_session_state(session_id)
|
||||||
user_msg_clean = user_message.lower().strip(" .!?:;")
|
if not session_state:
|
||||||
|
db_purge_ephemeral_messages(session_id)
|
||||||
|
|
||||||
# --- [FAST-PATH 1: ПЕРЕХВАТ ЗАВЕРШЕНИЯ НАСТРОЙКИ С ОЧИСТКОЙ ЭФЕМЕРНОЙ ПАМЯТИ] ---
|
full_user_content = f"{user_message}\n\n[СОДЕРЖИМОЕ ПРИКРЕПЛЕННОГО ФАЙЛА]:\n{file_context}" if file_context else user_message
|
||||||
if user_msg_clean in ["нет, спасибо", "нет, закончить настройку", "закончить настройку", "завершить", "нет"]:
|
|
||||||
db_clear_session_state(session_id)
|
|
||||||
deleted_count = db_purge_ephemeral_messages(session_id)
|
|
||||||
logger.info(f"🧹 Завершена работа с инструментом. Удалено эфемерных сообщений: {deleted_count}")
|
|
||||||
|
|
||||||
reply_text = "Хорошо. Настройка завершена, контекст диалога чист. Чем я могу помочь дальше?"
|
|
||||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=0)
|
|
||||||
return reply_text, db_get_chat_history(session_id), None
|
|
||||||
|
|
||||||
# --- [FAST-PATH 2: ПЕРЕХВАТ ПОДТВЕРЖДЕНИЯ / ОТМЕНЫ ПРЕВЬЮ ПРОМПТА] ---
|
# 1. Быстрый перехват строго системных кнопок UI (подтверждение превью промпта)
|
||||||
if session_state and session_state.get("state_type") == "PROMPT_PREVIEW":
|
fast_path_res = handle_fast_path_intercept(session_id, user_message, full_user_content, session_state)
|
||||||
if user_msg_clean in ["подтверждаю", "подтвердить", "да", "сохранить", "применить", "ок", "хорошо"]:
|
if fast_path_res:
|
||||||
db_add_system_prompt("main_agent", session_state.get("pending_data", ""))
|
return fast_path_res
|
||||||
db_clear_session_state(session_id)
|
|
||||||
|
|
||||||
action_cfg = db_get_tool_action("db_confirm_prompt_preview")
|
|
||||||
reply_text = action_cfg["success_template"] if action_cfg else "✅ Системный промпт успешно сохранен и применен в базе данных."
|
|
||||||
if action_cfg and action_cfg.get("follow_up_question"):
|
|
||||||
reply_text += f"\n\n{action_cfg['follow_up_question']}"
|
|
||||||
|
|
||||||
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=1)
|
|
||||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
|
||||||
return reply_text, db_get_chat_history(session_id), {
|
|
||||||
"type": action_cfg.get("action_type") if action_cfg else "FOLLOW_UP_ACTION",
|
|
||||||
"buttons": action_cfg.get("buttons", []) if action_cfg else []
|
|
||||||
}
|
|
||||||
|
|
||||||
elif user_msg_clean in ["отмена", "отменить", "отклонить", "назад", "стоп"]:
|
# 2. Фиксация сообщения пользователя
|
||||||
db_clear_session_state(session_id)
|
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=0)
|
||||||
|
|
||||||
action_cfg = db_get_tool_action("db_cancel_prompt_preview")
|
|
||||||
reply_text = action_cfg["success_template"] if action_cfg else "❌ Изменения системного промпта отменены."
|
|
||||||
if action_cfg and action_cfg.get("follow_up_question"):
|
|
||||||
reply_text += f"\n\n{action_cfg['follow_up_question']}"
|
|
||||||
|
|
||||||
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=1)
|
|
||||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
|
||||||
return reply_text, db_get_chat_history(session_id), {
|
|
||||||
"type": action_cfg.get("action_type") if action_cfg else "FOLLOW_UP_ACTION",
|
|
||||||
"buttons": action_cfg.get("buttons", []) if action_cfg else []
|
|
||||||
}
|
|
||||||
|
|
||||||
# 3.2. Сохраняем входящее сообщение в историю диалога
|
|
||||||
is_user_ephemeral = 1 if session_state else 0
|
|
||||||
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=is_user_ephemeral)
|
|
||||||
|
|
||||||
# 3.3. Извлекаем полную актуальную историю для передачи в LLM
|
|
||||||
db_history = db_get_chat_history(session_id, limit=20)
|
db_history = db_get_chat_history(session_id, limit=20)
|
||||||
|
|
||||||
dynamic_prompt_text = db_get_active_system_prompt()
|
# ANCHOR[AGENT_SYSTEM_PROMPT]
|
||||||
calendar_context = get_dynamic_calendar_context()
|
calendar_context = get_dynamic_calendar_context()
|
||||||
user_info = f"Пользователь ID={user_id}" if user_id != 0 else "Гость"
|
user_info = f"Пользователь ID={user_id}" if user_id != 0 else "Гость"
|
||||||
|
|
||||||
system_prompt_content = (
|
current_state_type = session_state.get("state_type") if session_state else None
|
||||||
f"Ты — интеллектуальный ассистент SCUD Orion AI. "
|
state_data = session_state.get("data_json") or {} if session_state else {}
|
||||||
f"Ты управляешь задачами, аналитикой СКУД и системными настройками с помощью инструментов (tools).\n\n"
|
if not isinstance(state_data, dict):
|
||||||
f"[ОКРУЖЕНИЕ]\n"
|
state_data = {}
|
||||||
f"- Пользователь: {user_info}\n"
|
idle_turns = state_data.get("idle_turns", 0)
|
||||||
f"- {calendar_context}\n\n"
|
|
||||||
f"[ПРАВИЛА ИСПОЛЬЗОВАНИЯ ИНСТРУМЕНТОВ]\n"
|
|
||||||
f"1. Для любых изменений системного промпта (добавить, удалить, изменить пункт) ВСЕГДА вызывай функцию db_preview_prompt_merge(prompt_text=...).\n"
|
|
||||||
f"2. Для просмотра системного промпта ВСЕГДА вызывай db_get_system_prompt().\n"
|
|
||||||
f"3. Для просмотра задач ВСЕГДА вызывай db_get_tasks().\n"
|
|
||||||
f"4. Никогда не симулируй выполнение функций в виде обычного текста. Если требуется действие — сразу вызывай соответствующий инструмент.\n\n"
|
|
||||||
f"[ПРИМЕРЫ ВЫЗОВА ИНСТРУМЕНТОВ]:\n"
|
|
||||||
f"- Пользователь: 'добавь пункт 3.4. Работать от сюда и до заката.' -> Вызов: db_preview_prompt_merge(prompt_text='3.4. Работать от сюда и до заката.')\n"
|
|
||||||
f"- Пользователь: 'удали пункт 3.4' -> Вызов: db_preview_prompt_merge(prompt_text='3.4')\n"
|
|
||||||
f"- Пользователь: 'покажи системный промпт' -> Вызов: db_get_system_prompt()\n"
|
|
||||||
f"- Пользователь: 'покажи мои задачи' -> Вызов: db_get_tasks()\n\n"
|
|
||||||
f"[ТЕКУЩИЙ АКТИВНЫЙ СИСТЕМНЫЙ ПРОМПТ]:\n{dynamic_prompt_text}"
|
|
||||||
)
|
|
||||||
|
|
||||||
llm_options = {
|
active_state_context = ""
|
||||||
"num_predict": 8192,
|
if current_state_type == "PROMPT_PREVIEW":
|
||||||
"num_ctx": 8192,
|
active_state_context = (
|
||||||
"temperature": 0.1,
|
"\n[ТЕКУЩИЙ РЕЖИМ: ПРЕДПРОСМОТР СИСТЕМНОГО ПРОМПТА]\n"
|
||||||
"repeat_penalty": 1.1,
|
"- Открыт предпросмотр изменений промпта. Для любых правок вызывай db_prompt_node_edit.\n"
|
||||||
"presence_penalty": 0.5,
|
)
|
||||||
"top_p": 0.9
|
elif current_state_type == "PROMPT_FOLLOWUP":
|
||||||
}
|
active_state_context = (
|
||||||
|
"\n[ТЕКУЩИЙ РЕЖИМ: СЕССИЯ РЕДАКТИРОВАНИЯ ПРОМПТА]\n"
|
||||||
|
"- Оператор просматривает или редактирует системный промпт.\n"
|
||||||
|
"- На любые команды вида 'удали пункт X.Y' или 'удали X.Y' ТЫ ОБЯЗАН ВЫЗВАТЬ db_prompt_node_edit с action='DELETE', section_id=X, item_id=Y.\n"
|
||||||
|
"- На любые команды 'добавь пункт X.Y ...' вызывай action='ADD'.\n"
|
||||||
|
"- Запрещено путать ADD и DELETE.\n"
|
||||||
|
)
|
||||||
|
elif current_state_type == "SNAPSHOTS_VIEW":
|
||||||
|
active_date = state_data.get("query_date", "выбранную дату")
|
||||||
|
active_state_context = (
|
||||||
|
f"\n[ТЕКУЩИЙ РЕЖИМ: ПРОСМОТР СНАПШОТОВ СКУД]\n"
|
||||||
|
f"- Сейчас на экране отображаются снапшоты за {active_date}.\n"
|
||||||
|
f"- Ты можешь форматировать, фильтровать или анализировать этот текущий срез.\n"
|
||||||
|
f"- Если оператор запрашивает ДРУГУЮ дату или день недели, отличную от {active_date} (например: 'за вчера', 'а за 13.08', 'покажи за сегодня') — "
|
||||||
|
f"ТЫ ОБЯЗАН СРАЗУ ВЫЗВАТЬ db_get_snapshots(date_str=...).\n"
|
||||||
|
f"- Запрещено генерировать текст за другую дату по памяти.\n"
|
||||||
|
)
|
||||||
|
elif current_state_type == "SNAPSHOT_DELETE_CONFIRM":
|
||||||
|
active_state_context = (
|
||||||
|
"\n[ВНИМАНИЕ: ОЖИДАЕТСЯ ПОДТВЕРЖДЕНИЕ УДАЛЕНИЯ СНАПШОТОВ]\n"
|
||||||
|
)
|
||||||
|
elif current_state_type == "TASK_DELETE_CONFIRM":
|
||||||
|
active_state_context = (
|
||||||
|
"\n[ВНИМАНИЕ: ОЖИДАЕТСЯ ПОДТВЕРЖДЕНИЕ УДАЛЕНИЯ ЗАДАЧИ]\n"
|
||||||
|
)
|
||||||
|
|
||||||
|
system_prompt_content = (
|
||||||
|
f"Ты — интеллектуальный ассистент SCUD Orion AI.\n"
|
||||||
|
f"Ты управляешь задачами, аналитикой СКУД и системными настройками исключительно через инструменты (tools).\n\n"
|
||||||
|
f"СТРОГИЕ ПРАВИЛА:\n"
|
||||||
|
f"1. К оператору всегда обращайся на Вы.\n"
|
||||||
|
f"2. Для получения данных всегда вызывай соответствующий инструмент:\n"
|
||||||
|
f" - Снапшоты и срезы логов СКУД за любые даты и дни недели -> db_get_snapshots\n"
|
||||||
|
f" - Удаление снапшотов -> db_delete_snapshots\n"
|
||||||
|
f" - Задачи и бэклог (просмотр, создание, смена статуса, удаление, экспорт) -> db_get_tasks, db_tasks_edit\n"
|
||||||
|
f" - Системный промпт -> db_get_system_prompt, db_prompt_node_edit\n"
|
||||||
|
f" - Аномалии СКУД/1С -> db_get_anomalies\n"
|
||||||
|
f" - База знаний -> db_get_rules\n"
|
||||||
|
f" - Статистика БД -> db_get_stats\n"
|
||||||
|
f" - Справка -> db_get_reference\n"
|
||||||
|
f"3. Запрещено сочинять данные от себя без вызова инструментов.\n\n"
|
||||||
|
f"[СИСТЕМНЫЙ КАЛЕНДАРЬ СЕРВЕРА]\n"
|
||||||
|
f"- Пользователь: {user_info}\n"
|
||||||
|
f"- {calendar_context}\n"
|
||||||
|
f"{active_state_context}"
|
||||||
|
)
|
||||||
|
|
||||||
user_msg_object = {"role": "user", "content": full_user_content}
|
user_msg_object = {"role": "user", "content": full_user_content}
|
||||||
|
|
||||||
# --- [SECTION 4: ROUTING & OLLAMA PAYLOAD] --- # ANCHOR[PAYLOAD_BUILD]
|
# ANCHOR[AGENT_TOOL_DISPATCHER]
|
||||||
if image_b64:
|
|
||||||
user_msg_object["images"] = [image_b64]
|
|
||||||
messages = [
|
|
||||||
{"role": "system", "content": "Ты — строгий модуль OCR. Перепиши весь текст с изображения буква в букву без отсебятины."},
|
|
||||||
user_msg_object
|
|
||||||
]
|
|
||||||
payload = {"model": VISION_MODEL, "messages": messages, "stream": False, "options": llm_options}
|
|
||||||
else:
|
|
||||||
clean_db_history = [dict(m) for m in db_history]
|
|
||||||
for m in clean_db_history:
|
|
||||||
m.pop("images", None)
|
|
||||||
messages = [{"role": "system", "content": system_prompt_content}] + clean_db_history + [user_msg_object]
|
|
||||||
payload = {"model": TEXT_MODEL, "messages": messages, "tools": TOOLS_SCHEMA, "stream": False, "options": llm_options}
|
|
||||||
|
|
||||||
# --- [SECTION 5: EXECUTION & TOOL ROUTING] --- # ANCHOR[TOOL_ROUTER]
|
|
||||||
try:
|
try:
|
||||||
req = urllib.request.Request(
|
if image_b64:
|
||||||
OLLAMA_URL,
|
user_msg_object["images"] = [image_b64]
|
||||||
data=json.dumps(payload).encode("utf-8"),
|
messages = [
|
||||||
headers={"Content-Type": "application/json"}
|
{"role": "system", "content": "Ты — строгий модуль OCR. Перепиши весь текст с изображения буква в букву."},
|
||||||
)
|
user_msg_object
|
||||||
with urllib.request.urlopen(req) as response:
|
]
|
||||||
res_data = json.loads(response.read().decode("utf-8"))
|
msg = call_ollama_chat(messages, is_vision=True)
|
||||||
msg = res_data.get("message", {})
|
else:
|
||||||
tool_calls = msg.get("tool_calls", [])
|
clean_db_history = [dict(m) for m in db_history]
|
||||||
raw_text_content = msg.get("content", "")
|
for m in clean_db_history:
|
||||||
|
m.pop("images", None)
|
||||||
|
messages = [{"role": "system", "content": system_prompt_content}] + clean_db_history + [user_msg_object]
|
||||||
|
msg = call_ollama_chat(messages, tools=TOOLS_SCHEMA, is_vision=False)
|
||||||
|
|
||||||
#tool_calls = inject_tools_if_needed(user_message, raw_text_content, tool_calls)
|
raw_text_reply = msg.get("content", "")
|
||||||
|
tool_calls = msg.get("tool_calls", [])
|
||||||
|
|
||||||
if tool_calls:
|
if not tool_calls:
|
||||||
logger.info(f"Ответ от Ollama получен. Tool calls: True (кол-во: {len(tool_calls)})")
|
tool_calls = inject_tools_if_needed(user_message, raw_text_reply, tool_calls)
|
||||||
messages.append(msg)
|
|
||||||
|
|
||||||
for tool in tool_calls:
|
|
||||||
fn_name = tool["function"]["name"]
|
|
||||||
fn_args = tool["function"].get("arguments", {})
|
|
||||||
logger.info(f"🚀 Вызов функции (Tool): {fn_name} с аргументами: {fn_args}")
|
|
||||||
tool_result_content = ""
|
|
||||||
|
|
||||||
action_cfg = db_get_tool_action(fn_name)
|
if tool_calls:
|
||||||
|
logger.info(f"Ответ от Ollama получен. Tool calls: True (кол-во: {len(tool_calls)})")
|
||||||
|
|
||||||
|
tool = tool_calls[0]
|
||||||
|
fn_name = tool["function"]["name"]
|
||||||
|
fn_args = tool["function"].get("arguments", {})
|
||||||
|
if isinstance(fn_args, str):
|
||||||
|
try:
|
||||||
|
fn_args = json.loads(fn_args)
|
||||||
|
except Exception:
|
||||||
|
fn_args = {}
|
||||||
|
|
||||||
|
logger.info(f"Вызов функции (Tool): {fn_name} с аргументами: {fn_args}")
|
||||||
|
|
||||||
|
# Ротация контекста: закрываем старую сессию инструмента
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason=f"ACTIVATE_{fn_name}")
|
||||||
|
session_state = None
|
||||||
|
mark_last_user_message_ephemeral(session_id)
|
||||||
|
|
||||||
|
# 1. Задачи (Просмотр)
|
||||||
|
if fn_name == "db_get_tasks":
|
||||||
|
raw_tasks = db_get_tasks(user_id, status=fn_args.get("status"))
|
||||||
|
reply_text = "Вот интерактивный список ваших текущих задач:"
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "TASK_INTERACTIVE_CARD",
|
||||||
|
"tasks": raw_tasks
|
||||||
|
}
|
||||||
|
|
||||||
|
# 2. Единый диспетчер задач
|
||||||
|
elif fn_name in ["db_tasks_edit", "db_add_task", "db_update_task_status", "db_delete_task"]:
|
||||||
|
action = fn_args.get("action", "UPDATE").upper()
|
||||||
|
if fn_name == "db_add_task": action = "ADD"
|
||||||
|
elif fn_name == "db_delete_task": action = "DELETE"
|
||||||
|
elif fn_name == "db_update_task_status": action = "UPDATE"
|
||||||
|
|
||||||
|
if action == "DELETE":
|
||||||
|
task_id_raw = str(fn_args.get("task_id", "")).replace("#", "").replace("TASK-", "").strip()
|
||||||
|
db_set_session_state(session_id, "TASK_DELETE_CONFIRM", {"task_id": task_id_raw, "idle_turns": 0})
|
||||||
|
reply_text = f"Вы действительно хотите удалить задачу #{task_id_raw}?"
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "TASK_DELETE_CONFIRM",
|
||||||
|
"buttons": [
|
||||||
|
{"label": f"Удалить #{task_id_raw}", "value": f"подтверждаю удаление задачи {task_id_raw}", "style": "danger"},
|
||||||
|
{"label": "Отмена", "value": "отмена", "style": "secondary"}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
elif action == "EXPORT":
|
||||||
|
export_res = db_export_tasks_markdown(
|
||||||
|
user_id=user_id,
|
||||||
|
filename=fn_args.get("filename"),
|
||||||
|
status_filter=fn_args.get("status")
|
||||||
|
)
|
||||||
|
reply_text = export_res.get("message", "Отчет по задачам успешно экспортирован.")
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=0)
|
||||||
|
|
||||||
if fn_name == "db_confirm_prompt_preview":
|
action_payload = None
|
||||||
if session_state and session_state.get("state_type") == "PROMPT_PREVIEW":
|
if export_res.get("status") == "success":
|
||||||
db_add_system_prompt("main_agent", session_state.get("pending_data", ""))
|
action_payload = {
|
||||||
db_clear_session_state(session_id)
|
"type": "FILE_DOWNLOAD_CARD",
|
||||||
|
"filename": export_res.get("filename"),
|
||||||
if action_cfg and action_cfg.get("bypass_llm"):
|
"download_url": export_res.get("download_url"),
|
||||||
reply_text = action_cfg["success_template"]
|
"tasks_count": export_res.get("tasks_count")
|
||||||
if action_cfg.get("follow_up_question"):
|
|
||||||
reply_text += f"\n\n{action_cfg['follow_up_question']}"
|
|
||||||
|
|
||||||
is_eph = action_cfg.get("is_ephemeral", 1)
|
|
||||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=is_eph)
|
|
||||||
return reply_text, db_get_chat_history(session_id), {
|
|
||||||
"type": action_cfg.get("action_type"),
|
|
||||||
"buttons": action_cfg.get("buttons", [])
|
|
||||||
}
|
|
||||||
tool_result_content = json.dumps({"status": "success"}, ensure_ascii=False)
|
|
||||||
else:
|
|
||||||
err_reply = "⚠️ Нет активного превью для подтверждения."
|
|
||||||
db_save_chat_message(session_id, "assistant", err_reply, is_ephemeral=1)
|
|
||||||
return err_reply, db_get_chat_history(session_id), None
|
|
||||||
|
|
||||||
elif fn_name == "db_cancel_prompt_preview":
|
|
||||||
db_clear_session_state(session_id)
|
|
||||||
if action_cfg and action_cfg.get("bypass_llm"):
|
|
||||||
reply_text = action_cfg["success_template"]
|
|
||||||
if action_cfg.get("follow_up_question"):
|
|
||||||
reply_text += f"\n\n{action_cfg['follow_up_question']}"
|
|
||||||
|
|
||||||
is_eph = action_cfg.get("is_ephemeral", 1)
|
|
||||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=is_eph)
|
|
||||||
return reply_text, db_get_chat_history(session_id), {
|
|
||||||
"type": action_cfg.get("action_type"),
|
|
||||||
"buttons": action_cfg.get("buttons", [])
|
|
||||||
}
|
|
||||||
tool_result_content = json.dumps({"status": "cancelled"}, ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name == "db_get_snapshots":
|
|
||||||
snapshots_res = db_get_snapshots(session_id=session_id, date_str=fn_args.get("date_str"), original_user_message=user_message)
|
|
||||||
tool_result_content = json.dumps(snapshots_res, ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name == "db_get_current_server_time":
|
|
||||||
tool_result_content = json.dumps(db_get_current_server_time(), ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name == "db_get_tasks":
|
|
||||||
tool_result_content = json.dumps(db_get_tasks(user_id), ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name in ["db_get_system_prompt", "db_get_system_prompts"]:
|
|
||||||
tool_result_content = json.dumps({"system_prompt": db_get_active_system_prompt()}, ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name == "db_get_stats":
|
|
||||||
tool_result_content = json.dumps(db_get_stats(), ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name == "db_get_anomalies":
|
|
||||||
tool_result_content = json.dumps(db_get_anomalies(limit=fn_args.get("limit", 100), date_str=fn_args.get("date_str")), ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name == "db_get_session_states":
|
|
||||||
tool_result_content = json.dumps(db_get_session_states(), ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name == "db_delete_snapshots":
|
|
||||||
res = db_delete_snapshots(snapshot_id=fn_args.get("snapshot_id"), day_str=fn_args.get("day_str"))
|
|
||||||
if action_cfg and action_cfg.get("bypass_llm"):
|
|
||||||
reply_text = action_cfg["success_template"]
|
|
||||||
if action_cfg.get("follow_up_question"):
|
|
||||||
reply_text += f"\n\n{action_cfg['follow_up_question']}"
|
|
||||||
is_eph = action_cfg.get("is_ephemeral", 1)
|
|
||||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=is_eph)
|
|
||||||
return reply_text, db_get_chat_history(session_id), {
|
|
||||||
"type": action_cfg.get("action_type"),
|
|
||||||
"buttons": action_cfg.get("buttons", [])
|
|
||||||
}
|
|
||||||
tool_result_content = json.dumps(res, ensure_ascii=False)
|
|
||||||
|
|
||||||
elif fn_name == "db_get_reference":
|
|
||||||
tool_result_content = json.dumps(db_get_reference(category=fn_args.get("category")), ensure_ascii=False)
|
|
||||||
|
|
||||||
# --- [SECTION 6: PROMPT MERGE & PREVIEW ENGINE] --- # ANCHOR[PROMPT_MERGE_LOGIC]
|
|
||||||
elif fn_name == "db_preview_prompt_merge":
|
|
||||||
proposed_text = fn_args.get("prompt_text") or fn_args.get("proposed_prompt") or ""
|
|
||||||
if isinstance(fn_args, str):
|
|
||||||
proposed_text = fn_args
|
|
||||||
|
|
||||||
current_prompt = db_get_active_system_prompt()
|
|
||||||
user_msg_lower = user_message.lower()
|
|
||||||
|
|
||||||
# 1. ОБРАБОТКА УДАЛЕНИЯ ПУНКТА
|
|
||||||
if any(w in user_msg_lower for w in ["удали", "стереть", "убрать", "вырежи", "удалить"]):
|
|
||||||
target_num_match = re.search(r'\d+(\.\d+)*', user_message)
|
|
||||||
target_num = target_num_match.group(0) if target_num_match else ""
|
|
||||||
|
|
||||||
lines = current_prompt.splitlines()
|
|
||||||
if target_num:
|
|
||||||
new_lines = [line for line in lines if not line.strip().startswith(f"{target_num}.")]
|
|
||||||
else:
|
|
||||||
new_lines = lines
|
|
||||||
proposed_text = "\n".join(new_lines)
|
|
||||||
|
|
||||||
# 2. ОБРАБОТКА ДОБАВЛЕНИЯ / ИЗМЕНЕНИЯ ПУНКТА
|
|
||||||
elif proposed_text:
|
|
||||||
if len(proposed_text) < 500:
|
|
||||||
clean_item = proposed_text.strip()
|
|
||||||
for prefix in ["добавь пункт", "добавить пункт", "вставь пункт", "добавь"]:
|
|
||||||
if prefix in clean_item.lower():
|
|
||||||
clean_item = re.sub(prefix, "", clean_item, flags=re.IGNORECASE).strip(" .:")
|
|
||||||
|
|
||||||
lines = current_prompt.splitlines()
|
|
||||||
new_lines = []
|
|
||||||
inserted = False
|
|
||||||
for line in lines:
|
|
||||||
new_lines.append(line)
|
|
||||||
if "3.3." in line and not inserted:
|
|
||||||
item_str = clean_item if re.match(r'^\d+\.\d+\.', clean_item) else f"3.4. {clean_item}"
|
|
||||||
new_lines.append(f" {item_str}")
|
|
||||||
inserted = True
|
|
||||||
if not inserted:
|
|
||||||
new_lines.append(f" {clean_item}")
|
|
||||||
proposed_text = "\n".join(new_lines)
|
|
||||||
|
|
||||||
db_set_session_state(session_id, "PROMPT_PREVIEW", proposed_text)
|
|
||||||
preview_reply = (
|
|
||||||
f"Ваше изменение успешно предпросмотрено. Полный обновленный системный промпт теперь выглядит так:\n\n"
|
|
||||||
f"{proposed_text}\n\n"
|
|
||||||
f"Для применения изменений подтвердите действие («подтверждаю») или отмените («отмена»)."
|
|
||||||
)
|
|
||||||
db_save_chat_message(session_id, "assistant", preview_reply, is_ephemeral=1)
|
|
||||||
return clean_raw_tool_tags(preview_reply), db_get_chat_history(session_id), {
|
|
||||||
"type": "PROMPT_PREVIEW",
|
|
||||||
"buttons": [
|
|
||||||
{"label": "Подтвердить", "value": "подтверждаю", "style": "primary"},
|
|
||||||
{"label": "Отменить", "value": "отмена", "style": "danger"}
|
|
||||||
]
|
|
||||||
}
|
}
|
||||||
|
return reply_text, db_get_chat_history(session_id), action_payload
|
||||||
|
|
||||||
elif fn_name == "db_get_rules":
|
else:
|
||||||
tool_result_content = json.dumps(db_get_rules(), ensure_ascii=False)
|
res = db_tasks_edit(
|
||||||
|
user_id=user_id,
|
||||||
|
action=action,
|
||||||
|
task_id=fn_args.get("task_id"),
|
||||||
|
title=fn_args.get("title"),
|
||||||
|
priority=fn_args.get("priority", "MEDIUM"),
|
||||||
|
status=fn_args.get("status"),
|
||||||
|
module=fn_args.get("module", "general"),
|
||||||
|
due_date=fn_args.get("due_date")
|
||||||
|
)
|
||||||
|
raw_tasks = db_get_tasks(user_id)
|
||||||
|
reply_text = res.get("message", "Операция над задачами выполнена.")
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "TASK_INTERACTIVE_CARD",
|
||||||
|
"tasks": raw_tasks
|
||||||
|
}
|
||||||
|
|
||||||
elif fn_name == "db_add_task":
|
# 3. Системный промпт
|
||||||
res = db_add_task(user_id=user_id, module=fn_args.get("module", "general"), title=fn_args.get("title"), priority=fn_args.get("priority", "MEDIUM"), due_date=fn_args.get("due_date"))
|
elif fn_name == "db_get_system_prompt":
|
||||||
tool_result_content = json.dumps(res, ensure_ascii=False)
|
active_prompt = db_get_active_system_prompt()
|
||||||
|
reply_text = f"Актуальный системный промпт:\n\n{active_prompt}"
|
||||||
|
db_set_session_state(session_id, "PROMPT_FOLLOWUP", {"idle_turns": 0})
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
elif fn_name == "db_update_task_status":
|
elif fn_name == "db_prompt_node_edit":
|
||||||
res = db_update_task_status(user_id=user_id, task_id=str(fn_args.get("task_id")), status=fn_args.get("status", "COMPLETED"), due_date=fn_args.get("due_date"))
|
action = str(fn_args.get("action", "ADD")).upper()
|
||||||
tool_result_content = json.dumps(res, ensure_ascii=False)
|
|
||||||
|
try:
|
||||||
|
sec_id = int(str(fn_args.get("section_id", 3)).strip())
|
||||||
|
except Exception:
|
||||||
|
sec_id = 3
|
||||||
|
|
||||||
elif fn_name == "db_delete_task":
|
try:
|
||||||
res = db_delete_task(user_id=user_id, task_id=str(fn_args.get("task_id", "")).upper())
|
itm_id = int(str(fn_args.get("item_id", 1)).strip())
|
||||||
if action_cfg and action_cfg.get("bypass_llm"):
|
except Exception:
|
||||||
reply_text = action_cfg["success_template"]
|
itm_id = 1
|
||||||
if action_cfg.get("follow_up_question"):
|
|
||||||
reply_text += f"\n\n{action_cfg['follow_up_question']}"
|
|
||||||
is_eph = action_cfg.get("is_ephemeral", 1)
|
|
||||||
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=is_eph)
|
|
||||||
return reply_text, db_get_chat_history(session_id), {
|
|
||||||
"type": action_cfg.get("action_type"),
|
|
||||||
"buttons": action_cfg.get("buttons", [])
|
|
||||||
}
|
|
||||||
tool_result_content = json.dumps(res, ensure_ascii=False)
|
|
||||||
|
|
||||||
messages.append({"role": "tool", "content": tool_result_content})
|
content = str(fn_args.get("content", "")).strip()
|
||||||
|
baseline_prompt = db_get_active_system_prompt()
|
||||||
|
|
||||||
# --- [SECTION 7: SECONDARY LLM PASS] --- # ANCHOR[SECONDARY_PASS]
|
with get_db_connection() as temp_conn:
|
||||||
second_payload = {"model": TEXT_MODEL, "messages": messages, "stream": False, "options": llm_options}
|
temp_cursor = temp_conn.cursor()
|
||||||
sec_req = urllib.request.Request(OLLAMA_URL, data=json.dumps(second_payload).encode("utf-8"), headers={"Content-Type": "application/json"})
|
temp_cursor.execute("""
|
||||||
with urllib.request.urlopen(sec_req) as sec_response:
|
SELECT section_id, item_id, content
|
||||||
sec_res_data = json.loads(sec_response.read().decode("utf-8"))
|
FROM system_prompt_nodes
|
||||||
raw_content = sec_res_data.get("message", {}).get("content", "").strip().replace("**", "")
|
WHERE prompt_name = 'main_agent' AND is_active = 1
|
||||||
final_content = clean_raw_tool_tags(clean_output(raw_content))
|
ORDER BY section_id, item_id
|
||||||
db_save_chat_message(session_id, "assistant", final_content, is_ephemeral=0)
|
""")
|
||||||
return final_content, db_get_chat_history(session_id), None
|
existing_nodes = temp_cursor.fetchall()
|
||||||
|
|
||||||
# Если вызовов функций не было
|
nodes_dict = {(sec, itm): txt for sec, itm, txt in existing_nodes}
|
||||||
raw_str = msg.get("content", "").strip().replace("**", "")
|
nodes_dict_for_draft = {k: v for k, v in nodes_dict.items() if k != (sec_id, itm_id)} if action == "DELETE" else dict(nodes_dict)
|
||||||
content_str = clean_raw_tool_tags(clean_output(raw_str))
|
if action != "DELETE":
|
||||||
final_reply = content_str or "Запрос обработан."
|
nodes_dict_for_draft[(sec_id, itm_id)] = content
|
||||||
db_save_chat_message(session_id, "assistant", final_reply, is_ephemeral=0)
|
|
||||||
return final_reply, db_get_chat_history(session_id), None
|
draft_lines = []
|
||||||
|
curr_sec = None
|
||||||
|
for (s_id, i_id), txt in sorted(nodes_dict_for_draft.items()):
|
||||||
|
if i_id == 0:
|
||||||
|
if curr_sec is not None: draft_lines.append("")
|
||||||
|
draft_lines.append(f"{s_id}. {txt}")
|
||||||
|
curr_sec = s_id
|
||||||
|
else:
|
||||||
|
draft_lines.append(f" {s_id}.{i_id}. {txt}")
|
||||||
|
merged_prompt = "\n".join(draft_lines)
|
||||||
|
|
||||||
|
diff_lines = []
|
||||||
|
curr_sec = None
|
||||||
|
display_nodes = dict(nodes_dict)
|
||||||
|
if action != "DELETE":
|
||||||
|
display_nodes[(sec_id, itm_id)] = content
|
||||||
|
|
||||||
|
for (s_id, i_id), txt in sorted(display_nodes.items()):
|
||||||
|
if i_id == 0:
|
||||||
|
if curr_sec is not None: diff_lines.append("")
|
||||||
|
diff_lines.append(f"{s_id}. {txt}")
|
||||||
|
curr_sec = s_id
|
||||||
|
else:
|
||||||
|
if s_id == sec_id and i_id == itm_id:
|
||||||
|
line_str = f' <span class="line-through text-rose-600 font-bold bg-rose-50 px-1.5 py-0.5 rounded border border-rose-300 opacity-80">{s_id}.{i_id}. {txt} [УДАЛЕНИЕ]</span>' if action == "DELETE" else f' <span class="text-rose-600 font-bold bg-rose-50 px-1.5 py-0.5 rounded border border-rose-300">{s_id}.{i_id}. {txt}</span>'
|
||||||
|
else:
|
||||||
|
line_str = f" {s_id}.{i_id}. {txt}"
|
||||||
|
diff_lines.append(line_str)
|
||||||
|
|
||||||
|
diff_html = "\n".join(diff_lines)
|
||||||
|
|
||||||
|
db_set_session_state(session_id, "PROMPT_PREVIEW", {
|
||||||
|
"draft_text": merged_prompt,
|
||||||
|
"action": action,
|
||||||
|
"section_id": sec_id,
|
||||||
|
"item_id": itm_id,
|
||||||
|
"content": content,
|
||||||
|
"idle_turns": 0
|
||||||
|
})
|
||||||
|
|
||||||
|
preview_reply = f"Предпросмотр изменений системного промпта:\n\n{diff_html}\n\nДля применения подтвердите действие, отредактируйте или отмените."
|
||||||
|
db_save_chat_message(session_id, "assistant", preview_reply, is_ephemeral=1)
|
||||||
|
return preview_reply, db_get_chat_history(session_id), {
|
||||||
|
"type": "PROMPT_PREVIEW",
|
||||||
|
"raw_draft": merged_prompt,
|
||||||
|
"baseline_prompt": baseline_prompt,
|
||||||
|
"buttons": [
|
||||||
|
{"label": "Подтвердить", "value": "подтверждаю", "style": "primary"},
|
||||||
|
{"label": "Отменить", "value": "отмена", "style": "danger"},
|
||||||
|
{"label": "✏️ Редактировать", "value": "action:open_editor", "style": "secondary"}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
# 4. Снапшоты СКУД
|
||||||
|
elif fn_name == "db_get_snapshots":
|
||||||
|
snapshots_res = db_get_snapshots(session_id=session_id, date_str=fn_args.get("date_str"), original_user_message=user_message)
|
||||||
|
query_date = snapshots_res.get("query_date", "выбранную дату")
|
||||||
|
reply_text = f"Реестр срезов СКУД за {query_date}:"
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "SNAPSHOTS_CARD",
|
||||||
|
"data": snapshots_res
|
||||||
|
}
|
||||||
|
|
||||||
|
elif fn_name == "db_delete_snapshots":
|
||||||
|
raw_id = fn_args.get("snapshot_id") or fn_args.get("day_str")
|
||||||
|
raw_ids = fn_args.get("snapshot_ids") or []
|
||||||
|
is_confirmed = fn_args.get("confirmed", False)
|
||||||
|
|
||||||
|
if raw_id and not raw_ids:
|
||||||
|
if isinstance(raw_id, str) and "," in raw_id:
|
||||||
|
raw_ids = [s.strip() for s in raw_id.split(",")]
|
||||||
|
else:
|
||||||
|
raw_ids = [raw_id]
|
||||||
|
|
||||||
|
safe_ids = [s.strip() for s in raw_ids if s and not str(s).strip().startswith("Y")]
|
||||||
|
if not safe_ids:
|
||||||
|
reply_text = "⚠️ Итоговый срез Y защищен от удаления. Выберите дневные снапшоты."
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
if not is_confirmed:
|
||||||
|
query_date = state_data.get("query_date", "")
|
||||||
|
db_set_session_state(session_id, "SNAPSHOT_DELETE_CONFIRM", {
|
||||||
|
"snapshot_ids": safe_ids,
|
||||||
|
"query_date": query_date,
|
||||||
|
"idle_turns": 0
|
||||||
|
})
|
||||||
|
ids_str = ", ".join(safe_ids)
|
||||||
|
reply_text = f"Вы действительно хотите удалить дневные снапшоты: {ids_str}?"
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "SNAPSHOT_DELETE_CONFIRM",
|
||||||
|
"buttons": [
|
||||||
|
{"label": f"Удалить ({len(safe_ids)} шт.)", "value": f"подтверждаю удаление снапшотов {ids_str}", "style": "danger"},
|
||||||
|
{"label": "Отмена", "value": "отмена", "style": "secondary"}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
else:
|
||||||
|
db_delete_snapshots(snapshot_ids=safe_ids)
|
||||||
|
query_date = state_data.get("query_date", "")
|
||||||
|
updated_snapshots_res = db_get_snapshots(session_id=session_id, date_str=query_date)
|
||||||
|
|
||||||
|
reply_text = f"✅ Успешно удалено снапшотов: {len(safe_ids)} шт."
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "SNAPSHOTS_CARD",
|
||||||
|
"data": updated_snapshots_res
|
||||||
|
}
|
||||||
|
|
||||||
|
elif fn_name == "db_delete_snapshots":
|
||||||
|
raw_id = fn_args.get("snapshot_id") or fn_args.get("day_str")
|
||||||
|
raw_ids = fn_args.get("snapshot_ids") or ([raw_id] if raw_id else [])
|
||||||
|
|
||||||
|
# Защита: итоговые Y-срезы удалять запрещено
|
||||||
|
safe_ids = [s for s in raw_ids if not str(s).startswith("Y")]
|
||||||
|
if not safe_ids:
|
||||||
|
reply_text = "⚠️ Итоговый срез Y защищен от удаления. Выберите дневные снапшоты."
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
# Прямое выполнение удаления в SQLite без лишних текстовых подтверждений
|
||||||
|
del_res = db_delete_snapshots(snapshot_ids=safe_ids)
|
||||||
|
|
||||||
|
# Получаем свежий реестр за ту же дату
|
||||||
|
query_date = fn_args.get("day_str", "")
|
||||||
|
updated_snapshots_res = db_get_snapshots(session_id=session_id, date_str=query_date)
|
||||||
|
|
||||||
|
reply_text = f"✅ Успешно удалено снапшотов: {len(safe_ids)} шт."
|
||||||
|
db_save_chat_message(session_id, "assistant", reply_text, is_ephemeral=1)
|
||||||
|
return reply_text, db_get_chat_history(session_id), {
|
||||||
|
"type": "SNAPSHOTS_CARD",
|
||||||
|
"data": updated_snapshots_res
|
||||||
|
}
|
||||||
|
|
||||||
|
# 5. Сервисные запросы
|
||||||
|
elif fn_name == "db_get_current_server_time":
|
||||||
|
tool_result_content = json.dumps(db_get_current_server_time(), ensure_ascii=False)
|
||||||
|
elif fn_name == "db_get_stats":
|
||||||
|
tool_result_content = json.dumps(db_get_stats(), ensure_ascii=False)
|
||||||
|
elif fn_name == "db_get_anomalies":
|
||||||
|
tool_result_content = json.dumps(db_get_anomalies(limit=fn_args.get("limit", 100), date_str=fn_args.get("date_str")), ensure_ascii=False)
|
||||||
|
elif fn_name == "db_get_rules":
|
||||||
|
tool_result_content = json.dumps(db_get_rules(), ensure_ascii=False)
|
||||||
|
elif fn_name == "db_get_reference":
|
||||||
|
tool_result_content = json.dumps(db_get_reference(category=fn_args.get("category")), ensure_ascii=False)
|
||||||
|
else:
|
||||||
|
tool_result_content = "{}"
|
||||||
|
|
||||||
|
messages.append(msg)
|
||||||
|
messages.append({"role": "tool", "content": tool_result_content})
|
||||||
|
sec_msg = call_ollama_chat(messages, is_vision=False)
|
||||||
|
raw_content = sec_msg.get("content", "").strip().replace("**", "").replace("*", "")
|
||||||
|
final_content = clean_raw_tool_tags(clean_output(raw_content)) or "Запрос выполнен."
|
||||||
|
|
||||||
|
for artifact in ["почемучка,", "почемучка!", "почемучка?", "почемучка", "почемучто", "почему-то"]:
|
||||||
|
if final_content.lower().startswith(artifact):
|
||||||
|
final_content = final_content[len(artifact):].lstrip(",.!?:; -")
|
||||||
|
|
||||||
|
is_ephem = 1 if fn_name in ["db_get_snapshots", "db_get_system_prompt", "db_prompt_node_edit"] else 0
|
||||||
|
db_save_chat_message(session_id, "assistant", final_content, is_ephemeral=is_ephem)
|
||||||
|
return final_content, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
# ANCHOR[AGENT_TOPIC_DRIFT]
|
||||||
|
raw_str = msg.get("content", "").strip().replace("**", "").replace("*", "")
|
||||||
|
final_reply = clean_raw_tool_tags(clean_output(raw_str)) or "Запрос обработан."
|
||||||
|
|
||||||
|
for artifact in ["почемучка,", "почемучка!", "почемучка?", "почемучка", "почемучто", "почему-то"]:
|
||||||
|
if final_reply.lower().startswith(artifact):
|
||||||
|
final_reply = final_reply[len(artifact):].lstrip(",.!?:; -")
|
||||||
|
|
||||||
|
action_payload = None
|
||||||
|
|
||||||
|
if session_state and session_state.get("state_type") in ["PROMPT_FOLLOWUP", "PROMPT_PREVIEW"]:
|
||||||
|
idle_turns += 1
|
||||||
|
if idle_turns > 3:
|
||||||
|
db_clear_session_state(session_id)
|
||||||
|
db_purge_ephemeral_messages(session_id)
|
||||||
|
action_payload = None
|
||||||
|
session_state = None
|
||||||
|
elif idle_turns == 3:
|
||||||
|
tool_action = db_get_tool_action("db_confirm_prompt_preview")
|
||||||
|
guard_question = tool_action.get("follow_up_question", "Желаете продолжить работу с системным промптом?") if tool_action else "Желаете продолжить работу с системным промптом?"
|
||||||
|
buttons = tool_action.get("buttons", []) if tool_action else []
|
||||||
|
final_reply += f"\n\n💡 *Напоминание:* {guard_question}"
|
||||||
|
action_payload = {"type": "PROMPT_FOLLOWUP", "buttons": buttons}
|
||||||
|
db_set_session_state(session_id, "PROMPT_FOLLOWUP", {"idle_turns": idle_turns})
|
||||||
|
else:
|
||||||
|
db_set_session_state(session_id, session_state.get("state_type"), {"idle_turns": idle_turns})
|
||||||
|
|
||||||
|
db_save_chat_message(session_id, "assistant", final_reply, is_ephemeral=0)
|
||||||
|
return final_reply, db_get_chat_history(session_id), action_payload
|
||||||
|
|
||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
logger.exception(f"Непредвиденная ошибка: {ex}")
|
logger.exception(f"Непредвиденная ошибка агента: {ex}")
|
||||||
error_reply = f"Внутренняя ошибка сервера: {ex}"
|
error_reply = f"Внутренняя ошибка сервера: {ex}"
|
||||||
return error_reply, db_get_chat_history(session_id), None
|
return error_reply, db_get_chat_history(session_id), None
|
||||||
@@ -0,0 +1,64 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/llm/core/context_manager.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: web_api / llm / core
|
||||||
|
ROLE: Интеллектуальный менеджер контекста: разделение служебных транзакций
|
||||||
|
инструментов и содержательного диалога с сохранением Topic Drift.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from typing import Dict, Any, Optional, List
|
||||||
|
|
||||||
|
from ..db.connection import get_db_connection
|
||||||
|
from ..db_tools import (
|
||||||
|
db_save_chat_message,
|
||||||
|
db_get_chat_history,
|
||||||
|
db_purge_ephemeral_messages,
|
||||||
|
db_clear_session_state,
|
||||||
|
db_set_session_state
|
||||||
|
)
|
||||||
|
|
||||||
|
logger = logging.getLogger("CONTEXT_MANAGER")
|
||||||
|
|
||||||
|
|
||||||
|
def save_tool_interaction(session_id: str, user_content: str, assistant_reply: str) -> None:
|
||||||
|
"""
|
||||||
|
Сохраняет синхронную служебную пару инструмента (и вопрос, и ответ = 1).
|
||||||
|
При очистке удалятся оба сообщения, не оставляя сирот.
|
||||||
|
"""
|
||||||
|
db_save_chat_message(session_id, "user", user_content, is_ephemeral=1)
|
||||||
|
db_save_chat_message(session_id, "assistant", assistant_reply, is_ephemeral=1)
|
||||||
|
|
||||||
|
|
||||||
|
def save_dialog_interaction(session_id: str, user_content: str, assistant_reply: str) -> None:
|
||||||
|
"""
|
||||||
|
Сохраняет содержательный диалог пользователя и ассистента (и вопрос, и ответ = 0).
|
||||||
|
Эти сообщения остаются в истории навсегда (включая Topic Drift).
|
||||||
|
"""
|
||||||
|
db_save_chat_message(session_id, "user", user_content, is_ephemeral=0)
|
||||||
|
db_save_chat_message(session_id, "assistant", assistant_reply, is_ephemeral=0)
|
||||||
|
|
||||||
|
|
||||||
|
def mark_last_user_message_ephemeral(session_id: str) -> None:
|
||||||
|
"""Помечает последнее сообщение пользователя как эфемерное при активации инструмента."""
|
||||||
|
with get_db_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
UPDATE chat_messages
|
||||||
|
SET is_ephemeral = 1
|
||||||
|
WHERE id = (SELECT MAX(id) FROM chat_messages WHERE session_id = ? AND role = 'user')
|
||||||
|
""", (session_id,))
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def close_tool_session_and_cleanup(session_id: str, close_reason: str = "COMPLETED") -> int:
|
||||||
|
"""
|
||||||
|
Закрывает сессию инструмента и удаляет ТОЛЬКО служебные карточки/команды.
|
||||||
|
Весь содержательный диалог сохраняется.
|
||||||
|
"""
|
||||||
|
db_clear_session_state(session_id)
|
||||||
|
deleted_count = db_purge_ephemeral_messages(session_id)
|
||||||
|
logger.info(f"[ContextManager] Сессия инструмента закрыта ({close_reason}). Очищено служебных сообщений: {deleted_count}")
|
||||||
|
return deleted_count
|
||||||
@@ -0,0 +1,168 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/llm/core/fast_path.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: web_api / llm / core
|
||||||
|
ROLE: Детерминированный мгновенный перехват нажатий кнопок подтверждения
|
||||||
|
(без задержек LLM и обращения к Ollama).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[FAST_PATH_MAIN]: Точка входа handle_fast_path_intercept.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
from typing import Dict, Any, Tuple, Optional
|
||||||
|
|
||||||
|
# Прямые вызовы чистых доменных сервисов
|
||||||
|
from services.prompts.service import save_full_prompt_draft, apply_prompt_action, get_active_system_prompt
|
||||||
|
from services.tasks.service import get_tasks, delete_task
|
||||||
|
from services.snapshots.service import get_snapshots_registry, delete_snapshots_safely
|
||||||
|
|
||||||
|
# Чат и сессии
|
||||||
|
from modules.web_api.llm.db.db_chat import db_save_chat_message, db_get_chat_history, db_purge_ephemeral_messages
|
||||||
|
from modules.web_api.llm.db.db_prompts import db_set_session_state, db_clear_session_state, db_get_tool_action
|
||||||
|
from modules.web_api.llm.core.context_manager import close_tool_session_and_cleanup
|
||||||
|
|
||||||
|
logger = logging.getLogger("FAST_PATH")
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[FAST_PATH_MAIN]
|
||||||
|
def handle_fast_path_intercept(
|
||||||
|
session_id: str,
|
||||||
|
user_message: str,
|
||||||
|
full_user_content: str,
|
||||||
|
session_state: Optional[Dict[str, Any]]
|
||||||
|
) -> Optional[Tuple[str, list, Optional[Dict[str, Any]]]]:
|
||||||
|
"""
|
||||||
|
Мгновенный перехват нажатий кнопок подтверждения (Fast-Path).
|
||||||
|
Возвращает (reply, history, action_type) или None, если требуется передать управление LLM.
|
||||||
|
"""
|
||||||
|
if not session_state:
|
||||||
|
return None
|
||||||
|
|
||||||
|
state_type = session_state.get("state_type")
|
||||||
|
state_data = session_state.get("data_json") or {}
|
||||||
|
if not isinstance(state_data, dict):
|
||||||
|
state_data = {}
|
||||||
|
|
||||||
|
msg_lower = user_message.strip().lower()
|
||||||
|
|
||||||
|
# -------------------------------------------------------------------------
|
||||||
|
# 1. ПОДТВЕРЖДЕНИЕ ПРЕВЬЮ СИСТЕМНОГО ПРОМПТА
|
||||||
|
# -------------------------------------------------------------------------
|
||||||
|
if state_type == "PROMPT_PREVIEW":
|
||||||
|
is_confirm = msg_lower in ["подтверждаю", "да", "сохраняй", "применить", "ок", "подтвердить"]
|
||||||
|
is_cancel = msg_lower in ["отмена", "отменить", "нет", "отклонить"]
|
||||||
|
|
||||||
|
if is_confirm:
|
||||||
|
action = state_data.get("action", "MANUAL_EDIT")
|
||||||
|
draft_text = state_data.get("draft_text", "")
|
||||||
|
|
||||||
|
# Применение изменений
|
||||||
|
if action == "MANUAL_EDIT" and draft_text:
|
||||||
|
save_full_prompt_draft(draft_text, prompt_name="main_agent")
|
||||||
|
else:
|
||||||
|
sec_id = state_data.get("section_id")
|
||||||
|
itm_id = state_data.get("item_id")
|
||||||
|
content = state_data.get("content", "")
|
||||||
|
if sec_id is not None and itm_id is not None:
|
||||||
|
apply_prompt_action(action=action, section_id=sec_id, item_id=itm_id, content=content)
|
||||||
|
elif draft_text:
|
||||||
|
save_full_prompt_draft(draft_text, prompt_name="main_agent")
|
||||||
|
|
||||||
|
# Закрытие сессии и зачистка
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason="PROMPT_APPLIED_SUCCESSFULLY")
|
||||||
|
db_clear_session_state(session_id)
|
||||||
|
|
||||||
|
tool_action = db_get_tool_action("db_confirm_prompt_preview")
|
||||||
|
reply = tool_action.get("success_template", "✅ Системный промпт успешно сохранен и применен в базе данных.") if tool_action else "✅ Системный промпт успешно сохранен и применен в базе данных."
|
||||||
|
|
||||||
|
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=0)
|
||||||
|
db_save_chat_message(session_id, "assistant", reply, is_ephemeral=0)
|
||||||
|
return reply, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
elif is_cancel:
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason="PROMPT_EDIT_CANCELLED")
|
||||||
|
db_clear_session_state(session_id)
|
||||||
|
|
||||||
|
tool_action = db_get_tool_action("db_cancel_prompt_preview")
|
||||||
|
reply = tool_action.get("success_template", "❌ Изменения системного промпта отменены.") if tool_action else "❌ Изменения системного промпта отменены."
|
||||||
|
|
||||||
|
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=0)
|
||||||
|
db_save_chat_message(session_id, "assistant", reply, is_ephemeral=0)
|
||||||
|
return reply, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
# -------------------------------------------------------------------------
|
||||||
|
# 2. ПОДТВЕРЖДЕНИЕ УДАЛЕНИЯ СНАПШОТОВ СКУД
|
||||||
|
# -------------------------------------------------------------------------
|
||||||
|
elif state_type == "SNAPSHOT_DELETE_CONFIRM":
|
||||||
|
is_confirm = "подтверждаю удаление снапшот" in msg_lower or msg_lower in ["подтверждаю", "да", "удалить"]
|
||||||
|
is_cancel = msg_lower in ["отмена", "отменить", "нет"]
|
||||||
|
|
||||||
|
if is_confirm:
|
||||||
|
snap_ids = state_data.get("snapshot_ids", [])
|
||||||
|
query_date = state_data.get("query_date", "")
|
||||||
|
|
||||||
|
# Безопасное удаление через сервис
|
||||||
|
del_res = delete_snapshots_safely(snapshot_ids=snap_ids)
|
||||||
|
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason="SNAPSHOTS_DELETED")
|
||||||
|
|
||||||
|
# Получаем свежий список за ту же дату
|
||||||
|
updated_data = get_snapshots_registry(date_str=query_date if query_date else None)
|
||||||
|
db_set_session_state(session_id, "SNAPSHOTS_VIEW", updated_data)
|
||||||
|
|
||||||
|
reply = f"✅ Успешно удалено снапшотов: {len(snap_ids)} шт."
|
||||||
|
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=1)
|
||||||
|
db_save_chat_message(session_id, "assistant", reply, is_ephemeral=1)
|
||||||
|
|
||||||
|
return reply, db_get_chat_history(session_id), {
|
||||||
|
"type": "SNAPSHOTS_CARD",
|
||||||
|
"data": updated_data
|
||||||
|
}
|
||||||
|
|
||||||
|
elif is_cancel:
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason="SNAPSHOT_DELETE_CANCELLED")
|
||||||
|
db_clear_session_state(session_id)
|
||||||
|
|
||||||
|
reply = "❌ Удаление снапшотов отменено."
|
||||||
|
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=0)
|
||||||
|
db_save_chat_message(session_id, "assistant", reply, is_ephemeral=0)
|
||||||
|
return reply, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
# -------------------------------------------------------------------------
|
||||||
|
# 3. ПОДТВЕРЖДЕНИЕ УДАЛЕНИЯ ЗАДАЧИ
|
||||||
|
# -------------------------------------------------------------------------
|
||||||
|
elif state_type == "TASK_DELETE_CONFIRM":
|
||||||
|
is_confirm = "подтверждаю удаление задачи" in msg_lower or msg_lower in ["подтверждаю", "да", "удалить"]
|
||||||
|
is_cancel = msg_lower in ["отмена", "отменить", "нет"]
|
||||||
|
|
||||||
|
if is_confirm:
|
||||||
|
task_id = state_data.get("task_id")
|
||||||
|
delete_task(user_id=1, task_id=str(task_id))
|
||||||
|
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason="TASK_DELETED")
|
||||||
|
db_clear_session_state(session_id)
|
||||||
|
|
||||||
|
raw_tasks = get_tasks(user_id=1)
|
||||||
|
reply = f"🗑 Задача #{task_id} удалена."
|
||||||
|
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=1)
|
||||||
|
db_save_chat_message(session_id, "assistant", reply, is_ephemeral=1)
|
||||||
|
|
||||||
|
return reply, db_get_chat_history(session_id), {
|
||||||
|
"type": "TASK_INTERACTIVE_CARD",
|
||||||
|
"tasks": raw_tasks
|
||||||
|
}
|
||||||
|
|
||||||
|
elif is_cancel:
|
||||||
|
close_tool_session_and_cleanup(session_id, close_reason="TASK_DELETE_CANCELLED")
|
||||||
|
db_clear_session_state(session_id)
|
||||||
|
|
||||||
|
reply = "❌ Удаление задачи отменено."
|
||||||
|
db_save_chat_message(session_id, "user", full_user_content, is_ephemeral=0)
|
||||||
|
db_save_chat_message(session_id, "assistant", reply, is_ephemeral=0)
|
||||||
|
return reply, db_get_chat_history(session_id), None
|
||||||
|
|
||||||
|
return None
|
||||||
@@ -0,0 +1,61 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/llm/core/ollama_client.py
|
||||||
|
ROLE: Транспортный клиент HTTP взаимодействия с локальным API Ollama.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
# ANCHOR[OLLAMA_CLIENT_IMPORTS]
|
||||||
|
import json
|
||||||
|
import urllib.request
|
||||||
|
import urllib.error
|
||||||
|
import logging
|
||||||
|
from typing import Dict, Any, Optional
|
||||||
|
|
||||||
|
logger = logging.getLogger("OLLAMA_CLIENT")
|
||||||
|
|
||||||
|
OLLAMA_URL = "http://10.121.17.227:11434/api/chat"
|
||||||
|
TEXT_MODEL = "qwen2.5:14b-instruct-q8_0"
|
||||||
|
VISION_MODEL = "qwen2.5vl:7b-q8_0"
|
||||||
|
|
||||||
|
LLM_OPTIONS = {
|
||||||
|
"num_predict": 8192,
|
||||||
|
"num_ctx": 8192,
|
||||||
|
"temperature": 0.0,
|
||||||
|
"repeat_penalty": 1.0,
|
||||||
|
"presence_penalty": 0.0,
|
||||||
|
"top_p": 0.9
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[OLLAMA_REQUEST_DISPATCHER]
|
||||||
|
def call_ollama_chat(
|
||||||
|
messages: list,
|
||||||
|
tools: Optional[list] = None,
|
||||||
|
is_vision: bool = False,
|
||||||
|
timeout: int = 120
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
Отправляет подготовленный массив сообщений в API Ollama.
|
||||||
|
Возвращает разобранный словарь сообщения ответа или генерирует исключение.
|
||||||
|
"""
|
||||||
|
model_name = VISION_MODEL if is_vision else TEXT_MODEL
|
||||||
|
payload = {
|
||||||
|
"model": model_name,
|
||||||
|
"messages": messages,
|
||||||
|
"stream": False,
|
||||||
|
"options": LLM_OPTIONS
|
||||||
|
}
|
||||||
|
|
||||||
|
if tools and not is_vision:
|
||||||
|
payload["tools"] = tools
|
||||||
|
|
||||||
|
req = urllib.request.Request(
|
||||||
|
OLLAMA_URL,
|
||||||
|
data=json.dumps(payload).encode("utf-8"),
|
||||||
|
headers={"Content-Type": "application/json"}
|
||||||
|
)
|
||||||
|
|
||||||
|
with urllib.request.urlopen(req, timeout=timeout) as response:
|
||||||
|
res_data = json.loads(response.read().decode("utf-8"))
|
||||||
|
return res_data.get("message", {})
|
||||||
@@ -1,39 +1,138 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/llm/core/tool_injector.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: web_api / llm / core
|
||||||
|
ROLE: Семантический анализ намерений оператора (Intent Classifier) и
|
||||||
|
детерминированная сборка вызовов инструментов при сбоях нативного Function Calling.
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[INTENT_INJECTOR_MAIN]: Точка входа inject_tools_if_needed.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
import re
|
import re
|
||||||
import json
|
|
||||||
import logging
|
import logging
|
||||||
from typing import List, Dict, Any
|
from typing import List, Dict, Any
|
||||||
|
|
||||||
logger = logging.getLogger("TOOL_INJECTOR")
|
logger = logging.getLogger("TOOL_INJECTOR")
|
||||||
|
|
||||||
|
|
||||||
def clean_raw_tool_tags(text: str) -> str:
|
def clean_raw_tool_tags(text: str) -> str:
|
||||||
|
"""Удаляет сырые теги вызова инструментов и системный шум."""
|
||||||
if not text:
|
if not text:
|
||||||
return ""
|
return ""
|
||||||
text = re.sub(r'\{"name":\s*"db_[^}]+\}\s*(</tool_call>)?', '', text)
|
cleaned = re.sub(r'<tool_call>.*?</tool_call>', '', text, flags=re.DOTALL)
|
||||||
text = re.sub(r'<tool_call>.*?</tool_call>', '', text, flags=re.DOTALL)
|
cleaned = re.sub(r'<\|.*?\|>', '', cleaned)
|
||||||
text = re.sub(r'</tool_call>\w*\[\]\(\)', '', text)
|
return cleaned.strip()
|
||||||
text = re.sub(r'</tool_call>', '', text)
|
|
||||||
return text.strip()
|
|
||||||
|
|
||||||
def clean_output(text: str) -> str:
|
def clean_output(text: str) -> str:
|
||||||
if not text:
|
"""Очищает маркеры форматирования."""
|
||||||
return text
|
return text.strip() if text else ""
|
||||||
artifacts = ["почемучто", "почто", "почему что"]
|
|
||||||
lower_text = text.lower()
|
|
||||||
for art in artifacts:
|
|
||||||
if lower_text.startswith(art):
|
|
||||||
text = text[len(art):].lstrip(",.!?:; -")
|
|
||||||
return text.strip()
|
|
||||||
|
|
||||||
def inject_tools_if_needed(user_message: str, raw_text_content: str, tool_calls: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
|
|
||||||
if tool_calls:
|
# ANCHOR[INTENT_INJECTOR_MAIN]
|
||||||
return tool_calls
|
def inject_tools_if_needed(user_message: str, raw_reply: str, existing_tool_calls: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
|
||||||
if '{"name":' in raw_text_content or '<tool_call>' in raw_text_content:
|
"""
|
||||||
try:
|
Гибридный семантический анализатор:
|
||||||
match = re.search(r'\{"name":\s*"([^"]+)",\s*"(?:params|arguments|properties)":\s*(\{.*?\})\}', raw_text_content)
|
Если модель ответила текстом с сомнениями или пропустила tool_call,
|
||||||
if match:
|
распознает доменное намерение и конструирует синтетический tool_call.
|
||||||
fn_name = match.group(1)
|
"""
|
||||||
fn_args = json.loads(match.group(2))
|
if existing_tool_calls:
|
||||||
return [{"function": {"name": fn_name, "arguments": fn_args}}]
|
return existing_tool_calls
|
||||||
except Exception as parse_err:
|
|
||||||
logger.debug(f"Ошибка парсинга сырого tool call: {parse_err}")
|
msg_clean = user_message.strip().lower()
|
||||||
return tool_calls
|
|
||||||
|
# 1. СЕМАНТИКА: Просмотр системного промпта
|
||||||
|
# Паттерны: "покажи системный промпт", "выведи промпт", "текущий промпт", "какой сейчас системный промпт"
|
||||||
|
if "промпт" in msg_clean and any(kw in msg_clean for kw in ["покажи", "выведи", "какой", "дай", "текст", "актуальн"]):
|
||||||
|
logger.info("[IntentInjector] Распознано намерение просмотра системного промпта")
|
||||||
|
return [{
|
||||||
|
"function": {
|
||||||
|
"name": "db_get_system_prompt",
|
||||||
|
"arguments": {}
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
|
||||||
|
# 2. СЕМАНТИКА: Удаление задач
|
||||||
|
# Паттерны: "удали задачу 37", "убери 37 задачу", "сотри таску #37", "сними с повестки задачу 37"
|
||||||
|
if any(kw in msg_clean for kw in ["удали", "удалить", "убери", "сотри", "сними"]) and any(kw in msg_clean for kw in ["задач", "таск"]):
|
||||||
|
task_match = re.search(r'#?\s*(\d+)', msg_clean)
|
||||||
|
if task_match:
|
||||||
|
task_id = task_match.group(1)
|
||||||
|
logger.info(f"[IntentInjector] Распознано намерение удаления задачи: #{task_id}")
|
||||||
|
return [{
|
||||||
|
"function": {
|
||||||
|
"name": "db_tasks_edit",
|
||||||
|
"arguments": {"action": "DELETE", "task_id": task_id}
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
|
||||||
|
# 3. СЕМАНТИКА: Управление системным промптом (удаление и мульти-удаление)
|
||||||
|
# Паттерны: "удали 1.8 и 3.4", "удали пункт 2.3", "вычеркни 1.8, 3.4 из промпта"
|
||||||
|
if any(kw in msg_clean for kw in ["удали", "удалить", "убери", "вычеркни", "сотри"]) and not any(kw in msg_clean for kw in ["задач", "снапшот", "срез"]):
|
||||||
|
node_matches = re.findall(r'(\d+)[\.\s]+(\d+)', user_message)
|
||||||
|
if node_matches:
|
||||||
|
formatted_nodes = [f"{s}.{i}" for s, i in node_matches]
|
||||||
|
logger.info(f"[IntentInjector] Распознано намерение удаления узлов промпта: {formatted_nodes}")
|
||||||
|
return [{
|
||||||
|
"function": {
|
||||||
|
"name": "db_prompt_node_edit",
|
||||||
|
"arguments": {
|
||||||
|
"action": "BATCH_DELETE" if len(formatted_nodes) > 1 else "DELETE",
|
||||||
|
"section_id": int(node_matches[0][0]),
|
||||||
|
"item_id": int(node_matches[0][1]),
|
||||||
|
"nodes_list": formatted_nodes,
|
||||||
|
"content": ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
|
||||||
|
# 4. СЕМАНТИКА: Добавление пункта промпта
|
||||||
|
# Паттерны: "добавь 3.4 Текст", "добавь пункт 3.4 Текст", "впиши в 3.4 Текст"
|
||||||
|
if any(kw in msg_clean for kw in ["добавь", "добавить", "впиши", "запиши"]) and not any(kw in msg_clean for kw in ["задач", "таск"]):
|
||||||
|
add_match = re.search(r'(\d+)[\.\s]+(\d+)[\.\s\:\-]+(.*)', user_message)
|
||||||
|
if add_match:
|
||||||
|
sec_id = int(add_match.group(1))
|
||||||
|
itm_id = int(add_match.group(2))
|
||||||
|
content = add_match.group(3).strip()
|
||||||
|
logger.info(f"[IntentInjector] Распознано намерение добавления узла промпта: {sec_id}.{itm_id}")
|
||||||
|
return [{
|
||||||
|
"function": {
|
||||||
|
"name": "db_prompt_node_edit",
|
||||||
|
"arguments": {
|
||||||
|
"action": "ADD",
|
||||||
|
"section_id": sec_id,
|
||||||
|
"item_id": itm_id,
|
||||||
|
"content": content
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
|
||||||
|
# 5. СЕМАНТИКА: Смена статуса задач
|
||||||
|
if any(kw in msg_clean for kw in ["в работу", "начни", "стартуй", "за работу"]):
|
||||||
|
task_match = re.search(r'#?\s*(\d+)', msg_clean)
|
||||||
|
if task_match:
|
||||||
|
task_id = task_match.group(1)
|
||||||
|
logger.info(f"[IntentInjector] Распознано намерение взятия в работу задачи: #{task_id}")
|
||||||
|
return [{
|
||||||
|
"function": {
|
||||||
|
"name": "db_tasks_edit",
|
||||||
|
"arguments": {"action": "UPDATE", "task_id": task_id, "status": "IN_PROGRESS"}
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
|
||||||
|
if any(kw in msg_clean for kw in ["заверши", "закрой", "готово", "выполнено"]):
|
||||||
|
task_match = re.search(r'#?\s*(\d+)', msg_clean)
|
||||||
|
if task_match:
|
||||||
|
task_id = task_match.group(1)
|
||||||
|
logger.info(f"[IntentInjector] Распознано намерение закрытия задачи: #{task_id}")
|
||||||
|
return [{
|
||||||
|
"function": {
|
||||||
|
"name": "db_tasks_edit",
|
||||||
|
"arguments": {"action": "UPDATE", "task_id": task_id, "status": "COMPLETED"}
|
||||||
|
}
|
||||||
|
}]
|
||||||
|
|
||||||
|
return existing_tool_calls
|
||||||
@@ -1,6 +1,13 @@
|
|||||||
"""
|
"""
|
||||||
|
===============================================================================
|
||||||
FILE: modules/web_api/llm/db/db_prompts.py
|
FILE: modules/web_api/llm/db/db_prompts.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: web_api / llm / db
|
||||||
|
ROLE: Реляционное управление системным промптом (таблица system_prompt_nodes),
|
||||||
|
базой знаний, реестром действий и сессионными стейтами.
|
||||||
|
===============================================================================
|
||||||
"""
|
"""
|
||||||
|
import re
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
from typing import List, Dict, Any, Optional
|
from typing import List, Dict, Any, Optional
|
||||||
@@ -8,16 +15,116 @@ from .connection import get_db_connection
|
|||||||
|
|
||||||
logger = logging.getLogger("DB_PROMPTS")
|
logger = logging.getLogger("DB_PROMPTS")
|
||||||
|
|
||||||
def db_get_active_system_prompt() -> str:
|
|
||||||
conn = get_db_connection()
|
def init_prompt_nodes_table():
|
||||||
cursor = conn.cursor()
|
"""Создает реляционную таблицу узлов промпта и заполняет базовыми данными."""
|
||||||
cursor.execute("SELECT prompt_text FROM system_prompts WHERE is_active = 1 ORDER BY id DESC LIMIT 1")
|
with get_db_connection() as conn:
|
||||||
row = cursor.fetchone()
|
cursor = conn.cursor()
|
||||||
conn.close()
|
cursor.execute("""
|
||||||
return row["prompt_text"] if row else "Ты — ИИ-ассистент SCUD Orion AI."
|
CREATE TABLE IF NOT EXISTS system_prompt_nodes (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
prompt_name TEXT DEFAULT 'main_agent',
|
||||||
|
section_id INTEGER NOT NULL,
|
||||||
|
item_id INTEGER NOT NULL,
|
||||||
|
content TEXT NOT NULL,
|
||||||
|
is_active INTEGER DEFAULT 1,
|
||||||
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
UNIQUE(prompt_name, section_id, item_id)
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
cursor.execute("CREATE INDEX IF NOT EXISTS idx_prompt_nodes ON system_prompt_nodes(prompt_name, section_id, item_id);")
|
||||||
|
|
||||||
|
cursor.execute("SELECT COUNT(*) FROM system_prompt_nodes WHERE prompt_name = 'main_agent'")
|
||||||
|
if cursor.fetchone()[0] == 0:
|
||||||
|
seed_nodes = [
|
||||||
|
# Раздел 1
|
||||||
|
(1, 0, "РОЛЬ И ЗАДАЧИ АССИСТЕНТА"),
|
||||||
|
(1, 1, "Управление бэклогом задач проекта (через db_get_tasks, db_add_task, db_update_task_status, db_delete_task)."),
|
||||||
|
(1, 2, "Консультация по правилам и арбитражу кадровых данных/СКУД из базы знаний (через db_get_rules)."),
|
||||||
|
(1, 3, "Предоставление справки о возможностях и примерах команд (СТРОГО через db_get_reference)."),
|
||||||
|
(1, 4, "Просмотр аномалий СКУД ⟷ 1С (СТРОГО через db_get_anomalies)."),
|
||||||
|
(1, 5, "Поддержка диалога с операторами и администраторами системы."),
|
||||||
|
(1, 7, "Работа с логами, снапшотами и срезами СКУД (через db_get_snapshots)."),
|
||||||
|
|
||||||
|
# Раздел 2
|
||||||
|
(2, 0, "ПРАВИЛА ВЫЗОВА ИНСТРУМЕНТОВ И ДАТ"),
|
||||||
|
(2, 1, "ОБЯЗАТЕЛЬНЫЙ ПРЕВЬЮ-МЕРДЖ: Категорически ЗАПРЕЩЕНО изменять промпт напрямую! При ЛЮБОМ запросе пользователя на изменение системного промпта Ты ОБЯЗАН вызвать инструмент db_prompt_node_edit."),
|
||||||
|
(2, 2, "БЕЗУСЛОВНОЕ ПОДТВЕРЖДЕНИЕ ПРЕВЬЮ: Инструмент db_confirm_prompt_preview вызывается СТРОГО после того, как пользователь напишет 'подтверждаю', 'да', 'сохраняй'."),
|
||||||
|
(2, 3, "ПОДТВЕРЖДЕНИЕ УДАЛЕНИЯ ЗАДАЧ: При запросе на удаление задач КАТЕГОРИЧЕСКИ ЗАПРЕЩЕНО сразу вызывать db_delete_task! Ты ОБЯЗАН сначала спросить пользователя подтверждение."),
|
||||||
|
(2, 4, "ИСПОЛЬЗОВАНИЕ КАЛЕНДАРЯ: При любых вопросах про даты БЕРИ ТОЧНУЮ ДАТУ ИЗ [СИСТЕМНОГО КАЛЕНДАРЯ СЕРВЕРА] В НАЧАЛЕ КОНТЕКСТА."),
|
||||||
|
(2, 5, "СТРОГИЙ ВЫЗОВ АНОМАЛИЙ: При запросах аномалий или нарушений Ты ОБЯЗАН вызвать инструмент db_get_anomalies."),
|
||||||
|
(2, 6, "СТРОГИЙ ВЫЗОВ СПРАВОЧНИКА: При запросах о возможностях, примерах запросов или командах Ты ОБЯЗАН вызывать db_get_reference."),
|
||||||
|
(2, 7, "Запрещено писать названия функций или код вызова текстом на экран."),
|
||||||
|
(2, 8, "СТРОГИЙ ВЫЗОВ СНАПШОТОВ: При ЛЮБЫХ запросах про снапшоты, срезы или логи СКУД за дату/день недели Ты ОБЯЗАН вызывать инструмент db_get_snapshots."),
|
||||||
|
|
||||||
|
# Раздел 3
|
||||||
|
(3, 0, "ПРАВИЛА СТИЛЯ"),
|
||||||
|
(3, 1, "Никогда не начинай ответ со склеек или слов-паразитов."),
|
||||||
|
(3, 2, "Отвечай в чистом текстовом формате (plain text) без спецсимволов или ###, если прямо не попросили."),
|
||||||
|
(3, 3, "Сохраняй инженерный, лаконичный и профессиональный стиль.")
|
||||||
|
]
|
||||||
|
cursor.executemany("""
|
||||||
|
INSERT OR IGNORE INTO system_prompt_nodes (prompt_name, section_id, item_id, content)
|
||||||
|
VALUES ('main_agent', ?, ?, ?)
|
||||||
|
""", seed_nodes)
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def db_get_active_system_prompt(prompt_name: str = "main_agent") -> str:
|
||||||
|
"""Собирает структурированный текст промпта из реляционной таблицы узлов."""
|
||||||
|
init_prompt_nodes_table()
|
||||||
|
with get_db_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT section_id, item_id, content
|
||||||
|
FROM system_prompt_nodes
|
||||||
|
WHERE prompt_name = ? AND is_active = 1
|
||||||
|
ORDER BY section_id ASC, item_id ASC
|
||||||
|
""", (prompt_name,))
|
||||||
|
rows = cursor.fetchall()
|
||||||
|
|
||||||
|
if not rows:
|
||||||
|
return "Ты — ИИ-ассистент SCUD Orion AI."
|
||||||
|
|
||||||
|
lines = []
|
||||||
|
current_section = None
|
||||||
|
|
||||||
|
for sec_id, itm_id, content in rows:
|
||||||
|
if itm_id == 0:
|
||||||
|
if current_section is not None:
|
||||||
|
lines.append("")
|
||||||
|
lines.append(f"{sec_id}. {content}")
|
||||||
|
current_section = sec_id
|
||||||
|
else:
|
||||||
|
lines.append(f" {sec_id}.{itm_id}. {content}")
|
||||||
|
|
||||||
|
return "\n".join(lines)
|
||||||
|
|
||||||
|
|
||||||
|
def db_apply_prompt_node_action(action: str, section_id: int, item_id: int, content: str = "", prompt_name: str = "main_agent"):
|
||||||
|
"""Прямое добавление, изменение или удаление узла в БД."""
|
||||||
|
init_prompt_nodes_table()
|
||||||
|
with get_db_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
action_clean = action.upper()
|
||||||
|
if action_clean in ["ADD", "UPDATE"]:
|
||||||
|
cursor.execute("""
|
||||||
|
INSERT INTO system_prompt_nodes (prompt_name, section_id, item_id, content, is_active, updated_at)
|
||||||
|
VALUES (?, ?, ?, ?, 1, CURRENT_TIMESTAMP)
|
||||||
|
ON CONFLICT(prompt_name, section_id, item_id) DO UPDATE SET
|
||||||
|
content = excluded.content,
|
||||||
|
is_active = 1,
|
||||||
|
updated_at = CURRENT_TIMESTAMP
|
||||||
|
""", (prompt_name, section_id, item_id, content))
|
||||||
|
elif action_clean == "DELETE":
|
||||||
|
cursor.execute("""
|
||||||
|
DELETE FROM system_prompt_nodes
|
||||||
|
WHERE prompt_name = ? AND section_id = ? AND item_id = ?
|
||||||
|
""", (prompt_name, section_id, item_id))
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
def db_get_tool_action(tool_name: str) -> Optional[Dict[str, Any]]:
|
def db_get_tool_action(tool_name: str) -> Optional[Dict[str, Any]]:
|
||||||
"""Получение шаблона, кнопок и настроек эфемерности инструмента из SQLite."""
|
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
cursor.execute("""
|
cursor.execute("""
|
||||||
@@ -35,28 +142,6 @@ def db_get_tool_action(tool_name: str) -> Optional[Dict[str, Any]]:
|
|||||||
return res
|
return res
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def db_add_system_prompt(name: str, prompt_text: str) -> Dict[str, Any]:
|
|
||||||
try:
|
|
||||||
with get_db_connection() as conn:
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("BEGIN IMMEDIATE;")
|
|
||||||
cursor.execute("SELECT id FROM system_prompts WHERE name = ?", (name,))
|
|
||||||
existing = cursor.fetchone()
|
|
||||||
if existing:
|
|
||||||
cursor.execute(
|
|
||||||
"UPDATE system_prompts SET prompt_text = ?, is_active = 1, updated_at = CURRENT_TIMESTAMP WHERE name = ?",
|
|
||||||
(prompt_text, name)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
cursor.execute(
|
|
||||||
"INSERT INTO system_prompts (name, prompt_text, is_active) VALUES (?, ?, 1)",
|
|
||||||
(name, prompt_text)
|
|
||||||
)
|
|
||||||
conn.commit()
|
|
||||||
return {"status": "success", "message": "Системный промпт успешно обновлен"}
|
|
||||||
except Exception as e:
|
|
||||||
logger.error(f"Ошибка при сохранении промпта в БД: {e}")
|
|
||||||
return {"status": "error", "error": str(e)}
|
|
||||||
|
|
||||||
def db_get_rules() -> List[Dict[str, Any]]:
|
def db_get_rules() -> List[Dict[str, Any]]:
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
@@ -66,9 +151,12 @@ def db_get_rules() -> List[Dict[str, Any]]:
|
|||||||
conn.close()
|
conn.close()
|
||||||
return [dict(r) for r in rows]
|
return [dict(r) for r in rows]
|
||||||
|
|
||||||
def db_set_session_state(session_id: str, state_type: str, data: str):
|
|
||||||
|
def db_set_session_state(session_id: str, state_type: str, data: Any) -> None:
|
||||||
|
"""Сохраняет состояние сессии в SQLite."""
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
|
payload_str = json.dumps(data, ensure_ascii=False) if isinstance(data, (dict, list)) else (str(data) if data is not None else "")
|
||||||
cursor.execute("""
|
cursor.execute("""
|
||||||
INSERT INTO session_states (session_id, state_type, pending_data, updated_at)
|
INSERT INTO session_states (session_id, state_type, pending_data, updated_at)
|
||||||
VALUES (?, ?, ?, CURRENT_TIMESTAMP)
|
VALUES (?, ?, ?, CURRENT_TIMESTAMP)
|
||||||
@@ -76,37 +164,40 @@ def db_set_session_state(session_id: str, state_type: str, data: str):
|
|||||||
state_type = excluded.state_type,
|
state_type = excluded.state_type,
|
||||||
pending_data = excluded.pending_data,
|
pending_data = excluded.pending_data,
|
||||||
updated_at = CURRENT_TIMESTAMP
|
updated_at = CURRENT_TIMESTAMP
|
||||||
""", (session_id, state_type, data))
|
""", (session_id, state_type, payload_str))
|
||||||
conn.commit()
|
conn.commit()
|
||||||
conn.close()
|
conn.close()
|
||||||
|
|
||||||
|
|
||||||
def db_get_session_state(session_id: str) -> Optional[Dict[str, Any]]:
|
def db_get_session_state(session_id: str) -> Optional[Dict[str, Any]]:
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
cursor.execute("SELECT state_type, pending_data FROM session_states WHERE session_id = ?", (session_id,))
|
cursor.execute("SELECT session_id, state_type, pending_data, updated_at FROM session_states WHERE session_id = ?", (session_id,))
|
||||||
row = cursor.fetchone()
|
row = cursor.fetchone()
|
||||||
conn.close()
|
conn.close()
|
||||||
return dict(row) if row else None
|
if not row:
|
||||||
|
return None
|
||||||
|
res = dict(row)
|
||||||
|
raw_data = res.get("pending_data") or ""
|
||||||
|
try:
|
||||||
|
res["data_json"] = json.loads(raw_data) if raw_data.strip().startswith(("{", "[")) else None
|
||||||
|
except Exception:
|
||||||
|
res["data_json"] = None
|
||||||
|
return res
|
||||||
|
|
||||||
def db_clear_session_state(session_id: str):
|
|
||||||
|
def db_clear_session_state(session_id: str) -> None:
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
cursor.execute("DELETE FROM session_states WHERE session_id = ?", (session_id,))
|
cursor.execute("DELETE FROM session_states WHERE session_id = ?", (session_id,))
|
||||||
conn.commit()
|
conn.commit()
|
||||||
conn.close()
|
conn.close()
|
||||||
|
|
||||||
def db_get_session_states() -> Dict[str, Any]:
|
|
||||||
conn = get_db_connection()
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("SELECT session_id, state_type, updated_at FROM session_states")
|
|
||||||
rows = cursor.fetchall()
|
|
||||||
conn.close()
|
|
||||||
return {"status": "success", "active_sessions": [dict(r) for r in rows]}
|
|
||||||
|
|
||||||
def db_get_stats() -> Dict[str, Any]:
|
def db_get_stats() -> Dict[str, Any]:
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
tables = ['scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history', 'ai_knowledge_base', 'system_prompts', 'session_states', 'tasks']
|
tables = ['scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history', 'ai_knowledge_base', 'system_prompt_nodes', 'session_states', 'tasks']
|
||||||
stats = {}
|
stats = {}
|
||||||
for t in tables:
|
for t in tables:
|
||||||
try:
|
try:
|
||||||
@@ -117,6 +208,7 @@ def db_get_stats() -> Dict[str, Any]:
|
|||||||
conn.close()
|
conn.close()
|
||||||
return {"status": "success", "tables_stats": stats}
|
return {"status": "success", "tables_stats": stats}
|
||||||
|
|
||||||
|
|
||||||
def db_get_anomalies(limit: int = 100, date_str: Optional[str] = None) -> Dict[str, Any]:
|
def db_get_anomalies(limit: int = 100, date_str: Optional[str] = None) -> Dict[str, Any]:
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
@@ -132,6 +224,7 @@ def db_get_anomalies(limit: int = 100, date_str: Optional[str] = None) -> Dict[s
|
|||||||
conn.close()
|
conn.close()
|
||||||
return {"status": "success", "count": len(rows), "anomalies": [dict(r) for r in rows]}
|
return {"status": "success", "count": len(rows), "anomalies": [dict(r) for r in rows]}
|
||||||
|
|
||||||
|
|
||||||
def db_get_reference(category: Optional[str] = None) -> Dict[str, Any]:
|
def db_get_reference(category: Optional[str] = None) -> Dict[str, Any]:
|
||||||
conn = get_db_connection()
|
conn = get_db_connection()
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
@@ -144,4 +237,49 @@ def db_get_reference(category: Optional[str] = None) -> Dict[str, Any]:
|
|||||||
cursor.execute(query, params)
|
cursor.execute(query, params)
|
||||||
rows = cursor.fetchall()
|
rows = cursor.fetchall()
|
||||||
conn.close()
|
conn.close()
|
||||||
return {"status": "success", "count": len(rows), "reference_items": [dict(r) for r in rows]}
|
return {"status": "success", "count": len(rows), "reference_items": [dict(r) for r in rows]}
|
||||||
|
|
||||||
|
def db_add_system_prompt(name: str, prompt_text: str) -> Dict[str, Any]:
|
||||||
|
"""Надежный парсер: восстанавливает разделы и пункты с авто-выравниванием отступов."""
|
||||||
|
try:
|
||||||
|
init_prompt_nodes_table()
|
||||||
|
with get_db_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM system_prompt_nodes WHERE prompt_name = ?", (name,))
|
||||||
|
|
||||||
|
current_sec = 1
|
||||||
|
current_itm = 0
|
||||||
|
|
||||||
|
for raw_line in prompt_text.splitlines():
|
||||||
|
clean_line = re.sub(r'<[^>]+>', '', raw_line).strip()
|
||||||
|
if not clean_line:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# 1. Проверяем подпункт (например: "1.8. Текст", "1.8 Текст", "1. 8 Текст")
|
||||||
|
sub_match = re.match(r'^(\d+)[\.\s]+(\d+)[\.\s\:\-]+(.*)$', clean_line)
|
||||||
|
# 2. Проверяем заголовок раздела (например: "1. РОЛЬ И ЗАДАЧИ", "1 РОЛЬ И ЗАДАЧИ")
|
||||||
|
sec_match = re.match(r'^(\d+)[\.\s\:\-]+(.*)$', clean_line)
|
||||||
|
|
||||||
|
if sub_match:
|
||||||
|
current_sec = int(sub_match.group(1))
|
||||||
|
current_itm = int(sub_match.group(2))
|
||||||
|
content = sub_match.group(3).strip()
|
||||||
|
elif sec_match and not any(c.islower() for c in sec_match.group(2)[:15]):
|
||||||
|
# Заголовок раздела (обычно капсом)
|
||||||
|
current_sec = int(sec_match.group(1))
|
||||||
|
current_itm = 0
|
||||||
|
content = sec_match.group(2).strip()
|
||||||
|
else:
|
||||||
|
current_itm += 1
|
||||||
|
content = clean_line
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
INSERT OR REPLACE INTO system_prompt_nodes (prompt_name, section_id, item_id, content, is_active)
|
||||||
|
VALUES (?, ?, ?, ?, 1)
|
||||||
|
""", (name, current_sec, current_itm, content))
|
||||||
|
|
||||||
|
conn.commit()
|
||||||
|
return {"status": "success", "message": "Системный промпт успешно сохранен по узлам"}
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Ошибка при разборе промпта в узлы: {e}")
|
||||||
|
return {"status": "error", "error": str(e)}
|
||||||
@@ -1,49 +0,0 @@
|
|||||||
"""
|
|
||||||
FILE: modules/web_api/llm/db/db_snapshots.py
|
|
||||||
"""
|
|
||||||
import json
|
|
||||||
from typing import Dict, Any, Optional
|
|
||||||
from .connection import get_db_connection
|
|
||||||
from .db_prompts import db_set_session_state
|
|
||||||
|
|
||||||
def db_get_snapshots(session_id: str = "web_session_main", date_str: Optional[str] = None, original_user_message: str = "") -> Dict[str, Any]:
|
|
||||||
conn = get_db_connection()
|
|
||||||
cursor = conn.cursor()
|
|
||||||
query = "SELECT snapshot_id, log_date, snapshot_time, COUNT(*) as record_count FROM scud_logs "
|
|
||||||
params = []
|
|
||||||
if date_str:
|
|
||||||
iso_date = date_str
|
|
||||||
if "." in date_str:
|
|
||||||
parts = date_str.split(".")
|
|
||||||
if len(parts) == 3:
|
|
||||||
iso_date = f"{parts[2]}-{parts[1]}-{parts[0]}"
|
|
||||||
query += " WHERE log_date = ? OR log_date = ? OR snapshot_time LIKE ? "
|
|
||||||
params.extend([date_str, iso_date, f"{iso_date}%"])
|
|
||||||
|
|
||||||
query += " GROUP BY snapshot_id ORDER BY id DESC LIMIT 50"
|
|
||||||
cursor.execute(query, params)
|
|
||||||
rows = cursor.fetchall()
|
|
||||||
snapshots = [dict(r) for r in rows]
|
|
||||||
|
|
||||||
result_data = {
|
|
||||||
"query_date": date_str or "все",
|
|
||||||
"snapshots_count": len(snapshots),
|
|
||||||
"snapshots": snapshots
|
|
||||||
}
|
|
||||||
db_set_session_state(session_id=session_id, state_type="SNAPSHOTS_VIEW", data=json.dumps(result_data, ensure_ascii=False))
|
|
||||||
conn.close()
|
|
||||||
return result_data
|
|
||||||
|
|
||||||
def db_delete_snapshots(snapshot_id: Optional[str] = None, day_str: Optional[str] = None) -> Dict[str, Any]:
|
|
||||||
if not snapshot_id and not day_str:
|
|
||||||
return {"status": "error", "message": "Необходимо указать snapshot_id или day_str (ДД.ММ.ГГГГ)."}
|
|
||||||
conn = get_db_connection()
|
|
||||||
cursor = conn.cursor()
|
|
||||||
if snapshot_id:
|
|
||||||
cursor.execute("DELETE FROM scud_logs WHERE snapshot_id = ?", (snapshot_id,))
|
|
||||||
else:
|
|
||||||
cursor.execute("DELETE FROM scud_logs WHERE log_date = ? OR snapshot_id LIKE ?", (day_str, f"%{day_str.replace('.', '')}%"))
|
|
||||||
deleted = cursor.rowcount
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
return {"status": "success", "message": f"Успешно удалено записей: {deleted}"}
|
|
||||||
@@ -1,82 +0,0 @@
|
|||||||
"""
|
|
||||||
FILE: modules/web_api/llm/db/db_tasks.py
|
|
||||||
"""
|
|
||||||
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()
|
|
||||||
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", "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} удалена"}
|
|
||||||
@@ -1,32 +1,96 @@
|
|||||||
"""
|
"""
|
||||||
|
===============================================================================
|
||||||
FILE: modules/web_api/llm/db_tools.py
|
FILE: modules/web_api/llm/db_tools.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: web_api / llm
|
||||||
|
ROLE: Фасадная точка доступа к доменным сервисам (Domain Facade).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[FACADE_EXPORTS]: Экспорт методов предметных сервисов для LLM и API.
|
||||||
|
===============================================================================
|
||||||
"""
|
"""
|
||||||
from datetime import datetime
|
|
||||||
from .db.connection import DB_PATH, get_db_connection
|
# ANCHOR[FACADE_EXPORTS]
|
||||||
from .db.db_chat import db_save_chat_message, db_get_chat_history, db_purge_ephemeral_messages
|
from typing import Optional, List, Dict, Any
|
||||||
from .db.db_tasks import normalize_task_id, db_get_tasks, db_add_task, db_update_task_status, db_delete_task
|
from core.connection import DB_PATH, get_connection as get_db_connection
|
||||||
from .db.db_snapshots import db_get_snapshots, db_delete_snapshots
|
|
||||||
|
# Домен: Задачи
|
||||||
|
from services.tasks.service import (
|
||||||
|
get_tasks as db_get_tasks,
|
||||||
|
add_task as db_add_task,
|
||||||
|
update_task_details as db_update_task_details,
|
||||||
|
delete_task as db_delete_task,
|
||||||
|
execute_task_action as db_tasks_edit
|
||||||
|
)
|
||||||
|
from services.tasks.exporter import export_tasks_to_markdown as db_export_tasks_markdown
|
||||||
|
from services.tasks.repository import normalize_task_id
|
||||||
|
|
||||||
|
# Домен: Системный промпт
|
||||||
|
from services.prompts.service import (
|
||||||
|
get_active_system_prompt as db_get_active_system_prompt,
|
||||||
|
apply_prompt_action as db_apply_prompt_node_action,
|
||||||
|
save_full_prompt_draft,
|
||||||
|
create_prompt_preview
|
||||||
|
)
|
||||||
|
|
||||||
|
# Домен: База знаний
|
||||||
|
from services.knowledge.service import (
|
||||||
|
get_rules as db_get_rules,
|
||||||
|
add_rule as db_add_rule
|
||||||
|
)
|
||||||
|
|
||||||
|
# Чат, сессии, статистика
|
||||||
|
from .db.db_chat import (
|
||||||
|
db_save_chat_message,
|
||||||
|
db_get_chat_history,
|
||||||
|
db_purge_ephemeral_messages,
|
||||||
|
db_clear_chat_history
|
||||||
|
)
|
||||||
from .db.db_prompts import (
|
from .db.db_prompts import (
|
||||||
db_get_active_system_prompt,
|
|
||||||
db_add_system_prompt,
|
|
||||||
db_get_tool_action,
|
db_get_tool_action,
|
||||||
db_get_rules,
|
|
||||||
db_set_session_state,
|
db_set_session_state,
|
||||||
db_get_session_state,
|
db_get_session_state,
|
||||||
db_clear_session_state,
|
db_clear_session_state,
|
||||||
db_get_session_states,
|
|
||||||
db_get_stats,
|
db_get_stats,
|
||||||
db_get_anomalies,
|
db_get_anomalies,
|
||||||
db_get_reference,
|
db_get_reference
|
||||||
db_get_tool_action
|
|
||||||
)
|
)
|
||||||
|
from .core.calendar_utils import get_dynamic_calendar_context as db_get_current_server_time
|
||||||
|
|
||||||
def db_get_current_server_time():
|
|
||||||
now = datetime.now()
|
def db_add_system_prompt(name_or_text: str, draft_text: str = None) -> None:
|
||||||
days_ru = ["понедельник", "вторник", "среда", "четверг", "пятница", "суббота", "воскресенье"]
|
"""Совместимая обертка для сохранения системного промпта."""
|
||||||
return {
|
if draft_text is not None:
|
||||||
"current_date": now.strftime("%d.%m.%Y"),
|
save_full_prompt_draft(draft_text, prompt_name=name_or_text)
|
||||||
"current_time": now.strftime("%H:%M:%S"),
|
else:
|
||||||
"day_of_week": days_ru[now.weekday()],
|
save_full_prompt_draft(name_or_text, prompt_name="main_agent")
|
||||||
"iso_date": now.strftime("%Y-%m-%d")
|
|
||||||
}
|
|
||||||
|
def db_get_snapshots(session_id: str = "web_session_main", date_str: str = None, original_user_message: str = "") -> Dict[str, Any]:
|
||||||
|
"""Совместимый фасад выборки снапшотов с сохранением стейта сессии."""
|
||||||
|
from services.snapshots.service import get_snapshots_registry
|
||||||
|
from .core.calendar_utils import parse_relative_date_ru
|
||||||
|
|
||||||
|
clean_date = (date_str or "").strip()
|
||||||
|
if not clean_date and original_user_message:
|
||||||
|
clean_date = parse_relative_date_ru(original_user_message)
|
||||||
|
|
||||||
|
res = get_snapshots_registry(date_str=clean_date if clean_date else None)
|
||||||
|
db_set_session_state(session_id=session_id, state_type="SNAPSHOTS_VIEW", data=res)
|
||||||
|
return res
|
||||||
|
|
||||||
|
|
||||||
|
def db_delete_snapshots(snapshot_id: str = None, snapshot_ids: List[str] = None, day_str: str = None) -> Dict[str, Any]:
|
||||||
|
"""Совместимый фасад безопасного удаления снапшотов."""
|
||||||
|
from services.snapshots.service import delete_snapshots_safely
|
||||||
|
|
||||||
|
target_ids = []
|
||||||
|
if snapshot_ids:
|
||||||
|
target_ids.extend(snapshot_ids)
|
||||||
|
if snapshot_id:
|
||||||
|
if isinstance(snapshot_id, str) and "," in snapshot_id:
|
||||||
|
target_ids.extend([s.strip() for s in snapshot_id.split(",")])
|
||||||
|
else:
|
||||||
|
target_ids.append(snapshot_id)
|
||||||
|
|
||||||
|
return delete_snapshots_safely(snapshot_ids=target_ids)
|
||||||
+194
-169
@@ -1,15 +1,194 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/llm/schemas.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: web_api / llm
|
||||||
|
ROLE: Декларативная схема нативных инструментов (Function Calling) для Ollama.
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[SCHEMA_PROMPT_EDIT]: Схема управления узлами системного промпта.
|
||||||
|
- ANCHOR[SCHEMA_TASKS_EDIT]: Консолидированная схема управления задачами.
|
||||||
|
- ANCHOR[SCHEMA_SNAPSHOTS]: Схема доступа к логам и срезам СКУД.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
# ANCHOR[SCHEMA_PROMPT_EDIT]
|
||||||
TOOLS_SCHEMA = [
|
TOOLS_SCHEMA = [
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "db_prompt_node_edit",
|
||||||
|
"description": (
|
||||||
|
"Управление элементами системного промпта (добавление, изменение, удаление пунктов).\n"
|
||||||
|
"Поддерживает как одиночные пункты (section_id, item_id), так и список пунктов для удаления (nodes_list=['1.8', '3.4']).\n"
|
||||||
|
"При любом запросе на удаление или добавление пунктов системного промпта ТЫ ОБЯЗАН вызвать этот инструмент."
|
||||||
|
),
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"action": {
|
||||||
|
"type": "string",
|
||||||
|
"enum": ["ADD", "EDIT", "DELETE", "BATCH_DELETE"],
|
||||||
|
"description": "Тип действия"
|
||||||
|
},
|
||||||
|
"section_id": {
|
||||||
|
"type": "integer",
|
||||||
|
"description": "Номер раздела (например 1)"
|
||||||
|
},
|
||||||
|
"item_id": {
|
||||||
|
"type": "integer",
|
||||||
|
"description": "Номер пункта (например 8)"
|
||||||
|
},
|
||||||
|
"nodes_list": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {"type": "string"},
|
||||||
|
"description": "Список пунктов для удаления/изменения, например ['1.8', '3.4']"
|
||||||
|
},
|
||||||
|
"content": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Текст пункта"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"required": ["action"]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "db_get_system_prompt",
|
||||||
|
"description": "Просмотр текущего активного системного промпта ассистента из базы данных.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
# ANCHOR[SCHEMA_TASKS_EDIT]
|
||||||
{
|
{
|
||||||
"type": "function",
|
"type": "function",
|
||||||
"function": {
|
"function": {
|
||||||
"name": "db_get_tasks",
|
"name": "db_get_tasks",
|
||||||
"description": "ПОЛУЧИТЬ СПИСОК ЗАДАЧ ТЕКУЩЕГО ПОЛЬЗОВАТЕЛЯ. Вызывай СРАЗУ при запросе 'покажи мои задачи' или 'список задач'. ВАЖНОЕ ПРАВИЛО ВЫВОДА: Выводи задачи ЕДИНЫМ плоским списком (нумерованным или маркированным) по порядку ID. КАТЕГОРИЧЕСКИ ЗАПРЕЩЕНО группировать задачи по статусам (В процессе, Бэклог, Завершены) или создавать подзаголовки, если оператор явно не попросил о группировке!",
|
"description": "Просмотр реестра задач и бэклога текущего пользователя.",
|
||||||
"parameters": {
|
"parameters": {
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"properties": {
|
"properties": {
|
||||||
"status": {
|
"status": {
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"description": "Фильтр статуса: BACKLOG, IN_PROGRESS или COMPLETED."
|
"enum": ["ALL", "IN_PROGRESS", "PLANNED", "COMPLETED"],
|
||||||
|
"description": "Опциональный фильтр статуса задач"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "db_tasks_edit",
|
||||||
|
"description": (
|
||||||
|
"Единый инструмент управления задачами: создание (ADD), изменение статуса/срока/названия (UPDATE), удаление (DELETE) и экспорт (EXPORT).\n"
|
||||||
|
"СТРОГИЕ ПРАВИЛА ВЫЗОВА:\n"
|
||||||
|
"- На любые фразы вида 'удали задачу N', 'удалить N', 'убери задачу N' ТЫ ОБЯЗАН СРАЗУ вызвать инструмент с action='DELETE', task_id='N'.\n"
|
||||||
|
"- На фразы 'возьми в работу N' вызывай action='UPDATE', task_id='N', status='IN_PROGRESS'.\n"
|
||||||
|
"- На фразы 'заверши N', 'готово N' вызывай action='UPDATE', task_id='N', status='COMPLETED'.\n"
|
||||||
|
"- КАТЕГОРИЧЕСКИ ЗАПРЕЩЕНО писать текстовые вопросы или запрашивать подтверждения словами! ТЫ ОБЯЗАН СРАЗУ вызвать инструмент."
|
||||||
|
),
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"action": {
|
||||||
|
"type": "string",
|
||||||
|
"enum": ["ADD", "UPDATE", "DELETE", "EXPORT"],
|
||||||
|
"description": "Тип действия: ADD, UPDATE, DELETE или EXPORT"
|
||||||
|
},
|
||||||
|
"task_id": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Номер задачи (например: '37')"
|
||||||
|
},
|
||||||
|
"title": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Описание или текст задачи"
|
||||||
|
},
|
||||||
|
"status": {
|
||||||
|
"type": "string",
|
||||||
|
"enum": ["IN_PROGRESS", "COMPLETED", "BACKLOG"],
|
||||||
|
"description": "Новый статус задачи: IN_PROGRESS (В работу), COMPLETED (Завершено), BACKLOG (В планы)"
|
||||||
|
},
|
||||||
|
"priority": {
|
||||||
|
"type": "string",
|
||||||
|
"enum": ["LOW", "MEDIUM", "HIGH", "CRITICAL"],
|
||||||
|
"description": "Приоритет задачи"
|
||||||
|
},
|
||||||
|
"module": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Модуль проекта"
|
||||||
|
},
|
||||||
|
"due_date": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Срок в формате ГГГГ-ММ-ДД"
|
||||||
|
},
|
||||||
|
"filename": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Имя файла для экспорта"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"required": ["action"]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
# ANCHOR[SCHEMA_SNAPSHOTS]
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "db_get_snapshots",
|
||||||
|
"description": "Получение списка снапшотов и срезов логов СКУД из базы данных за конкретную дату.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"date_str": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Дата в формате ДД.ММ.ГГГГ или относительное слово"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "db_delete_snapshots",
|
||||||
|
"description": "Удаление снапшотов СКУД по идентификатору или дате.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"snapshot_id": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Идентификатор конкретного снапшота"
|
||||||
|
},
|
||||||
|
"day_str": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Дата всех снапшотов за день"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "function",
|
||||||
|
"function": {
|
||||||
|
"name": "db_get_anomalies",
|
||||||
|
"description": "Просмотр истории аномалий и расхождений между СКУД и 1С.",
|
||||||
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"date_str": {
|
||||||
|
"type": "string",
|
||||||
|
"description": "Опциональная дата в формате ДД.ММ.ГГГГ"
|
||||||
|
},
|
||||||
|
"limit": {
|
||||||
|
"type": "integer",
|
||||||
|
"description": "Лимит записей"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -19,108 +198,10 @@ TOOLS_SCHEMA = [
|
|||||||
"type": "function",
|
"type": "function",
|
||||||
"function": {
|
"function": {
|
||||||
"name": "db_get_rules",
|
"name": "db_get_rules",
|
||||||
"description": "ПОЛУЧИТЬ БАЗУ ЗНАНИЙ ИИ И ПРАВИЛА АРБИТРАЖА (ai_knowledge_base). Вызывай когда пользователь просит показать базу знаний, правила, инструкции или промпты.",
|
"description": "Просмотр базы знаний и правил кадрового арбитража компании.",
|
||||||
"parameters": {"type": "object", "properties": {}}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_get_system_prompt",
|
|
||||||
"description": "ВЫЗЫВАЙ ВСЕГДА при наличии в сообщении фраз: 'покажи системный промпт', 'покажи промпт', 'выведи промпт', 'системный промпт'. Запрещено отвечать текстом без вызова этого инструмента.",
|
|
||||||
"parameters": {
|
"parameters": {
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"properties": {},
|
"properties": {}
|
||||||
"required": []
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_update_task_status",
|
|
||||||
"description": "Изменить статус и/или срок выполнения задачи в реестре.",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {
|
|
||||||
"task_id": {"type": "string", "description": "Идентификатор задачи, например TASK-17"},
|
|
||||||
"status": {"type": "string", "description": "Новый статус: COMPLETED, IN_PROGRESS или BACKLOG"},
|
|
||||||
"due_date": {"type": "string", "description": "Срок выполнения задачи"}
|
|
||||||
},
|
|
||||||
"required": ["task_id"]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_delete_task",
|
|
||||||
"description": "Удалить задачу из реестра по её task_id (например, TASK-18).",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {
|
|
||||||
"task_id": {"type": "string", "description": "Идентификатор задачи для удаления, например TASK-18"}
|
|
||||||
},
|
|
||||||
"required": ["task_id"]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_get_snapshots",
|
|
||||||
"description": "ПОЛУЧИТЬ АКТУАЛЬНЫЙ СПИСОК СНАПШОТОВ ИЗ БАЗЫ SQLITE. Вызывай ЭТУ ФУНКЦИЮ ВСЕГДА, даже если список снапшотов уже есть в истории чата или пользователь просит 'обновить', 'повторить запрос', 'проверить снова'. ЗАПРЕЩЕНО беречь контекст и выводить старые данные из истории!",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {
|
|
||||||
"date_str": {
|
|
||||||
"type": "string",
|
|
||||||
"description": "Точная дата в формате ДД.ММ.ГГГГ (например, '12.08.2026'), взятая из [СИСТЕМНОГО КАЛЕНДАРЯ СЕРВЕРА]."
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_delete_snapshots",
|
|
||||||
"description": "УДАЛИТЬ СНАПШОТ(Ы) ИЗ БАЗЫ ДАННЫХ. Вызывай, когда пользователь явно просит удалить конкретный снапшот по ID или все снапшоты за выбранный день.",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {
|
|
||||||
"snapshot_id": {"type": "string", "description": "Идентификатор снапшота для удаления, например Y20260805-007"},
|
|
||||||
"day_str": {"type": "string", "description": "Дата в формате ДД.ММ.ГГГГ для удаления всех снапшотов за день"}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_get_current_server_time",
|
|
||||||
"description": "ПОЛУЧИТЬ ТЕКУЩУЮ ДАТУ, ВРЕМЯ И ДЕНЬ НЕДЕЛИ СЕРВЕРА. Вызывай МГНОВЕННО при любых вопросах пользователя про точное текущее время или текущую дату.",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {},
|
|
||||||
"required": []
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_add_task",
|
|
||||||
"description": "Добавить новую задачу в бэклог проекта.",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {
|
|
||||||
"title": {"type": "string", "description": "Краткое описание задачи"},
|
|
||||||
"priority": {"type": "string", "description": "Приоритет: HIGH, MEDIUM, LOW"},
|
|
||||||
"module": {"type": "string", "description": "Модуль проекта, например general или services/scud_export"},
|
|
||||||
"due_date": {"type": "string", "description": "Срок выполнения задачи, например '2026-08-07 12:00'"}
|
|
||||||
},
|
|
||||||
"required": ["title"]
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
@@ -128,94 +209,38 @@ TOOLS_SCHEMA = [
|
|||||||
"type": "function",
|
"type": "function",
|
||||||
"function": {
|
"function": {
|
||||||
"name": "db_get_stats",
|
"name": "db_get_stats",
|
||||||
"description": "ПОЛУЧИТЬ ОБЩУЮ СТАТИСТИКУ БАЗЫ ДАННЫХ. Вызывай, когда пользователь просит показать общую статистику БД, количество записей в таблицах или размер базы.",
|
"description": "Получение статистики количества записей в таблицах базы данных.",
|
||||||
"parameters": {"type": "object", "properties": {}}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_get_anomalies",
|
|
||||||
"description": "ПОЛУЧИТЬ ИСТОРИЮ АНОМАЛИЙ СКУД ⟷ 1С. Вызывай при запросах на просмотр аномалий или расхождений. Передавай date_str если пользователь просит аномалии за конкретный день, или увеличенный limit (например 100) если просит все.",
|
|
||||||
"parameters": {
|
"parameters": {
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"properties": {
|
"properties": {}
|
||||||
"limit": {"type": "integer", "description": "Максимальное количество записей (по умолчанию 100)"},
|
|
||||||
"date_str": {"type": "string", "description": "Опциональная дата в формате ДД.ММ.ГГГГ"}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"type": "function",
|
"type": "function",
|
||||||
"function": {
|
"function": {
|
||||||
"name": "db_get_session_states",
|
"name": "db_get_current_server_time",
|
||||||
"description": "ПОЛУЧИТЬ АКТИВНЫЕ СЕССИИ И ПРЕВЬЮ (session_states). Вызывай, когда пользователь просит показать текущие сессии или статус превью.",
|
"description": "Получение текущего точного времени сервера.",
|
||||||
"parameters": {"type": "object", "properties": {}}
|
"parameters": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"type": "function",
|
"type": "function",
|
||||||
"function": {
|
"function": {
|
||||||
"name": "db_get_reference",
|
"name": "db_get_reference",
|
||||||
"description": "ПОЛУЧИТЬ СИСТЕМНЫЙ СПРАВОЧНИК И ПРИМЕРЫ КОМАНД ДЛЯ ОПЕРАТОРА (system_reference). Вызывай ВСЕГДА, когда пользователь спрашивает про возможности ассистента, список команд, примерах промптов или справе по работе с системой.",
|
"description": "Справка о возможностях ассистента и примеры доступных команд.",
|
||||||
"parameters": {
|
"parameters": {
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"properties": {
|
"properties": {
|
||||||
"category": {
|
"category": {
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"description": "Фильтр категории: scud, tasks, calendar или system. Если просят всё — не передавай параметр."
|
"description": "Категория справки"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
}
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_add_system_prompt",
|
|
||||||
"description": "Прямое сохранение системного промпта в БД без предварительного просмотра.",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {
|
|
||||||
"name": {"type": "string", "description": "Имя промпта, по умолчанию main_agent"},
|
|
||||||
"prompt_text": {"type": "string", "description": "Полный текст системного промпта"}
|
|
||||||
},
|
|
||||||
"required": ["prompt_text"]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_preview_prompt_merge",
|
|
||||||
"description": "Сформировать предпросмотр изменения системного промпта. Вызывается при любых запросах на добавление ('добавь пункт...'), изменение или удаление пунктов системного промпта ('удали пункт...').",
|
|
||||||
"parameters": {
|
|
||||||
"type": "object",
|
|
||||||
"properties": {
|
|
||||||
"prompt_text": {
|
|
||||||
"type": "string",
|
|
||||||
"description": "Полный текст добавляемого пункта или номер/описание удаляемого пункта."
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"required": ["prompt_text"]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_confirm_prompt_preview",
|
|
||||||
"description": "Подтвердить и сохранить текущее подготовленное превью в БД. Вызывай этот инструмент, когда пользователь говорит 'подтверждаю', 'да', 'вноси', 'применяй', 'сохраняй' или одобряет превью в любой форме.",
|
|
||||||
"parameters": {"type": "object", "properties": {}}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"type": "function",
|
|
||||||
"function": {
|
|
||||||
"name": "db_cancel_prompt_preview",
|
|
||||||
"description": "Отменить текущее превью системного промпта и сбросить изменения. Вызывай, когда пользователь явно отказывается от изменений.",
|
|
||||||
"parameters": {"type": "object", "properties": {}}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
]
|
]
|
||||||
+33
-250
@@ -2,68 +2,44 @@
|
|||||||
===============================================================================
|
===============================================================================
|
||||||
FILE: modules/web_api/main.py
|
FILE: modules/web_api/main.py
|
||||||
PROJECT: SCUD Orion AI (Unified Repository)
|
PROJECT: SCUD Orion AI (Unified Repository)
|
||||||
MODULE: web_api (FastAPI REST Server & Context Management)
|
MODULE: web_api (Main Application Entry Point)
|
||||||
ROLE: Главный шлюз веб-интерфейса, авторизация пользователей (JWT/Bcrypt),
|
ROLE: Инициализация FastAPI приложения, подключение роутеров и статики.
|
||||||
маршрутизация диалогов с LLM, OCR-парсинг файлов и управление задачами.
|
|
||||||
|
|
||||||
AI-CONTEXT-ANCHORS & INVARIANTS:
|
|
||||||
- ANCHOR[SYS_PATH]: Добавляет директорию модуля в sys.path для корректных импортов
|
|
||||||
независимо от рабочей директории запуска (root или web_api).
|
|
||||||
- ANCHOR[STATIC_MOUNT]: Рассчитывает абсолютный путь к папке static/ для надежного
|
|
||||||
рендеринга интерфейса и ассетов (css/js/favicon).
|
|
||||||
- ANCHOR[AUTH_JWT]: Изолирует персональные пространства задач по user_id (sub).
|
|
||||||
- ANCHOR[CHAT_PIPELINE]: Оркестрирует пайплайн парсинга вложений (file_parser) и
|
|
||||||
генерации ответов LLM (agent.process_chat_message).
|
|
||||||
|
|
||||||
DEPENDENCIES:
|
|
||||||
- modules/web_api/llm/agent.py (process_chat_message)
|
|
||||||
- modules/web_api/llm/db_tools.py (db_get_tasks, DB_PATH)
|
|
||||||
- modules/web_api/llm/file_parser.py (extract_text_from_file)
|
|
||||||
===============================================================================
|
===============================================================================
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# --- [SECTION 1: SYSTEM PATHS & IMPORTS] --- # ANCHOR[SYS_PATH]
|
# ANCHOR[APP_INIT_IMPORTS]
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import json
|
|
||||||
import sqlite3
|
|
||||||
import logging
|
import logging
|
||||||
import urllib.request
|
|
||||||
from datetime import datetime, timedelta
|
|
||||||
from typing import List, Dict, Any, Optional
|
|
||||||
|
|
||||||
# Гарантируем корректный импорт подмодулей web_api независимо от точки запуска
|
|
||||||
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
|
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||||
if CURRENT_DIR not in sys.path:
|
if CURRENT_DIR not in sys.path:
|
||||||
sys.path.insert(0, CURRENT_DIR)
|
sys.path.insert(0, CURRENT_DIR)
|
||||||
|
|
||||||
import jwt
|
ROOT_DIR = os.path.abspath(os.path.join(CURRENT_DIR, "../../"))
|
||||||
from passlib.context import CryptContext
|
|
||||||
from fastapi import FastAPI, Depends, HTTPException, status, UploadFile, File, Form
|
for p in [ROOT_DIR, CURRENT_DIR]:
|
||||||
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
|
if p not in sys.path:
|
||||||
|
sys.path.insert(0, p)
|
||||||
|
|
||||||
|
from fastapi import FastAPI, HTTPException
|
||||||
from fastapi.staticfiles import StaticFiles
|
from fastapi.staticfiles import StaticFiles
|
||||||
from fastapi.responses import FileResponse, JSONResponse
|
from fastapi.responses import FileResponse, JSONResponse
|
||||||
from fastapi.exceptions import RequestValidationError
|
from fastapi.exceptions import RequestValidationError
|
||||||
from pydantic import BaseModel
|
|
||||||
|
|
||||||
# Внутренние модули LLM и БД
|
from routers.auth import router as auth_router
|
||||||
from llm.agent import process_chat_message
|
from routers.admin import router as admin_router
|
||||||
from llm.db_tools import db_get_tasks, DB_PATH
|
from routers.tasks import router as tasks_router
|
||||||
from llm.file_parser import extract_text_from_file
|
from routers.chat import router as chat_router
|
||||||
|
from routers.files import router as files_router
|
||||||
|
|
||||||
# --- [SECTION 2: CONFIGURATION & SECURITY] --- # ANCHOR[AUTH_CONFIG]
|
# ANCHOR[APP_CONFIG]
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
level=logging.INFO,
|
level=logging.INFO,
|
||||||
format="%(asctime)s [%(levelname)s] %(message)s",
|
format="%(asctime)s [%(levelname)s] %(message)s",
|
||||||
handlers=[logging.StreamHandler()]
|
handlers=[logging.StreamHandler()]
|
||||||
)
|
)
|
||||||
|
|
||||||
JWT_SECRET = "scud_jwt_secret_key_2026_orion_ai_super_secure"
|
|
||||||
ALGORITHM = "HS256"
|
|
||||||
|
|
||||||
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
|
|
||||||
security = HTTPBearer()
|
|
||||||
|
|
||||||
STATIC_DIR = os.path.join(CURRENT_DIR, "static")
|
STATIC_DIR = os.path.join(CURRENT_DIR, "static")
|
||||||
|
|
||||||
app = FastAPI(title="SCUD Orion AI Context API", version="2.5")
|
app = FastAPI(title="SCUD Orion AI Context API", version="2.5")
|
||||||
@@ -79,57 +55,16 @@ async def validation_exception_handler(request, exc):
|
|||||||
content={"detail": exc.errors(), "body": str(exc)}
|
content={"detail": exc.errors(), "body": str(exc)}
|
||||||
)
|
)
|
||||||
|
|
||||||
# --- [SECTION 3: DATABASE & TOKEN HELPERS] --- # ANCHOR[DB_HELPERS]
|
# ANCHOR[ROUTER_REGISTRATION]
|
||||||
def get_db():
|
app.include_router(auth_router)
|
||||||
"""Создает безопасное соединение с SQLite БД модуля."""
|
app.include_router(admin_router)
|
||||||
conn = sqlite3.connect(DB_PATH)
|
app.include_router(tasks_router)
|
||||||
conn.row_factory = sqlite3.Row
|
app.include_router(chat_router)
|
||||||
return conn
|
app.include_router(files_router)
|
||||||
|
|
||||||
def create_access_token(user_id: int, username: str, is_admin: bool) -> str:
|
# ANCHOR[ROOT_STATIC_ROUTES]
|
||||||
payload = {
|
|
||||||
"sub": str(user_id),
|
|
||||||
"username": username,
|
|
||||||
"is_admin": is_admin,
|
|
||||||
"exp": datetime.utcnow() + timedelta(days=30)
|
|
||||||
}
|
|
||||||
return jwt.encode(payload, JWT_SECRET, algorithm=ALGORITHM)
|
|
||||||
|
|
||||||
def get_current_user(credentials: HTTPAuthorizationCredentials = Depends(security)) -> Dict[str, Any]:
|
|
||||||
try:
|
|
||||||
token = credentials.credentials
|
|
||||||
payload = jwt.decode(token, JWT_SECRET, algorithms=[ALGORITHM])
|
|
||||||
user_id = int(payload.get("sub"))
|
|
||||||
username = payload.get("username")
|
|
||||||
is_admin = bool(payload.get("is_admin", False))
|
|
||||||
return {"id": user_id, "username": username, "is_admin": is_admin}
|
|
||||||
except Exception as e:
|
|
||||||
logging.warning(f"Auth error: {e}")
|
|
||||||
raise HTTPException(
|
|
||||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
||||||
detail="Недействительный или просроченный токен авторизации",
|
|
||||||
headers={"WWW-Authenticate": "Bearer"},
|
|
||||||
)
|
|
||||||
|
|
||||||
# Pydantic-схемы валидации запросов
|
|
||||||
class AuthRequest(BaseModel):
|
|
||||||
username: str
|
|
||||||
password: str
|
|
||||||
|
|
||||||
class CreateUserRequest(BaseModel):
|
|
||||||
username: str
|
|
||||||
password: str
|
|
||||||
full_name: Optional[str] = None
|
|
||||||
is_admin: Optional[bool] = False
|
|
||||||
|
|
||||||
class ChangePasswordRequest(BaseModel):
|
|
||||||
old_password: str
|
|
||||||
new_password: str
|
|
||||||
|
|
||||||
# --- [SECTION 4: STATIC FILES & SPA ROUTES] --- # ANCHOR[STATIC_MOUNT]
|
|
||||||
@app.get("/")
|
@app.get("/")
|
||||||
def read_root():
|
def read_root():
|
||||||
"""Отдает главную страницу панели управления."""
|
|
||||||
index_path = os.path.join(STATIC_DIR, "index.html")
|
index_path = os.path.join(STATIC_DIR, "index.html")
|
||||||
if os.path.exists(index_path):
|
if os.path.exists(index_path):
|
||||||
return FileResponse(index_path)
|
return FileResponse(index_path)
|
||||||
@@ -142,175 +77,23 @@ async def favicon():
|
|||||||
return FileResponse(file_path)
|
return FileResponse(file_path)
|
||||||
raise HTTPException(status_code=404)
|
raise HTTPException(status_code=404)
|
||||||
|
|
||||||
# --- [SECTION 5: AUTHENTICATION & USER MANAGEMENT] --- # ANCHOR[AUTH_JWT]
|
|
||||||
@app.post("/api/v1/auth/login")
|
|
||||||
def login(req: AuthRequest):
|
|
||||||
username = req.username.strip().lower()
|
|
||||||
logging.info(f"===> Попытка входа для пользователя: {username}")
|
|
||||||
|
|
||||||
conn = get_db()
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("SELECT id, username, password_hash, is_admin FROM users WHERE username = ?", (username,))
|
|
||||||
user = cursor.fetchone()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
if not user or not pwd_context.verify(req.password, user["password_hash"]):
|
|
||||||
logging.warning(f"===> Ошибка: Неверный логин или пароль для {username}")
|
|
||||||
raise HTTPException(status_code=401, detail="Неверное имя пользователя или пароль")
|
|
||||||
|
|
||||||
is_admin = bool(user["is_admin"]) or (user["username"] == "puh")
|
|
||||||
token = create_access_token(user["id"], user["username"], is_admin)
|
|
||||||
logging.info(f"===> УСПЕХ: Авторизован пользователь {username}")
|
|
||||||
|
|
||||||
return {"status": "success", "token": token, "username": user["username"], "is_admin": is_admin}
|
|
||||||
|
|
||||||
@app.post("/api/v1/auth/change-password")
|
|
||||||
def change_password(req: ChangePasswordRequest, current_user: Dict[str, Any] = Depends(get_current_user)):
|
|
||||||
if not req.new_password or len(req.new_password) < 4:
|
|
||||||
raise HTTPException(status_code=400, detail="Новый пароль должен содержать минимум 4 символа")
|
|
||||||
|
|
||||||
conn = get_db()
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("SELECT password_hash FROM users WHERE id = ?", (current_user["id"],))
|
|
||||||
user = cursor.fetchone()
|
|
||||||
|
|
||||||
if not user or not pwd_context.verify(req.old_password, user["password_hash"]):
|
|
||||||
conn.close()
|
|
||||||
raise HTTPException(status_code=400, detail="Неверный старый пароль")
|
|
||||||
|
|
||||||
new_hash = pwd_context.hash(req.new_password)
|
|
||||||
cursor.execute("UPDATE users SET password_hash = ? WHERE id = ?", (new_hash, current_user["id"]))
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
logging.info(f"Пароль успешно изменен для пользователя ID: {current_user['id']}")
|
|
||||||
return {"status": "success", "message": "Пароль успешно изменен"}
|
|
||||||
|
|
||||||
@app.get("/api/v1/admin/users")
|
|
||||||
def list_users(current_user: Dict[str, Any] = Depends(get_current_user)):
|
|
||||||
if not current_user["is_admin"]:
|
|
||||||
raise HTTPException(status_code=403, detail="Доступ запрещен. Только для администратора.")
|
|
||||||
|
|
||||||
conn = get_db()
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("SELECT id, username, full_name, is_admin, created_at FROM users ORDER BY id ASC")
|
|
||||||
users = [dict(r) for r in cursor.fetchall()]
|
|
||||||
conn.close()
|
|
||||||
return users
|
|
||||||
|
|
||||||
@app.post("/api/v1/admin/users")
|
|
||||||
def create_user(req: CreateUserRequest, current_user: Dict[str, Any] = Depends(get_current_user)):
|
|
||||||
if not current_user["is_admin"]:
|
|
||||||
raise HTTPException(status_code=403, detail="Доступ запрещен. Только для администратора.")
|
|
||||||
|
|
||||||
username = req.username.strip().lower()
|
|
||||||
if not username or not req.password:
|
|
||||||
raise HTTPException(status_code=400, detail="Заполните имя пользователя и пароль")
|
|
||||||
|
|
||||||
conn = get_db()
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("SELECT id FROM users WHERE username = ?", (username,))
|
|
||||||
if cursor.fetchone():
|
|
||||||
conn.close()
|
|
||||||
raise HTTPException(status_code=400, detail="Пользователь с таким именем уже существует")
|
|
||||||
|
|
||||||
pwd_hash = pwd_context.hash(req.password)
|
|
||||||
full_name = req.full_name.strip() if req.full_name else None
|
|
||||||
is_admin = 1 if req.is_admin else 0
|
|
||||||
|
|
||||||
cursor.execute(
|
|
||||||
"INSERT INTO users (username, password_hash, full_name, is_admin) VALUES (?, ?, ?, ?)",
|
|
||||||
(username, pwd_hash, full_name, is_admin)
|
|
||||||
)
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
logging.info(f"Создан пользователь: {username} (admin={is_admin}) админом {current_user['username']}")
|
|
||||||
return {"status": "success", "message": f"Пользователь {username} создан"}
|
|
||||||
|
|
||||||
@app.delete("/api/v1/admin/users/{user_id}")
|
|
||||||
def delete_user(user_id: int, current_user: Dict[str, Any] = Depends(get_current_user)):
|
|
||||||
if not current_user["is_admin"]:
|
|
||||||
raise HTTPException(status_code=403, detail="Доступ запрещен. Только для администратора.")
|
|
||||||
|
|
||||||
if user_id == current_user["id"]:
|
|
||||||
raise HTTPException(status_code=400, detail="Нельзя удалить самого себя")
|
|
||||||
|
|
||||||
conn = get_db()
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute("DELETE FROM users WHERE id = ?", (user_id,))
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
logging.info(f"Удален пользователь ID: {user_id}")
|
|
||||||
return {"status": "success", "message": "Пользователь удален"}
|
|
||||||
|
|
||||||
# --- [SECTION 6: TASK TRACKER & LLM CHAT PIPELINE] --- # ANCHOR[CHAT_PIPELINE]
|
|
||||||
@app.get("/api/v1/tasks")
|
|
||||||
def get_tasks(user: Dict[str, Any] = Depends(get_current_user)):
|
|
||||||
"""Получить задачи текущего авторизованного пользователя."""
|
|
||||||
return db_get_tasks(user_id=user["id"])
|
|
||||||
|
|
||||||
# --- ЧАТ С АВТОРИЗАЦИЕЙ ---
|
|
||||||
@app.post("/api/v1/chat")
|
|
||||||
async def chat_endpoint(
|
|
||||||
session_id: str = Form("web_session_main"),
|
|
||||||
message: str = Form(""),
|
|
||||||
file: Optional[UploadFile] = File(default=None),
|
|
||||||
current_user: dict = Depends(get_current_user)
|
|
||||||
):
|
|
||||||
parsed_file = {"text": "", "image_b64": None}
|
|
||||||
if file and file.filename:
|
|
||||||
file_bytes = await file.read()
|
|
||||||
parsed_file = extract_text_from_file(file_bytes, file.filename)
|
|
||||||
|
|
||||||
reply, history, action_type = process_chat_message(
|
|
||||||
user_id=current_user["id"],
|
|
||||||
user_message=message,
|
|
||||||
file_context=parsed_file["text"],
|
|
||||||
image_b64=parsed_file["image_b64"],
|
|
||||||
session_id=session_id
|
|
||||||
)
|
|
||||||
return {"reply": reply, "history": history, "action_type": action_type}
|
|
||||||
|
|
||||||
|
|
||||||
# --- ГОСТЕВОЙ ЧАТ ---
|
|
||||||
@app.post("/api/v1/chat/guest")
|
|
||||||
async def guest_chat_endpoint(
|
|
||||||
session_id: str = Form("web_session_main"),
|
|
||||||
message: str = Form(""),
|
|
||||||
file: Optional[UploadFile] = File(default=None)
|
|
||||||
):
|
|
||||||
parsed_file = {"text": "", "image_b64": None}
|
|
||||||
if file and file.filename:
|
|
||||||
file_bytes = await file.read()
|
|
||||||
parsed_file = extract_text_from_file(file_bytes, file.filename)
|
|
||||||
|
|
||||||
reply, history, action_type = process_chat_message(
|
|
||||||
user_id=0,
|
|
||||||
user_message=message,
|
|
||||||
file_context=parsed_file["text"],
|
|
||||||
image_b64=parsed_file["image_b64"],
|
|
||||||
session_id=session_id
|
|
||||||
)
|
|
||||||
return {"reply": reply, "history": history, "action_type": action_type}
|
|
||||||
|
|
||||||
# --- [SECTION 7: STATIC FALLBACK ROUTER] --- # ANCHOR[STATIC_FALLBACK]
|
|
||||||
@app.get("/{file_path:path}")
|
@app.get("/{file_path:path}")
|
||||||
def serve_static_fallback(file_path: str):
|
def serve_static_fallback(file_path: str):
|
||||||
clean_path = file_path.lstrip("/")
|
clean_path = file_path.lstrip("/")
|
||||||
|
|
||||||
target = os.path.join(STATIC_DIR, clean_path)
|
target = os.path.join(STATIC_DIR, clean_path)
|
||||||
|
|
||||||
if os.path.isfile(target):
|
if os.path.isfile(target):
|
||||||
|
if clean_path.endswith(".js"):
|
||||||
|
return FileResponse(target, media_type="application/javascript")
|
||||||
|
elif clean_path.endswith(".css"):
|
||||||
|
return FileResponse(target, media_type="text/css")
|
||||||
return FileResponse(target)
|
return FileResponse(target)
|
||||||
|
|
||||||
filename = os.path.basename(clean_path)
|
filename = os.path.basename(clean_path)
|
||||||
target_js = os.path.join(STATIC_DIR, "js", filename)
|
for root, _, files in os.walk(STATIC_DIR):
|
||||||
if filename.endswith(".js") and os.path.isfile(target_js):
|
if filename in files:
|
||||||
return FileResponse(target_js, media_type="application/javascript")
|
full_path = os.path.join(root, filename)
|
||||||
|
media = "application/javascript" if filename.endswith(".js") else "text/css"
|
||||||
target_css = os.path.join(STATIC_DIR, "css", filename)
|
return FileResponse(full_path, media_type=media)
|
||||||
if filename.endswith(".css") and os.path.isfile(target_css):
|
|
||||||
return FileResponse(target_css, media_type="text/css")
|
|
||||||
|
|
||||||
raise HTTPException(status_code=404, detail="File not found")
|
raise HTTPException(status_code=404, detail="File not found")
|
||||||
@@ -0,0 +1,83 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/routers/admin.py
|
||||||
|
ROLE: Администрирование пользователей и прав доступа.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
# ANCHOR[ADMIN_ROUTER_IMPORTS]
|
||||||
|
import logging
|
||||||
|
from typing import Dict, Any, Optional
|
||||||
|
from fastapi import APIRouter, Depends, HTTPException
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
from .auth import get_current_user, get_db, pwd_context
|
||||||
|
|
||||||
|
router = APIRouter(prefix="/api/v1/admin", tags=["admin"])
|
||||||
|
|
||||||
|
# ANCHOR[ADMIN_SCHEMAS]
|
||||||
|
class CreateUserRequest(BaseModel):
|
||||||
|
username: str
|
||||||
|
password: str
|
||||||
|
full_name: Optional[str] = None
|
||||||
|
is_admin: Optional[bool] = False
|
||||||
|
|
||||||
|
# ANCHOR[ADMIN_ENDPOINTS]
|
||||||
|
@router.get("/users")
|
||||||
|
def list_users(current_user: Dict[str, Any] = Depends(get_current_user)):
|
||||||
|
if not current_user["is_admin"]:
|
||||||
|
raise HTTPException(status_code=403, detail="Доступ запрещен. Только для администратора.")
|
||||||
|
|
||||||
|
conn = get_db()
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT id, username, full_name, is_admin, created_at FROM users ORDER BY id ASC")
|
||||||
|
users = [dict(r) for r in cursor.fetchall()]
|
||||||
|
conn.close()
|
||||||
|
return users
|
||||||
|
|
||||||
|
@router.post("/users")
|
||||||
|
def create_user(req: CreateUserRequest, current_user: Dict[str, Any] = Depends(get_current_user)):
|
||||||
|
if not current_user["is_admin"]:
|
||||||
|
raise HTTPException(status_code=403, detail="Доступ запрещен. Только для администратора.")
|
||||||
|
|
||||||
|
username = req.username.strip().lower()
|
||||||
|
if not username or not req.password:
|
||||||
|
raise HTTPException(status_code=400, detail="Заполните имя пользователя и пароль")
|
||||||
|
|
||||||
|
conn = get_db()
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT id FROM users WHERE username = ?", (username,))
|
||||||
|
if cursor.fetchone():
|
||||||
|
conn.close()
|
||||||
|
raise HTTPException(status_code=400, detail="Пользователь с таким именем уже существует")
|
||||||
|
|
||||||
|
pwd_hash = pwd_context.hash(req.password)
|
||||||
|
full_name = req.full_name.strip() if req.full_name else None
|
||||||
|
is_admin = 1 if req.is_admin else 0
|
||||||
|
|
||||||
|
cursor.execute(
|
||||||
|
"INSERT INTO users (username, password_hash, full_name, is_admin) VALUES (?, ?, ?, ?)",
|
||||||
|
(username, pwd_hash, full_name, is_admin)
|
||||||
|
)
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
logging.info(f"Создан пользователь: {username} (admin={is_admin}) админом {current_user['username']}")
|
||||||
|
return {"status": "success", "message": f"Пользователь {username} создан"}
|
||||||
|
|
||||||
|
@router.delete("/users/{user_id}")
|
||||||
|
def delete_user(user_id: int, current_user: Dict[str, Any] = Depends(get_current_user)):
|
||||||
|
if not current_user["is_admin"]:
|
||||||
|
raise HTTPException(status_code=403, detail="Доступ запрещен. Только для администратора.")
|
||||||
|
|
||||||
|
if user_id == current_user["id"]:
|
||||||
|
raise HTTPException(status_code=400, detail="Нельзя удалить самого себя")
|
||||||
|
|
||||||
|
conn = get_db()
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM users WHERE id = ?", (user_id,))
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
logging.info(f"Удален пользователь ID: {user_id}")
|
||||||
|
return {"status": "success", "message": "Пользователь удален"}
|
||||||
@@ -0,0 +1,107 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/routers/auth.py
|
||||||
|
ROLE: Аутентификация, валидация JWT-токенов и управление паролями.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
from typing import Dict, Any, Optional
|
||||||
|
|
||||||
|
import jwt
|
||||||
|
from passlib.context import CryptContext
|
||||||
|
from fastapi import APIRouter, Depends, HTTPException, status
|
||||||
|
from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
from core.connection import get_connection
|
||||||
|
|
||||||
|
JWT_SECRET = "scud_jwt_secret_key_2026_orion_ai_super_secure"
|
||||||
|
ALGORITHM = "HS256"
|
||||||
|
|
||||||
|
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
|
||||||
|
security = HTTPBearer()
|
||||||
|
|
||||||
|
router = APIRouter(prefix="/api/v1/auth", tags=["auth"])
|
||||||
|
|
||||||
|
|
||||||
|
def get_db():
|
||||||
|
return get_connection(row_factory=True)
|
||||||
|
|
||||||
|
|
||||||
|
def create_access_token(user_id: int, username: str, is_admin: bool) -> str:
|
||||||
|
payload = {
|
||||||
|
"sub": str(user_id),
|
||||||
|
"username": username,
|
||||||
|
"is_admin": is_admin,
|
||||||
|
"exp": datetime.utcnow() + timedelta(days=30)
|
||||||
|
}
|
||||||
|
return jwt.encode(payload, JWT_SECRET, algorithm=ALGORITHM)
|
||||||
|
|
||||||
|
|
||||||
|
def get_current_user(credentials: HTTPAuthorizationCredentials = Depends(security)) -> Dict[str, Any]:
|
||||||
|
try:
|
||||||
|
token = credentials.credentials
|
||||||
|
payload = jwt.decode(token, JWT_SECRET, algorithms=[ALGORITHM])
|
||||||
|
user_id = int(payload.get("sub"))
|
||||||
|
username = payload.get("username")
|
||||||
|
is_admin = bool(payload.get("is_admin", False))
|
||||||
|
return {"id": user_id, "username": username, "is_admin": is_admin}
|
||||||
|
except Exception as e:
|
||||||
|
logging.warning(f"Auth error: {e}")
|
||||||
|
raise HTTPException(
|
||||||
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||||
|
detail="Недействительный или просроченный токен авторизации",
|
||||||
|
headers={"WWW-Authenticate": "Bearer"},
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class AuthRequest(BaseModel):
|
||||||
|
username: str
|
||||||
|
password: str
|
||||||
|
|
||||||
|
|
||||||
|
class ChangePasswordRequest(BaseModel):
|
||||||
|
old_password: str
|
||||||
|
new_password: str
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/login")
|
||||||
|
def login(req: AuthRequest):
|
||||||
|
username = req.username.strip().lower()
|
||||||
|
conn = get_db()
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT id, username, password_hash, is_admin FROM users WHERE username = ?", (username,))
|
||||||
|
user = cursor.fetchone()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
if not user or not pwd_context.verify(req.password, user["password_hash"]):
|
||||||
|
raise HTTPException(status_code=401, detail="Неверное имя пользователя или пароль")
|
||||||
|
|
||||||
|
is_admin = bool(user["is_admin"]) or (user["username"] == "puh")
|
||||||
|
token = create_access_token(user["id"], user["username"], is_admin)
|
||||||
|
|
||||||
|
return {"status": "success", "token": token, "username": user["username"], "is_admin": is_admin}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/change-password")
|
||||||
|
def change_password(req: ChangePasswordRequest, current_user: Dict[str, Any] = Depends(get_current_user)):
|
||||||
|
if not req.new_password or len(req.new_password) < 4:
|
||||||
|
raise HTTPException(status_code=400, detail="Новый пароль должен содержать минимум 4 символа")
|
||||||
|
|
||||||
|
conn = get_db()
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT password_hash FROM users WHERE id = ?", (current_user["id"],))
|
||||||
|
user = cursor.fetchone()
|
||||||
|
|
||||||
|
if not user or not pwd_context.verify(req.old_password, user["password_hash"]):
|
||||||
|
conn.close()
|
||||||
|
raise HTTPException(status_code=400, detail="Неверный старый пароль")
|
||||||
|
|
||||||
|
new_hash = pwd_context.hash(req.new_password)
|
||||||
|
cursor.execute("UPDATE users SET password_hash = ? WHERE id = ?", (new_hash, current_user["id"]))
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
return {"status": "success", "message": "Пароль успешно изменен"}
|
||||||
@@ -0,0 +1,97 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/routers/chat.py
|
||||||
|
ROLE: Обработка сообщений веб-чата с поддержкой токенов и гостевого доступа.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Header, HTTPException, Request
|
||||||
|
from pydantic import BaseModel
|
||||||
|
from typing import Optional, List, Dict, Any
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from services.text_reporter import ask_ollama
|
||||||
|
from services.knowledge_base import load_knowledge_base
|
||||||
|
from core.database import get_connection
|
||||||
|
|
||||||
|
logger = logging.getLogger("CHAT_API")
|
||||||
|
router = APIRouter(prefix="/api/v1", tags=["Chat"])
|
||||||
|
|
||||||
|
|
||||||
|
class ChatMessageRequest(BaseModel):
|
||||||
|
message: str
|
||||||
|
session_id: Optional[str] = "web_session_main"
|
||||||
|
user_id: Optional[int] = 1
|
||||||
|
|
||||||
|
|
||||||
|
def resolve_user_id(authorization: Optional[str] = None, explicit_user_id: Optional[int] = None) -> int:
|
||||||
|
"""
|
||||||
|
Извлекает ID пользователя из Bearer-токена.
|
||||||
|
Если токен не передан или сессия новая — использует user_id=1 по умолчанию,
|
||||||
|
не блокируя работу ошибкой 403 Forbidden.
|
||||||
|
"""
|
||||||
|
if explicit_user_id and explicit_user_id > 0:
|
||||||
|
return explicit_user_id
|
||||||
|
|
||||||
|
if authorization and authorization.startswith("Bearer "):
|
||||||
|
token = authorization.replace("Bearer ", "").strip()
|
||||||
|
# Если используется простой токен вида 'user_1' или JWT
|
||||||
|
if token.isdigit():
|
||||||
|
return int(token)
|
||||||
|
elif token.startswith("dev_token_"):
|
||||||
|
try:
|
||||||
|
return int(token.replace("dev_token_", ""))
|
||||||
|
except ValueError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Дефолтный пользователь (гостевой / основной аккаунт)
|
||||||
|
return 1
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/chat")
|
||||||
|
async def chat_endpoint(payload: ChatMessageRequest, authorization: Optional[str] = Header(None)):
|
||||||
|
user_id = resolve_user_id(authorization, payload.user_id)
|
||||||
|
session_id = payload.session_id or "web_session_main"
|
||||||
|
user_msg = payload.message.strip()
|
||||||
|
|
||||||
|
if not user_msg:
|
||||||
|
raise HTTPException(status_code=400, detail="Пустое сообщение")
|
||||||
|
|
||||||
|
logger.info(f"Сообщение от user_id={user_id}, session_id={session_id}: {user_msg}")
|
||||||
|
|
||||||
|
# Загружаем контекст базы знаний
|
||||||
|
kb = load_knowledge_base()
|
||||||
|
rules_text = "\n".join([f"- {r}" for r in kb.get("rules", [])])
|
||||||
|
|
||||||
|
system_prompt = (
|
||||||
|
"Ты — ИИ-ассистент системы кадровой безопасности и контроллинга СКУД Orion AI.\n"
|
||||||
|
"Отвечай четко, профессионально и на русском языке.\n"
|
||||||
|
f"Актуальные правила системы:\n{rules_text}"
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
reply_text = ask_ollama(user_msg, system_prompt=system_prompt)
|
||||||
|
|
||||||
|
# Сохранение истории в SQLite при необходимости
|
||||||
|
try:
|
||||||
|
with get_connection() as conn:
|
||||||
|
conn.execute(
|
||||||
|
"INSERT INTO chat_messages (session_id, role, content) VALUES (?, ?, ?), (?, ?, ?)",
|
||||||
|
(session_id, "user", user_msg, session_id, "assistant", reply_text)
|
||||||
|
)
|
||||||
|
conn.commit()
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
return {
|
||||||
|
"status": "success",
|
||||||
|
"user_id": user_id,
|
||||||
|
"session_id": session_id,
|
||||||
|
"response": reply_text
|
||||||
|
}
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Ошибка вызова нейросети: {e}")
|
||||||
|
return {
|
||||||
|
"status": "error",
|
||||||
|
"response": f"⚠️ Ошибка обработки запроса: {str(e)}"
|
||||||
|
}
|
||||||
@@ -0,0 +1,31 @@
|
|||||||
|
from fastapi import APIRouter, HTTPException
|
||||||
|
from pydantic import BaseModel
|
||||||
|
from typing import Optional, Dict, List
|
||||||
|
from services.exceptions_repo import get_all_exceptions_from_db, add_exception_to_db, remove_exception_from_db
|
||||||
|
|
||||||
|
router = APIRouter(prefix="/api/v1/exceptions", tags=["Exceptions"])
|
||||||
|
|
||||||
|
|
||||||
|
class ExceptionItem(BaseModel):
|
||||||
|
category: str
|
||||||
|
value: str
|
||||||
|
comment: Optional[str] = ""
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/")
|
||||||
|
def api_get_exceptions():
|
||||||
|
return get_all_exceptions_from_db()
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/")
|
||||||
|
def api_add_exception(item: ExceptionItem):
|
||||||
|
if not add_exception_to_db(item.category, item.value, item.comment):
|
||||||
|
raise HTTPException(status_code=400, detail="Ошибка добавления исключения")
|
||||||
|
return {"status": "success", "data": item}
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete("/")
|
||||||
|
def api_delete_exception(category: str, value: str):
|
||||||
|
if not remove_exception_from_db(category, value):
|
||||||
|
raise HTTPException(status_code=404, detail="Исключение не найдено")
|
||||||
|
return {"status": "success"}
|
||||||
@@ -0,0 +1,73 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/routers/files.py
|
||||||
|
ROLE: Раздача сформированных отчетов и выгрузок с сохранением оригинальных имен
|
||||||
|
через изолированные UUID-директории инструментов.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
import shutil
|
||||||
|
import urllib.parse
|
||||||
|
from fastapi import APIRouter, HTTPException
|
||||||
|
from fastapi.responses import FileResponse
|
||||||
|
|
||||||
|
router = APIRouter(prefix="/api/v1/files", tags=["Files"])
|
||||||
|
|
||||||
|
BASE_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../"))
|
||||||
|
WEB_OUTPUT_DIR = os.path.join(BASE_ROOT, "output", "web")
|
||||||
|
os.makedirs(WEB_OUTPUT_DIR, exist_ok=True)
|
||||||
|
|
||||||
|
SESSION_TTL_HOURS = 24 # Срок жизни временных сессионных выгрузок
|
||||||
|
|
||||||
|
|
||||||
|
def purge_old_tool_sessions(tool_dir_path: str):
|
||||||
|
"""Удаляет временные UUID-папки старше SESSION_TTL_HOURS внутри инструмента."""
|
||||||
|
if not os.path.exists(tool_dir_path):
|
||||||
|
return
|
||||||
|
now = time.time()
|
||||||
|
cutoff = now - (SESSION_TTL_HOURS * 3600)
|
||||||
|
try:
|
||||||
|
for entry in os.listdir(tool_dir_path):
|
||||||
|
subpath = os.path.join(tool_dir_path, entry)
|
||||||
|
if os.path.isdir(subpath):
|
||||||
|
if os.path.getmtime(subpath) < cutoff:
|
||||||
|
shutil.rmtree(subpath, ignore_errors=True)
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/download/{tool_name}/{session_uuid}/{filename}")
|
||||||
|
async def download_file(tool_name: str, session_uuid: str, filename: str):
|
||||||
|
"""
|
||||||
|
Безопасная отдача файла с каноническим именем из изолированной директории.
|
||||||
|
"""
|
||||||
|
safe_tool = os.path.basename(tool_name)
|
||||||
|
safe_uuid = os.path.basename(session_uuid)
|
||||||
|
safe_filename = os.path.basename(filename)
|
||||||
|
|
||||||
|
file_path = os.path.join(WEB_OUTPUT_DIR, safe_tool, safe_uuid, safe_filename)
|
||||||
|
|
||||||
|
if not os.path.exists(file_path) or not os.path.isfile(file_path):
|
||||||
|
raise HTTPException(status_code=404, detail="Файл не найден или срок его действия истек")
|
||||||
|
|
||||||
|
# Определение MIME-типа
|
||||||
|
media_type = "application/octet-stream"
|
||||||
|
if safe_filename.endswith(".md") or safe_filename.endswith(".txt"):
|
||||||
|
media_type = "text/markdown; charset=utf-8"
|
||||||
|
elif safe_filename.endswith(".xlsx"):
|
||||||
|
media_type = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
|
||||||
|
elif safe_filename.endswith(".pdf"):
|
||||||
|
media_type = "application/pdf"
|
||||||
|
|
||||||
|
# Корректная кодировка для кириллических имен файлов
|
||||||
|
encoded_filename = urllib.parse.quote(safe_filename)
|
||||||
|
|
||||||
|
return FileResponse(
|
||||||
|
path=file_path,
|
||||||
|
media_type=media_type,
|
||||||
|
headers={
|
||||||
|
"Content-Disposition": f"attachment; filename*=UTF-8''{encoded_filename}"
|
||||||
|
}
|
||||||
|
)
|
||||||
@@ -0,0 +1,76 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/routers/tasks.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: web_api / routers
|
||||||
|
ROLE: REST API эндпоинты реестра задач.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Depends, HTTPException
|
||||||
|
from pydantic import BaseModel
|
||||||
|
from typing import Optional, Dict, Any
|
||||||
|
|
||||||
|
from routers.auth import get_current_user
|
||||||
|
from services.tasks.service import get_tasks, add_task, update_task_details
|
||||||
|
|
||||||
|
router = APIRouter(prefix="/api/v1/tasks", tags=["Tasks"])
|
||||||
|
|
||||||
|
|
||||||
|
class TaskCreateRequest(BaseModel):
|
||||||
|
title: str
|
||||||
|
priority: Optional[str] = "MEDIUM"
|
||||||
|
module: Optional[str] = "general"
|
||||||
|
due_date: Optional[str] = None
|
||||||
|
status: Optional[str] = "BACKLOG"
|
||||||
|
|
||||||
|
|
||||||
|
class TaskUpdateRequest(BaseModel):
|
||||||
|
title: Optional[str] = None
|
||||||
|
priority: Optional[str] = None
|
||||||
|
due_date: Optional[str] = None
|
||||||
|
status: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
|
def resolve_user_id(current_user: Dict[str, Any]) -> int:
|
||||||
|
if not current_user:
|
||||||
|
return 1
|
||||||
|
return current_user.get("id") or current_user.get("user_id") or 1
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("")
|
||||||
|
async def get_tasks_endpoint(status: Optional[str] = None, current_user = Depends(get_current_user)):
|
||||||
|
user_id = resolve_user_id(current_user)
|
||||||
|
return {"tasks": get_tasks(user_id=user_id, status=status)}
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("")
|
||||||
|
async def create_task_endpoint(req: TaskCreateRequest, current_user = Depends(get_current_user)):
|
||||||
|
user_id = resolve_user_id(current_user)
|
||||||
|
res = add_task(
|
||||||
|
user_id=user_id,
|
||||||
|
module=req.module,
|
||||||
|
title=req.title,
|
||||||
|
priority=req.priority,
|
||||||
|
due_date=req.due_date,
|
||||||
|
status=req.status
|
||||||
|
)
|
||||||
|
if "error" in res:
|
||||||
|
raise HTTPException(status_code=400, detail=res["error"])
|
||||||
|
return res
|
||||||
|
|
||||||
|
|
||||||
|
@router.patch("/{task_id}")
|
||||||
|
async def update_task_endpoint(task_id: str, req: TaskUpdateRequest, current_user = Depends(get_current_user)):
|
||||||
|
user_id = resolve_user_id(current_user)
|
||||||
|
res = update_task_details(
|
||||||
|
user_id=user_id,
|
||||||
|
task_id=task_id,
|
||||||
|
title=req.title,
|
||||||
|
priority=req.priority,
|
||||||
|
status=req.status,
|
||||||
|
due_date=req.due_date
|
||||||
|
)
|
||||||
|
if "error" in res:
|
||||||
|
raise HTTPException(status_code=404, detail=res["error"])
|
||||||
|
return res
|
||||||
@@ -1,47 +0,0 @@
|
|||||||
import os
|
|
||||||
import sqlite3
|
|
||||||
|
|
||||||
# Автопоиск файла базы данных в проекте
|
|
||||||
db_path = '/home/puh/scud_orion_ai_v2/data/scud_orion_ai.db' if os.path.exists('/home/puh/scud_orion_ai_v2/data/scud_orion_ai.db') else 'scud_orion_ai.db'
|
|
||||||
|
|
||||||
print("=" * 80)
|
|
||||||
print(f"🔍 ДИАГНОСТИКА СУБД SQLITE: {db_path}")
|
|
||||||
print("=" * 80)
|
|
||||||
|
|
||||||
if not os.path.exists(db_path):
|
|
||||||
print(f"❌ Файл базы данных {db_path} не найден!")
|
|
||||||
exit(1)
|
|
||||||
|
|
||||||
conn = sqlite3.connect(db_path)
|
|
||||||
cursor = conn.cursor()
|
|
||||||
|
|
||||||
# 1. Список всех таблиц и колонок
|
|
||||||
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' ORDER BY name")
|
|
||||||
tables = [t[0] for t in cursor.fetchall()]
|
|
||||||
|
|
||||||
print("\n📋 СТРУКТУРА ТАБЛИЦ И КОЛИЧЕСТВО ЗАПИСЕЙ:")
|
|
||||||
print("-" * 80)
|
|
||||||
for t_name in tables:
|
|
||||||
cursor.execute(f"PRAGMA table_info({t_name})")
|
|
||||||
cols = [c[1] for c in cursor.fetchall()]
|
|
||||||
|
|
||||||
cursor.execute(f"SELECT COUNT(*) FROM {t_name}")
|
|
||||||
count = cursor.fetchone()[0]
|
|
||||||
|
|
||||||
print(f"• [{t_name:<20}] — {count:>6} строк | Колонки: {cols}")
|
|
||||||
|
|
||||||
# 2. Просмотр правил Базы Знаний
|
|
||||||
if 'ai_knowledge_base' in tables:
|
|
||||||
print("\n" + "=" * 80)
|
|
||||||
print("🧠 АКТУАЛЬНЫЕ ПРАВИЛА БАЗЫ ЗНАНИЙ (ai_knowledge_base):")
|
|
||||||
print("=" * 80)
|
|
||||||
cursor.execute("SELECT id, rule_text, added_by FROM ai_knowledge_base ORDER BY id ASC")
|
|
||||||
rules = cursor.fetchall()
|
|
||||||
if not rules:
|
|
||||||
print("Таблица ai_knowledge_base пуста.")
|
|
||||||
else:
|
|
||||||
for r_id, r_text, r_author in rules:
|
|
||||||
print(f" {r_id}. [{r_author}] {r_text}\n")
|
|
||||||
|
|
||||||
conn.close()
|
|
||||||
print("=" * 80)
|
|
||||||
@@ -1,23 +0,0 @@
|
|||||||
import os
|
|
||||||
|
|
||||||
print("=" * 80)
|
|
||||||
print("📂 ТЕКУЩЕЕ СОСТОЯНИЕ ФАЙЛОВ ПРОЕКТА (scud_orion_context)")
|
|
||||||
print("=" * 80)
|
|
||||||
|
|
||||||
total_files = 0
|
|
||||||
total_size = 0
|
|
||||||
|
|
||||||
for root, dirs, files in os.walk('.'):
|
|
||||||
# Исключаем служебные каталоги
|
|
||||||
dirs[:] = [d for d in dirs if d not in ['.git', '__pycache__', 'venv', '.venv', 'extracted_project']]
|
|
||||||
|
|
||||||
for f in files:
|
|
||||||
p = os.path.join(root, f)
|
|
||||||
size = os.path.getsize(p)
|
|
||||||
total_files += 1
|
|
||||||
total_size += size
|
|
||||||
print(f"{p:<55} ({size:>10,} bytes)".replace(',', ' '))
|
|
||||||
|
|
||||||
print("-" * 80)
|
|
||||||
print(f"ИТОГО: файлов: {total_files} | Общий объем: {total_size / (1024 * 1024):.2f} MB")
|
|
||||||
print("=" * 80)
|
|
||||||
@@ -1,31 +0,0 @@
|
|||||||
import os
|
|
||||||
|
|
||||||
OUTPUT_SNAPSHOT = "api_code_snapshot.md"
|
|
||||||
|
|
||||||
# Расширения файлов для включения в снимок
|
|
||||||
ALLOWED_EXTENSIONS = {'.py', '.json', '.md', '.sh', '.ini', '.js', '.html', '.css'}
|
|
||||||
EXCLUDE_DIRS = {'.git', '__pycache__', 'venv', '.venv', 'output', 'logs', 'extracted_project'}
|
|
||||||
EXCLUDE_FILES = {OUTPUT_SNAPSHOT, 'scud_context_api.tar.gz', 'context_memory.db'}
|
|
||||||
|
|
||||||
print(f"🔄 Сборка полного контекстного слепка проекта в {OUTPUT_SNAPSHOT}...")
|
|
||||||
|
|
||||||
with open(OUTPUT_SNAPSHOT, 'w', encoding='utf-8') as out:
|
|
||||||
out.write("# 📦 ПОЛНЫЙ ИСХОДНЫЙ КОД И КОНФИГУРАЦИЯ ПРОЕКТА scud_context_api\n\n")
|
|
||||||
|
|
||||||
for root, dirs, files in os.walk('.'):
|
|
||||||
dirs[:] = [d for d in dirs if d not in EXCLUDE_DIRS]
|
|
||||||
|
|
||||||
for file in sorted(files):
|
|
||||||
ext = os.path.splitext(file)[1].lower()
|
|
||||||
if ext in ALLOWED_EXTENSIONS and file not in EXCLUDE_FILES:
|
|
||||||
filepath = os.path.join(root, file)
|
|
||||||
out.write(f"## File: `{filepath}`\n")
|
|
||||||
out.write("```" + (ext.replace('.', '') if ext != '.md' else '') + "\n")
|
|
||||||
try:
|
|
||||||
with open(filepath, 'r', encoding='utf-8', errors='replace') as f:
|
|
||||||
out.write(f.read())
|
|
||||||
except Exception as e:
|
|
||||||
out.write(f"// Ошибка чтения файла: {e}\n")
|
|
||||||
out.write("\n```\n\n")
|
|
||||||
|
|
||||||
print(f"✓ Успешно создан слепок проекта: {OUTPUT_SNAPSHOT} ({os.path.getsize(OUTPUT_SNAPSHOT):,} bytes)")
|
|
||||||
+146
-242
@@ -2,260 +2,164 @@
|
|||||||
<html lang="ru">
|
<html lang="ru">
|
||||||
<head>
|
<head>
|
||||||
<meta charset="UTF-8">
|
<meta charset="UTF-8">
|
||||||
<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no, viewport-fit=cover">
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
<title>SCUD Orion AI — Context Manager</title>
|
<title>SCUD Orion AI — Управление и Аналитика</title>
|
||||||
|
<!-- Tailwind CSS CDN -->
|
||||||
<script src="https://cdn.tailwindcss.com"></script>
|
<script src="https://cdn.tailwindcss.com"></script>
|
||||||
|
<!-- FontAwesome Icons -->
|
||||||
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css">
|
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css">
|
||||||
<link rel="stylesheet" href="/static/css/styles.css">
|
<link rel="stylesheet" href="/css/styles.css">
|
||||||
</head>
|
</head>
|
||||||
<body class="bg-slate-100 text-slate-800 h-[100dvh] w-full flex flex-col font-sans overflow-hidden">
|
<body class="bg-slate-100 font-sans h-screen flex overflow-hidden text-slate-800">
|
||||||
|
|
||||||
<!-- Окно авторизации -->
|
<!-- Боковая панель (Задачи и Навигация) -->
|
||||||
<div id="auth-modal" class="fixed inset-0 bg-slate-900/80 backdrop-blur-md z-50 flex items-center justify-center p-4">
|
<aside id="task-drawer" class="w-80 sm:w-96 bg-white border-r border-slate-200 flex flex-col shrink-0 h-full z-20 shadow-sm transition-all duration-300">
|
||||||
<div class="bg-white rounded-2xl p-6 sm:p-8 max-w-md w-full shadow-2xl border border-slate-200">
|
<!-- Шапка панели задач -->
|
||||||
<div class="flex items-center space-x-3 mb-6">
|
<div class="p-4 border-b border-slate-200 flex items-center justify-between bg-slate-50/70">
|
||||||
<div class="bg-indigo-600 text-white p-3 rounded-xl">
|
|
||||||
<i class="fa-solid fa-user-shield text-xl"></i>
|
|
||||||
</div>
|
|
||||||
<div>
|
|
||||||
<h2 class="text-lg font-bold text-slate-900">SCUD Orion AI</h2>
|
|
||||||
<p class="text-xs text-slate-500">Авторизация в системе</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="space-y-4">
|
|
||||||
<div>
|
|
||||||
<label class="block text-xs font-semibold text-slate-700 uppercase tracking-wider mb-1">Имя пользователя</label>
|
|
||||||
<input type="text" id="auth-username-input" placeholder="Введите логин..." required autocomplete="username"
|
|
||||||
class="w-full bg-slate-50 border border-slate-300 rounded-xl px-4 py-2.5 text-sm text-slate-900 focus:outline-none focus:border-indigo-600 focus:bg-white transition">
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div>
|
|
||||||
<label class="block text-xs font-semibold text-slate-700 uppercase tracking-wider mb-1">Пароль</label>
|
|
||||||
<input type="password" id="auth-password-input" placeholder="Введите пароль..." required autocomplete="current-password"
|
|
||||||
class="w-full bg-slate-50 border border-slate-300 rounded-xl px-4 py-2.5 text-sm text-slate-900 focus:outline-none focus:border-indigo-600 focus:bg-white transition">
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="auth-error" class="hidden text-xs text-red-600 font-medium bg-red-50 p-3 rounded-xl border border-red-200"></div>
|
|
||||||
|
|
||||||
<button type="button" onclick="handleLogin()" id="auth-btn" class="w-full bg-indigo-600 hover:bg-indigo-700 active:bg-indigo-800 text-white font-semibold py-3 rounded-xl text-sm transition shadow-md flex items-center justify-center gap-2">
|
|
||||||
<i class="fa-solid fa-right-to-bracket"></i>
|
|
||||||
<span>Войти в систему</span>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="relative my-4">
|
|
||||||
<div class="absolute inset-0 flex items-center"><div class="w-full border-t border-slate-200"></div></div>
|
|
||||||
<div class="relative flex justify-center text-xs uppercase"><span class="bg-white px-2 text-slate-400 font-medium">Или</span></div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<button type="button" onclick="enableGuestMode()" class="w-full bg-slate-100 hover:bg-slate-200 text-slate-700 font-semibold py-2.5 rounded-xl text-xs transition border border-slate-300 flex items-center justify-center gap-2">
|
|
||||||
<i class="fa-solid fa-user-ninja"></i>
|
|
||||||
<span>Войти как гость (Локальный ИИ)</span>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<!-- Модальное окно смены пароля -->
|
|
||||||
<div id="change-pwd-modal" class="fixed inset-0 bg-slate-900/60 backdrop-blur-sm z-50 hidden flex items-center justify-center p-4">
|
|
||||||
<div class="bg-white rounded-2xl p-6 max-w-sm w-full shadow-2xl border border-slate-200">
|
|
||||||
<div class="flex justify-between items-center mb-4">
|
|
||||||
<h3 class="font-bold text-slate-800 text-sm flex items-center gap-2">
|
|
||||||
<i class="fa-solid fa-key text-indigo-600"></i> Смена пароля
|
|
||||||
</h3>
|
|
||||||
<button type="button" onclick="closeChangePasswordModal()" class="text-slate-400 hover:text-slate-700">
|
|
||||||
<i class="fa-solid fa-xmark text-lg"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="space-y-3">
|
|
||||||
<div>
|
|
||||||
<label class="block text-[11px] font-semibold text-slate-600 uppercase mb-1">Старый пароль</label>
|
|
||||||
<input type="password" id="old-pwd-input" required class="w-full bg-slate-50 border border-slate-300 rounded-xl px-3 py-2 text-xs text-slate-900 focus:outline-none focus:border-indigo-600">
|
|
||||||
</div>
|
|
||||||
<div>
|
|
||||||
<label class="block text-[11px] font-semibold text-slate-600 uppercase mb-1">Новый пароль</label>
|
|
||||||
<input type="password" id="new-pwd-input" required class="w-full bg-slate-50 border border-slate-300 rounded-xl px-3 py-2 text-xs text-slate-900 focus:outline-none focus:border-indigo-600">
|
|
||||||
</div>
|
|
||||||
<div>
|
|
||||||
<label class="block text-[11px] font-semibold text-slate-600 uppercase mb-1">Повторите новый пароль</label>
|
|
||||||
<input type="password" id="confirm-pwd-input" required class="w-full bg-slate-50 border border-slate-300 rounded-xl px-3 py-2 text-xs text-slate-900 focus:outline-none focus:border-indigo-600">
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="pwd-error" class="hidden text-xs text-red-600 bg-red-50 p-2 rounded-lg border border-red-200"></div>
|
|
||||||
<div id="pwd-success" class="hidden text-xs text-emerald-600 bg-emerald-50 p-2 rounded-lg border border-emerald-200"></div>
|
|
||||||
|
|
||||||
<button type="button" onclick="handleChangePassword()" class="w-full bg-indigo-600 hover:bg-indigo-700 text-white font-semibold py-2.5 rounded-xl text-xs transition shadow-sm mt-2">
|
|
||||||
Сохранить новый пароль
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<!-- Модальное окно управления пользователями -->
|
|
||||||
<div id="admin-modal" class="fixed inset-0 bg-slate-900/60 backdrop-blur-sm z-50 hidden flex items-center justify-center p-4">
|
|
||||||
<div class="bg-white rounded-2xl p-6 max-w-lg w-full shadow-2xl border border-slate-200 flex flex-col max-h-[85vh]">
|
|
||||||
<div class="flex justify-between items-center mb-4 pb-2 border-b border-slate-200">
|
|
||||||
<h3 class="font-bold text-slate-800 text-sm flex items-center gap-2">
|
|
||||||
<i class="fa-solid fa-users-gear text-indigo-600"></i> Управление пользователями
|
|
||||||
</h3>
|
|
||||||
<button type="button" onclick="closeAdminModal()" class="text-slate-400 hover:text-slate-700">
|
|
||||||
<i class="fa-solid fa-xmark text-lg"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="space-y-2 mb-4 bg-slate-50 p-3.5 rounded-xl border border-slate-200 shrink-0">
|
|
||||||
<p class="text-[11px] font-bold text-slate-700 uppercase">Создать нового пользователя</p>
|
|
||||||
<div class="grid grid-cols-2 gap-2">
|
|
||||||
<input type="text" id="new-user-name" placeholder="Логин *" required class="bg-white border border-slate-300 rounded-lg px-2.5 py-1.5 text-xs">
|
|
||||||
<input type="password" id="new-user-pwd" placeholder="Пароль *" required class="bg-white border border-slate-300 rounded-lg px-2.5 py-1.5 text-xs">
|
|
||||||
</div>
|
|
||||||
<input type="text" id="new-user-fullname" placeholder="ФИО (необязательно)" class="w-full bg-white border border-slate-300 rounded-lg px-2.5 py-1.5 text-xs">
|
|
||||||
<div class="flex items-center justify-between pt-1">
|
|
||||||
<label class="flex items-center gap-2 text-xs text-slate-700 cursor-pointer">
|
|
||||||
<input type="checkbox" id="new-user-is-admin" class="rounded border-slate-300 text-indigo-600 focus:ring-indigo-500">
|
|
||||||
<span>Права администратора</span>
|
|
||||||
</label>
|
|
||||||
<button type="button" onclick="handleCreateUser()" class="bg-indigo-600 hover:bg-indigo-700 text-white font-semibold px-4 py-1.5 rounded-lg text-xs transition">
|
|
||||||
+ Добавить
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
<div id="admin-msg" class="hidden text-[11px] text-red-600 pt-1"></div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="flex-1 overflow-y-auto space-y-2 pr-1" id="admin-users-list">
|
|
||||||
<div class="text-xs text-slate-400 py-4 text-center">Загрузка пользователей...</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<!-- Хедер -->
|
|
||||||
<header class="bg-white border-b border-slate-200 px-4 py-2.5 flex justify-between items-center shadow-sm shrink-0 z-20">
|
|
||||||
<div class="flex items-center space-x-2.5">
|
|
||||||
<div class="bg-indigo-600 text-white p-2 rounded-xl shrink-0">
|
|
||||||
<i class="fa-solid fa-brain text-lg"></i>
|
|
||||||
</div>
|
|
||||||
<div>
|
|
||||||
<h1 class="text-sm font-bold text-slate-900 leading-tight">SCUD Orion AI</h1>
|
|
||||||
<span class="text-[11px] text-slate-500">Task & Context API</span>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="flex items-center space-x-1.5">
|
|
||||||
<span id="guest-badge" class="hidden text-[10px] text-amber-700 font-semibold bg-amber-50 px-2 py-0.5 rounded-full border border-amber-200">
|
|
||||||
Гость
|
|
||||||
</span>
|
|
||||||
|
|
||||||
<span id="username-badge" class="hidden text-xs text-indigo-700 font-bold bg-indigo-50 px-2.5 py-1 rounded-full border border-indigo-200">
|
|
||||||
puh
|
|
||||||
</span>
|
|
||||||
|
|
||||||
<button id="admin-users-btn" type="button" onclick="openAdminModal()" class="hidden text-slate-500 hover:text-indigo-600 transition p-2 rounded-xl" title="Управление пользователями">
|
|
||||||
<i class="fa-solid fa-users-gear text-base"></i>
|
|
||||||
</button>
|
|
||||||
|
|
||||||
<button id="change-pwd-btn" type="button" onclick="openChangePasswordModal()" class="hidden text-slate-500 hover:text-indigo-600 transition p-2 rounded-xl" title="Сменить пароль">
|
|
||||||
<i class="fa-solid fa-key text-base"></i>
|
|
||||||
</button>
|
|
||||||
|
|
||||||
<button id="tasks-drawer-btn" type="button" onclick="toggleDrawer()" class="bg-indigo-600 active:bg-indigo-700 text-white px-3 py-1.5 rounded-xl text-xs font-semibold flex items-center gap-1.5 shadow-sm">
|
|
||||||
<i class="fa-solid fa-list-check"></i>
|
|
||||||
<span>Задачи</span>
|
|
||||||
<span id="task-count-badge" class="bg-white text-indigo-700 text-[10px] font-bold px-1.5 py-0.2 rounded-full">0</span>
|
|
||||||
</button>
|
|
||||||
|
|
||||||
<button type="button" onclick="logout()" class="text-slate-400 hover:text-red-600 transition p-2 rounded-xl" title="Выйти">
|
|
||||||
<i class="fa-solid fa-right-from-bracket text-base"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</header>
|
|
||||||
|
|
||||||
<!-- Главный контейнер -->
|
|
||||||
<div class="flex-1 flex flex-col min-h-0 w-full max-w-4xl mx-auto bg-white relative overflow-hidden">
|
|
||||||
<div id="chat-window" class="flex-1 p-3.5 overflow-y-auto space-y-3 bg-slate-50/50">
|
|
||||||
<div id="drop-overlay" class="absolute inset-0 bg-indigo-600/10 backdrop-blur-sm border-2 border-dashed border-indigo-600 rounded-2xl hidden flex-col items-center justify-center z-30 transition-all pointer-events-none">
|
|
||||||
<div class="bg-white p-4 rounded-2xl shadow-xl flex flex-col items-center gap-2">
|
|
||||||
<i class="fa-solid fa-cloud-arrow-up text-3xl text-indigo-600 animate-bounce"></i>
|
|
||||||
<p class="text-sm font-bold text-slate-800">Перетащите файл сюда</p>
|
|
||||||
<p class="text-xs text-slate-500">Поддерживаются PDF, изображения, таблицы, TXT</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div class="bg-white border border-slate-200 rounded-2xl p-3.5 shadow-sm">
|
|
||||||
<p class="text-[11px] font-bold text-indigo-600 uppercase tracking-wider mb-1">
|
|
||||||
<i class="fa-solid fa-robot mr-1"></i> ИИ-Ассистент
|
|
||||||
</p>
|
|
||||||
<p class="text-slate-800 text-xs sm:text-sm leading-relaxed">
|
|
||||||
Привет! У каждого пользователя свое изолированное пространство задач. Вы можете задавать вопросы нейросети, прикреплять файлы или ставить персональные задачи.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<!-- Превью прикрепленного файла -->
|
|
||||||
<div id="file-preview-container" class="hidden px-4 py-2 bg-slate-100 border-t border-slate-200 flex items-center justify-between text-xs text-slate-700">
|
|
||||||
<div class="flex items-center gap-2 truncate">
|
|
||||||
<i class="fa-solid fa-paperclip text-indigo-600"></i>
|
|
||||||
<span id="file-name-display" class="font-medium truncate">file.pdf</span>
|
|
||||||
<span id="file-size-display" class="text-slate-400 text-[10px]">(0 KB)</span>
|
|
||||||
</div>
|
|
||||||
<button type="button" onclick="clearAttachedFile()" class="text-slate-400 hover:text-red-500 p-1 transition">
|
|
||||||
<i class="fa-solid fa-xmark text-sm"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div class="p-2.5 pb-6 bg-white border-t border-slate-200 shrink-0 z-10 shadow-lg">
|
|
||||||
<div class="flex items-center gap-2">
|
<div class="flex items-center gap-2">
|
||||||
<!-- Скрытый инпут и кнопка прикрепления файла -->
|
<div class="w-8 h-8 rounded-lg bg-indigo-600 flex items-center justify-center text-white shadow-sm">
|
||||||
<input type="file" id="file-input" class="hidden" onchange="handleFileSelect(event)" accept=".png,.jpg,.jpeg,.pdf,.txt,.csv,.xlsx">
|
<i class="fa-solid fa-list-check text-sm"></i>
|
||||||
<button type="button" onclick="document.getElementById('file-input').click()" class="text-slate-500 hover:text-indigo-600 p-2 rounded-xl transition" title="Прикрепить файл">
|
</div>
|
||||||
<i class="fa-solid fa-paperclip text-lg"></i>
|
<div>
|
||||||
</button>
|
<h2 class="font-bold text-sm text-slate-900 leading-tight">Бэклог задач</h2>
|
||||||
|
<p class="text-[11px] text-slate-500">SCUD Orion AI Roadmap</p>
|
||||||
<div class="flex-1 bg-slate-100 border border-slate-300 rounded-2xl px-3 py-1.5 focus-within:border-indigo-600 focus-within:bg-white transition">
|
|
||||||
<textarea id="user-input" rows="1" autocomplete="off" autocorrect="off" spellcheck="false"
|
|
||||||
placeholder="Команда, вопрос или перетащите файл сюда..."
|
|
||||||
class="w-full bg-transparent text-slate-900 text-sm focus:outline-none resize-none overflow-y-auto h-[24px] max-h-[120px] leading-[24px] fade-scroll-top no-scrollbar"></textarea>
|
|
||||||
</div>
|
</div>
|
||||||
<button type="button" id="send-btn" onclick="sendMessage()" class="bg-indigo-600 active:bg-indigo-800 text-white font-semibold px-3.5 py-2.5 rounded-2xl text-xs sm:text-sm transition flex items-center justify-center gap-1.5 shrink-0 shadow-sm">
|
|
||||||
<span>Отправить</span>
|
|
||||||
<i class="fa-solid fa-paper-plane text-xs"></i>
|
|
||||||
</button>
|
|
||||||
</div>
|
</div>
|
||||||
|
<button type="button" onclick="openAddTaskModal()" title="Добавить задачу"
|
||||||
|
class="px-2.5 py-1.5 bg-indigo-50 hover:bg-indigo-100 text-indigo-700 border border-indigo-200 rounded-lg text-xs font-semibold flex items-center gap-1 transition cursor-pointer">
|
||||||
|
<i class="fa-solid fa-plus"></i>
|
||||||
|
<span>Задача</span>
|
||||||
|
</button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
|
||||||
|
|
||||||
<!-- Выезжающая панель (Drawer) -->
|
<!-- Фильтры задач -->
|
||||||
<div id="drawer-backdrop" onclick="toggleDrawer()" class="fixed inset-0 bg-slate-900/50 backdrop-blur-sm hidden transition-opacity z-40"></div>
|
<div class="px-4 py-2.5 border-b border-slate-100 flex items-center justify-between gap-1 text-xs">
|
||||||
|
<button onclick="filterTasksByTab('IN_PROGRESS')" id="tab-in-progress" class="task-tab-btn font-semibold px-2.5 py-1 rounded-md text-indigo-600 bg-indigo-50 transition">В работе</button>
|
||||||
<aside id="task-drawer" class="fixed right-0 top-0 h-full w-full sm:w-[420px] bg-white border-l border-slate-200 shadow-2xl transform translate-x-full transition-transform duration-300 ease-in-out z-50 flex flex-col">
|
<button onclick="filterTasksByTab('BACKLOG')" id="tab-backlog" class="task-tab-btn font-medium px-2.5 py-1 rounded-md text-slate-600 hover:bg-slate-100 transition">В планах</button>
|
||||||
<div class="p-3.5 border-b border-slate-200 flex justify-between items-center bg-slate-50 shrink-0">
|
<button onclick="filterTasksByTab('COMPLETED')" id="tab-completed" class="task-tab-btn font-medium px-2.5 py-1 rounded-md text-slate-600 hover:bg-slate-100 transition">Готово</button>
|
||||||
<h2 class="font-bold text-slate-800 flex items-center gap-2 text-sm">
|
<button onclick="filterTasksByTab('ALL')" id="tab-all" class="task-tab-btn font-medium px-2.5 py-1 rounded-md text-slate-600 hover:bg-slate-100 transition">Все</button>
|
||||||
<i class="fa-solid fa-list-check text-indigo-600"></i> Мой реестр задач
|
</div>
|
||||||
</h2>
|
|
||||||
<div class="flex items-center gap-3">
|
<!-- Список задач с прокруткой (ID исправлен на tasks-list) -->
|
||||||
<button type="button" onclick="loadTasks()" class="text-xs text-slate-500 hover:text-indigo-600 transition p-1" title="Обновить">
|
<div id="tasks-list" class="flex-1 overflow-y-auto p-3 space-y-2.5">
|
||||||
<i class="fa-solid fa-rotate-right text-sm"></i>
|
<div class="text-center py-8 text-xs text-slate-400">Загрузка задач...</div>
|
||||||
</button>
|
</div>
|
||||||
<button type="button" onclick="toggleDrawer()" class="text-slate-500 hover:text-slate-800 transition p-1">
|
|
||||||
<i class="fa-solid fa-xmark text-lg"></i>
|
<!-- Подвал панели пользователя -->
|
||||||
</button>
|
<div class="p-3 border-t border-slate-200 bg-slate-50/50 flex items-center justify-between text-xs">
|
||||||
|
<div class="flex items-center gap-2">
|
||||||
|
<div class="w-7 h-7 rounded-full bg-slate-300 flex items-center justify-center text-slate-700 font-bold">
|
||||||
|
<i class="fa-solid fa-user text-xs"></i>
|
||||||
|
</div>
|
||||||
|
<div class="truncate">
|
||||||
|
<span id="current-username" class="font-semibold text-slate-900 block truncate">Александр Пушков</span>
|
||||||
|
<span id="user-role-badge" class="text-[10px] text-indigo-600 font-medium">Администратор</span>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
<button onclick="logout()" title="Выйти" class="p-1.5 text-slate-400 hover:text-rose-600 transition">
|
||||||
|
<i class="fa-solid fa-arrow-right-from-bracket"></i>
|
||||||
<div class="flex border-b border-slate-200 bg-white px-2 pt-2 text-xs font-semibold text-slate-500 gap-1 overflow-x-auto no-scrollbar shrink-0">
|
</button>
|
||||||
<button type="button" onclick="setFilter('ALL')" id="filter-ALL" class="px-3 py-1.5 rounded-t-lg border-b-2 border-indigo-600 text-indigo-600 font-bold whitespace-nowrap">Все</button>
|
|
||||||
<button type="button" onclick="setFilter('IN_PROGRESS')" id="filter-IN_PROGRESS" class="px-3 py-1.5 rounded-t-lg border-b-2 border-transparent whitespace-nowrap">В работе</button>
|
|
||||||
<button type="button" onclick="setFilter('BACKLOG')" id="filter-BACKLOG" class="px-3 py-1.5 rounded-t-lg border-b-2 border-transparent whitespace-nowrap">Бэклог</button>
|
|
||||||
<button type="button" onclick="setFilter('COMPLETED')" id="filter-COMPLETED" class="px-3 py-1.5 rounded-t-lg border-b-2 border-transparent whitespace-nowrap">Завершено</button>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div id="tasks-container" class="flex-1 overflow-y-auto p-3.5 space-y-3 bg-slate-50/50 pb-8">
|
|
||||||
<div class="text-center text-slate-400 py-8 text-xs">Загрузка ваших задач...</div>
|
|
||||||
</div>
|
</div>
|
||||||
</aside>
|
</aside>
|
||||||
|
|
||||||
<script src="/static/js/auth.js"></script>
|
<!-- Основная рабочая область чата -->
|
||||||
<script src="/static/js/tasks.js"></script>
|
<main class="flex-1 flex flex-col min-w-0 bg-white relative h-full overflow-hidden">
|
||||||
<script src="/static/js/chat.js"></script>
|
<!-- Верхний заголовок чата -->
|
||||||
<script src="/static/js/app.js"></script>
|
<header class="h-14 border-b border-slate-200 px-4 flex items-center justify-between bg-white shrink-0 z-10">
|
||||||
|
<div class="flex items-center gap-3">
|
||||||
|
<button type="button" onclick="toggleTaskDrawer()" class="p-1.5 text-slate-500 hover:text-indigo-600 hover:bg-slate-100 rounded-lg transition" title="Переключить боковую панель">
|
||||||
|
<i class="fa-solid fa-bars"></i>
|
||||||
|
</button>
|
||||||
|
<div>
|
||||||
|
<h1 class="font-bold text-sm text-slate-900 flex items-center gap-2">
|
||||||
|
<span>SCUD Orion AI Assistant</span>
|
||||||
|
<span class="inline-flex items-center px-1.5 py-0.5 rounded-full text-[10px] font-semibold bg-emerald-100 text-emerald-800">Online</span>
|
||||||
|
</h1>
|
||||||
|
<p class="text-[11px] text-slate-500">Система интеллектуального аудита и контроля СКУД / 1С</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="flex items-center gap-2 text-xs">
|
||||||
|
<button type="button" onclick="handleActionButtonClick('покажи системный промпт')" class="px-2.5 py-1 text-slate-600 hover:text-indigo-600 hover:bg-slate-100 rounded-md border border-slate-200 transition">
|
||||||
|
<i class="fa-solid fa-terminal mr-1"></i> Промпт
|
||||||
|
</button>
|
||||||
|
<button type="button" onclick="handleActionButtonClick('покажи снапшоты')" class="px-2.5 py-1 text-slate-600 hover:text-indigo-600 hover:bg-slate-100 rounded-md border border-slate-200 transition">
|
||||||
|
<i class="fa-solid fa-camera mr-1"></i> Срезы СКУД
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</header>
|
||||||
|
|
||||||
|
<!-- Drop Overlay при перетаскивании файлов -->
|
||||||
|
<div id="drop-overlay" class="hidden absolute inset-0 bg-indigo-600/10 backdrop-blur-[2px] border-2 border-dashed border-indigo-500 rounded-2xl m-4 z-50 items-center justify-center flex-col gap-2 pointer-events-none">
|
||||||
|
<i class="fa-solid fa-cloud-arrow-up text-3xl text-indigo-600 animate-bounce"></i>
|
||||||
|
<p class="text-xs font-bold text-indigo-900">Перетащите файл сюда для отправки в диалог</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- Окно сообщений с центрированием контента -->
|
||||||
|
<div id="chat-window" class="flex-1 p-4 overflow-y-auto bg-slate-50/50 scroll-smooth">
|
||||||
|
<!-- Центрирующая колонка для сообщений -->
|
||||||
|
<div class="max-w-4xl w-full mx-auto space-y-4">
|
||||||
|
|
||||||
|
<!-- Стартовое приветственное сообщение -->
|
||||||
|
<div class="chat-message-card bg-white border border-slate-200 rounded-2xl p-4 shadow-sm w-full">
|
||||||
|
<p class="text-[11px] font-bold text-indigo-600 uppercase tracking-wider mb-1">
|
||||||
|
<i class="fa-solid fa-robot mr-1"></i> ИИ-Ассистент SCUD Orion AI
|
||||||
|
</p>
|
||||||
|
<div class="text-slate-800 text-xs sm:text-sm leading-relaxed">
|
||||||
|
Привет! У каждого пользователя свое изолированное пространство задач. Вы можете задавать вопросы нейросети, прикреплять файлы или ставить персональные задачи.
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- Невидимая распорка в самом низу для свободного скролла любого вопроса наверх -->
|
||||||
|
<div id="chat-bottom-spacer" class="min-h-[85vh] pointer-events-none w-full"></div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- Центрированная нижняя панель ввода -->
|
||||||
|
<div class="p-3 bg-white border-t border-slate-200 shrink-0 z-10">
|
||||||
|
<div class="max-w-4xl w-full mx-auto">
|
||||||
|
<!-- Блок предпросмотра прикрепленного файла -->
|
||||||
|
<div id="file-preview-container" class="hidden mb-2 p-2 bg-indigo-50 border border-indigo-200 rounded-xl flex items-center justify-between">
|
||||||
|
<div class="flex items-center gap-2 min-w-0">
|
||||||
|
<i class="fa-solid fa-file-arrow-up text-indigo-600 text-sm"></i>
|
||||||
|
<span id="file-name-display" class="text-xs font-semibold text-slate-800 truncate"></span>
|
||||||
|
<span id="file-size-display" class="text-[10px] text-slate-500"></span>
|
||||||
|
</div>
|
||||||
|
<button type="button" onclick="clearAttachedFile()" class="text-slate-400 hover:text-rose-600 transition p-1">
|
||||||
|
<i class="fa-solid fa-xmark"></i>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- Форма отправки -->
|
||||||
|
<form id="chat-form" onsubmit="sendMessage(event)" class="flex items-end gap-2">
|
||||||
|
<input type="file" id="file-input" class="hidden" onchange="handleFileSelect(event)" />
|
||||||
|
|
||||||
|
<button type="button" onclick="document.getElementById('file-input').click()"
|
||||||
|
title="Прикрепить файл"
|
||||||
|
class="p-2.5 text-slate-500 hover:text-indigo-600 hover:bg-slate-100 rounded-xl transition shrink-0 cursor-pointer">
|
||||||
|
<i class="fa-solid fa-paperclip text-sm"></i>
|
||||||
|
</button>
|
||||||
|
|
||||||
|
<div class="flex-1 bg-slate-100 border border-slate-300 focus-within:border-indigo-600 focus-within:bg-white rounded-2xl p-1.5 transition flex items-center shadow-inner">
|
||||||
|
<textarea id="user-input" rows="1"
|
||||||
|
placeholder="Команда, вопрос или перетащите файл сюда..."
|
||||||
|
class="w-full bg-transparent px-2 text-xs sm:text-sm text-slate-800 focus:outline-none resize-none overflow-y-auto leading-relaxed"
|
||||||
|
style="height: 24px; max-height: 120px;"></textarea>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button type="submit" id="send-btn"
|
||||||
|
class="bg-indigo-600 hover:bg-indigo-700 active:bg-indigo-800 text-white p-2.5 rounded-xl transition shrink-0 shadow-sm cursor-pointer">
|
||||||
|
<i class="fa-solid fa-paper-plane text-sm"></i>
|
||||||
|
</button>
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</main>
|
||||||
|
|
||||||
|
<!-- Скрипты клиентской логики -->
|
||||||
|
<script src="/js/auth.js"></script>
|
||||||
|
<script src="/js/tasks.js"></script>
|
||||||
|
<script src="/js/chat/task_widget.js"></script>
|
||||||
|
<script src="/js/chat/core.js"></script>
|
||||||
|
<script src="/js/app.js"></script>
|
||||||
</body>
|
</body>
|
||||||
</html>
|
</html>
|
||||||
@@ -1,308 +1,62 @@
|
|||||||
function showAuthModal() {
|
/**
|
||||||
const el = document.getElementById("auth-modal");
|
* ===============================================================================
|
||||||
if (el) el.classList.remove("hidden");
|
* FILE: static/js/auth.js
|
||||||
}
|
* ROLE: Управление сессией пользователя, токенами и корректным выходом (Logout).
|
||||||
|
* ===============================================================================
|
||||||
|
*/
|
||||||
|
|
||||||
function hideAuthModal() {
|
const AuthManager = {
|
||||||
const el = document.getElementById("auth-modal");
|
getToken() {
|
||||||
if (el) el.classList.add("hidden");
|
let token = localStorage.getItem("auth_token") || localStorage.getItem("token");
|
||||||
}
|
if (!token) {
|
||||||
|
// Если токена нет — инициализируем рабочий дефолтный токен
|
||||||
|
token = "dev_token_1";
|
||||||
|
localStorage.setItem("auth_token", token);
|
||||||
|
localStorage.setItem("user_id", "1");
|
||||||
|
}
|
||||||
|
return token;
|
||||||
|
},
|
||||||
|
|
||||||
async function handleLogin(e) {
|
getUserId() {
|
||||||
if (e && e.preventDefault) e.preventDefault();
|
return parseInt(localStorage.getItem("user_id") || "1", 10);
|
||||||
|
},
|
||||||
const usernameInput = document.getElementById("auth-username-input");
|
|
||||||
const passwordInput = document.getElementById("auth-password-input");
|
|
||||||
const errorEl = document.getElementById("auth-error");
|
|
||||||
|
|
||||||
if (!usernameInput || !passwordInput) return;
|
logout() {
|
||||||
|
console.log("[Auth] Выполняется выход из учетной записи...");
|
||||||
|
|
||||||
const username = usernameInput.value.trim();
|
// 1. Полная очистка хранилищ браузера
|
||||||
const password = passwordInput.value;
|
localStorage.removeItem("auth_token");
|
||||||
|
localStorage.removeItem("token");
|
||||||
|
localStorage.removeItem("user_id");
|
||||||
|
sessionStorage.clear();
|
||||||
|
|
||||||
if (!username || !password) return;
|
// 2. Сброс авторизационных cookies (если присутствуют)
|
||||||
|
document.cookie.split(";").forEach((cookie) => {
|
||||||
if (errorEl) errorEl.classList.add("hidden");
|
const eqPos = cookie.indexOf("=");
|
||||||
|
const name = eqPos > -1 ? cookie.substr(0, eqPos).trim() : cookie.trim();
|
||||||
try {
|
document.cookie = `${name}=;expires=Thu, 01 Jan 1970 00:00:00 GMT;path=/`;
|
||||||
const res = await fetch("/api/v1/auth/login", {
|
|
||||||
method: "POST",
|
|
||||||
headers: { "Content-Type": "application/json" },
|
|
||||||
body: JSON.stringify({ username, password })
|
|
||||||
});
|
});
|
||||||
|
|
||||||
const data = await res.json();
|
// 3. Перезагрузка страницы для сброса состояния интерфейса
|
||||||
|
window.location.reload();
|
||||||
|
},
|
||||||
|
|
||||||
if (res.status === 200) {
|
init() {
|
||||||
// Используем прямой строковый ключ, чтобы избежать ошибки ReferenceError
|
// Привязка клика ко всем элементам с классом или id logout
|
||||||
API_TOKEN = data.token;
|
document.addEventListener("DOMContentLoaded", () => {
|
||||||
CURRENT_USERNAME = data.username;
|
const logoutBtns = document.querySelectorAll("#logout-btn, .logout-btn, [data-action='logout']");
|
||||||
IS_ADMIN = data.is_admin;
|
logoutBtns.forEach(btn => {
|
||||||
IS_GUEST = false;
|
btn.addEventListener("click", (e) => {
|
||||||
|
e.preventDefault();
|
||||||
localStorage.setItem("scud_api_auth_token", data.token);
|
AuthManager.logout();
|
||||||
localStorage.setItem("scud_username", data.username);
|
});
|
||||||
localStorage.setItem("scud_is_admin", data.is_admin ? "true" : "false");
|
});
|
||||||
localStorage.removeItem("scud_is_guest");
|
|
||||||
|
|
||||||
hideAuthModal();
|
|
||||||
updateUIState();
|
|
||||||
|
|
||||||
if (typeof loadTasks === 'function') {
|
|
||||||
loadTasks();
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if (errorEl) {
|
|
||||||
errorEl.innerText = data.detail || "Ошибка авторизации";
|
|
||||||
errorEl.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
console.error("[Auth Error]", err);
|
|
||||||
if (errorEl) {
|
|
||||||
errorEl.innerText = "Ошибка соединения с сервером";
|
|
||||||
errorEl.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function enableGuestMode() {
|
|
||||||
IS_GUEST = true;
|
|
||||||
API_TOKEN = "";
|
|
||||||
CURRENT_USERNAME = "Гость";
|
|
||||||
IS_ADMIN = false;
|
|
||||||
localStorage.setItem("scud_is_guest", "true");
|
|
||||||
hideAuthModal();
|
|
||||||
updateUIState();
|
|
||||||
}
|
|
||||||
|
|
||||||
function logout() {
|
|
||||||
localStorage.removeItem("scud_api_auth_token");
|
|
||||||
localStorage.removeItem("scud_username");
|
|
||||||
localStorage.removeItem("scud_is_admin");
|
|
||||||
localStorage.removeItem("scud_is_guest");
|
|
||||||
API_TOKEN = "";
|
|
||||||
CURRENT_USERNAME = "";
|
|
||||||
IS_ADMIN = false;
|
|
||||||
IS_GUEST = false;
|
|
||||||
showAuthModal();
|
|
||||||
}
|
|
||||||
|
|
||||||
function updateUIState() {
|
|
||||||
const tasksBtn = document.getElementById("tasks-drawer-btn");
|
|
||||||
const adminBtn = document.getElementById("admin-users-btn");
|
|
||||||
const changePwdBtn = document.getElementById("change-pwd-btn");
|
|
||||||
const guestBadge = document.getElementById("guest-badge");
|
|
||||||
const usernameBadge = document.getElementById("username-badge");
|
|
||||||
|
|
||||||
if (typeof IS_GUEST !== 'undefined' && IS_GUEST) {
|
|
||||||
if (tasksBtn) tasksBtn.classList.add("hidden");
|
|
||||||
if (adminBtn) adminBtn.classList.add("hidden");
|
|
||||||
if (changePwdBtn) changePwdBtn.classList.add("hidden");
|
|
||||||
if (guestBadge) guestBadge.classList.remove("hidden");
|
|
||||||
if (usernameBadge) usernameBadge.classList.add("hidden");
|
|
||||||
} else {
|
|
||||||
if (tasksBtn) tasksBtn.classList.remove("hidden");
|
|
||||||
if (changePwdBtn) changePwdBtn.classList.remove("hidden");
|
|
||||||
if (guestBadge) guestBadge.classList.add("hidden");
|
|
||||||
|
|
||||||
if (usernameBadge) {
|
|
||||||
usernameBadge.innerText = (typeof CURRENT_USERNAME !== 'undefined' && CURRENT_USERNAME) ? CURRENT_USERNAME : "User";
|
|
||||||
usernameBadge.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
|
|
||||||
if (adminBtn) {
|
|
||||||
const isAdminUser = (typeof IS_ADMIN !== 'undefined' && IS_ADMIN) || (typeof CURRENT_USERNAME !== 'undefined' && CURRENT_USERNAME === "puh");
|
|
||||||
if (isAdminUser) {
|
|
||||||
adminBtn.classList.remove("hidden");
|
|
||||||
} else {
|
|
||||||
adminBtn.classList.add("hidden");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function openChangePasswordModal() {
|
|
||||||
const el = document.getElementById("change-pwd-modal");
|
|
||||||
if (el) el.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
|
|
||||||
function closeChangePasswordModal() {
|
|
||||||
const el = document.getElementById("change-pwd-modal");
|
|
||||||
if (el) el.classList.add("hidden");
|
|
||||||
|
|
||||||
const err = document.getElementById("pwd-error");
|
|
||||||
const succ = document.getElementById("pwd-success");
|
|
||||||
if (err) err.classList.add("hidden");
|
|
||||||
if (succ) succ.classList.add("hidden");
|
|
||||||
|
|
||||||
document.getElementById("old-pwd-input").value = "";
|
|
||||||
document.getElementById("new-pwd-input").value = "";
|
|
||||||
const confirmInput = document.getElementById("confirm-pwd-input");
|
|
||||||
if (confirmInput) confirmInput.value = "";
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleChangePassword(e) {
|
|
||||||
if (e && e.preventDefault) e.preventDefault();
|
|
||||||
const old_password = document.getElementById("old-pwd-input").value;
|
|
||||||
const new_password = document.getElementById("new-pwd-input").value;
|
|
||||||
const confirmInput = document.getElementById("confirm-pwd-input");
|
|
||||||
const confirm_password = confirmInput ? confirmInput.value : new_password;
|
|
||||||
const errorEl = document.getElementById("pwd-error");
|
|
||||||
const successEl = document.getElementById("pwd-success");
|
|
||||||
|
|
||||||
if (errorEl) errorEl.classList.add("hidden");
|
|
||||||
if (successEl) successEl.classList.add("hidden");
|
|
||||||
|
|
||||||
if (new_password !== confirm_password) {
|
|
||||||
if (errorEl) {
|
|
||||||
errorEl.innerText = "Новые пароли не совпадают";
|
|
||||||
errorEl.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const token = typeof API_TOKEN !== 'undefined' ? API_TOKEN : localStorage.getItem("scud_api_auth_token");
|
|
||||||
const res = await fetch("/api/v1/auth/change-password", {
|
|
||||||
method: "POST",
|
|
||||||
headers: {
|
|
||||||
"Content-Type": "application/json",
|
|
||||||
"Authorization": "Bearer " + token
|
|
||||||
},
|
|
||||||
body: JSON.stringify({ old_password, new_password })
|
|
||||||
});
|
});
|
||||||
|
|
||||||
const data = await res.json();
|
|
||||||
|
|
||||||
if (res.status === 200) {
|
|
||||||
if (successEl) {
|
|
||||||
successEl.innerText = "Пароль успешно изменен!";
|
|
||||||
successEl.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
setTimeout(closeChangePasswordModal, 1500);
|
|
||||||
} else {
|
|
||||||
if (errorEl) {
|
|
||||||
errorEl.innerText = data.detail || "Ошибка при смене пароля";
|
|
||||||
errorEl.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
if (errorEl) {
|
|
||||||
errorEl.innerText = "Ошибка соединения с сервером";
|
|
||||||
errorEl.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
function openAdminModal() {
|
// Инициализация при загрузке скрипта
|
||||||
const el = document.getElementById("admin-modal");
|
AuthManager.init();
|
||||||
if (el) el.classList.remove("hidden");
|
|
||||||
loadUsersList();
|
|
||||||
}
|
|
||||||
|
|
||||||
function closeAdminModal() {
|
// Глобальная доступность функции для onclick в HTML
|
||||||
const el = document.getElementById("admin-modal");
|
window.logout = () => AuthManager.logout();
|
||||||
if (el) el.classList.add("hidden");
|
|
||||||
}
|
|
||||||
|
|
||||||
async function loadUsersList() {
|
|
||||||
const listEl = document.getElementById("admin-users-list");
|
|
||||||
if (!listEl) return;
|
|
||||||
listEl.innerHTML = '<div class="text-xs text-slate-400 py-4 text-center">Загрузка пользователей...</div>';
|
|
||||||
|
|
||||||
try {
|
|
||||||
const token = typeof API_TOKEN !== 'undefined' ? API_TOKEN : localStorage.getItem("scud_api_auth_token");
|
|
||||||
const res = await fetch("/api/v1/admin/users", {
|
|
||||||
headers: { "Authorization": "Bearer " + token }
|
|
||||||
});
|
|
||||||
const users = await res.json();
|
|
||||||
|
|
||||||
if (res.status === 200) {
|
|
||||||
listEl.innerHTML = users.map(u => {
|
|
||||||
const adminTag = u.is_admin ? '<span class="ml-1.5 text-[9px] bg-indigo-100 text-indigo-700 px-1.5 py-0.5 rounded font-bold">ADMIN</span>' : '<span class="ml-1.5 text-[9px] bg-slate-100 text-slate-600 px-1.5 py-0.5 rounded">USER</span>';
|
|
||||||
const fullNameHtml = u.full_name ? `<div class="text-[11px] text-slate-500 font-normal">${u.full_name}</div>` : '';
|
|
||||||
const dateStr = u.created_at ? u.created_at.split(' ')[0] : '—';
|
|
||||||
const deleteBtn = u.username !== CURRENT_USERNAME ? `<button type="button" onclick="deleteUser(${u.id}, '${u.username}')" class="text-red-500 hover:text-red-700 p-1"><i class="fa-solid fa-trash-can"></i></button>` : '<span class="text-[10px] text-slate-400">Вы</span>';
|
|
||||||
|
|
||||||
return `
|
|
||||||
<div class="flex justify-between items-center bg-slate-50 border border-slate-200 p-2.5 rounded-xl text-xs">
|
|
||||||
<div>
|
|
||||||
<div class="flex items-center">
|
|
||||||
<span class="font-bold text-slate-800">${u.username}</span>
|
|
||||||
${adminTag}
|
|
||||||
</div>
|
|
||||||
${fullNameHtml}
|
|
||||||
<div class="text-[10px] text-slate-400 mt-0.5">Создан: ${dateStr}</div>
|
|
||||||
</div>
|
|
||||||
${deleteBtn}
|
|
||||||
</div>
|
|
||||||
`;
|
|
||||||
}).join("");
|
|
||||||
} else {
|
|
||||||
listEl.innerHTML = `<div class="text-xs text-red-500 py-2">${users.detail}</div>`;
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
listEl.innerHTML = '<div class="text-xs text-red-500 py-2">Ошибка загрузки пользователей</div>';
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function handleCreateUser(e) {
|
|
||||||
if (e && e.preventDefault) e.preventDefault();
|
|
||||||
const username = document.getElementById("new-user-name").value.trim();
|
|
||||||
const password = document.getElementById("new-user-pwd").value;
|
|
||||||
const fullNameInput = document.getElementById("new-user-fullname");
|
|
||||||
const full_name = fullNameInput ? fullNameInput.value.trim() : "";
|
|
||||||
const adminCheckbox = document.getElementById("new-user-is-admin");
|
|
||||||
const is_admin = adminCheckbox ? adminCheckbox.checked : false;
|
|
||||||
const msgEl = document.getElementById("admin-msg");
|
|
||||||
|
|
||||||
if (msgEl) msgEl.classList.add("hidden");
|
|
||||||
|
|
||||||
try {
|
|
||||||
const token = typeof API_TOKEN !== 'undefined' ? API_TOKEN : localStorage.getItem("scud_api_auth_token");
|
|
||||||
const res = await fetch("/api/v1/admin/users", {
|
|
||||||
method: "POST",
|
|
||||||
headers: {
|
|
||||||
"Content-Type": "application/json",
|
|
||||||
"Authorization": "Bearer " + token
|
|
||||||
},
|
|
||||||
body: JSON.stringify({ username, password, full_name, is_admin })
|
|
||||||
});
|
|
||||||
|
|
||||||
const data = await res.json();
|
|
||||||
|
|
||||||
if (res.status === 200) {
|
|
||||||
document.getElementById("new-user-name").value = "";
|
|
||||||
document.getElementById("new-user-pwd").value = "";
|
|
||||||
if (fullNameInput) fullNameInput.value = "";
|
|
||||||
if (adminCheckbox) adminCheckbox.checked = false;
|
|
||||||
loadUsersList();
|
|
||||||
} else {
|
|
||||||
if (msgEl) {
|
|
||||||
msgEl.innerText = data.detail || "Ошибка";
|
|
||||||
msgEl.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch (err) {
|
|
||||||
if (msgEl) {
|
|
||||||
msgEl.innerText = "Ошибка связи с сервером";
|
|
||||||
msgEl.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function deleteUser(userId, username) {
|
|
||||||
if (!confirm("Удалить пользователя " + username + "?")) return;
|
|
||||||
|
|
||||||
try {
|
|
||||||
const token = typeof API_TOKEN !== 'undefined' ? API_TOKEN : localStorage.getItem("scud_api_auth_token");
|
|
||||||
await fetch("/api/v1/admin/users/" + userId, {
|
|
||||||
method: "DELETE",
|
|
||||||
headers: { "Authorization": "Bearer " + token }
|
|
||||||
});
|
|
||||||
loadUsersList();
|
|
||||||
} catch (err) {
|
|
||||||
alert("Ошибка при удалении");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,321 +0,0 @@
|
|||||||
// Вспомогательная функция для автоматического изменения высоты текстового поля
|
|
||||||
function updateInputHeight(el) {
|
|
||||||
if (!el) return;
|
|
||||||
el.style.height = "24px";
|
|
||||||
const newHeight = Math.min(el.scrollHeight, 120);
|
|
||||||
el.style.height = newHeight + "px";
|
|
||||||
}
|
|
||||||
|
|
||||||
let selectedFile = null;
|
|
||||||
|
|
||||||
function handleFileSelect(e) {
|
|
||||||
const file = e.target.files[0];
|
|
||||||
if (!file) return;
|
|
||||||
|
|
||||||
if (file.size > 15 * 1024 * 1024) {
|
|
||||||
alert("Файл слишком большой. Максимальный размер: 15 МБ");
|
|
||||||
e.target.value = "";
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
selectedFile = file;
|
|
||||||
const fileNameEl = document.getElementById("file-name-display");
|
|
||||||
const fileSizeEl = document.getElementById("file-size-display");
|
|
||||||
const previewContainer = document.getElementById("file-preview-container");
|
|
||||||
|
|
||||||
if (fileNameEl) fileNameEl.innerText = file.name;
|
|
||||||
if (fileSizeEl) fileSizeEl.innerText = `(${(file.size / 1024).toFixed(1)} KB)`;
|
|
||||||
if (previewContainer) previewContainer.classList.remove("hidden");
|
|
||||||
}
|
|
||||||
|
|
||||||
function clearAttachedFile() {
|
|
||||||
selectedFile = null;
|
|
||||||
const fileInput = document.getElementById("file-input");
|
|
||||||
const previewContainer = document.getElementById("file-preview-container");
|
|
||||||
if (fileInput) fileInput.value = "";
|
|
||||||
if (previewContainer) previewContainer.classList.add("hidden");
|
|
||||||
}
|
|
||||||
|
|
||||||
// Деактивация всех старых кнопок в истории
|
|
||||||
function disableAllActionButtons() {
|
|
||||||
const allBtnContainers = document.querySelectorAll(".action-buttons-container");
|
|
||||||
allBtnContainers.forEach(container => {
|
|
||||||
container.querySelectorAll("button").forEach(btn => {
|
|
||||||
btn.disabled = true;
|
|
||||||
btn.classList.add("opacity-40", "cursor-not-allowed");
|
|
||||||
});
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Быстрая отправка текста кнопки
|
|
||||||
function handleActionButtonClick(text) {
|
|
||||||
disableAllActionButtons();
|
|
||||||
const input = document.getElementById("user-input");
|
|
||||||
if (input) {
|
|
||||||
input.value = text;
|
|
||||||
sendMessage();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function sendMessage(e) {
|
|
||||||
if (e && e.preventDefault) e.preventDefault();
|
|
||||||
|
|
||||||
const input = document.getElementById("user-input");
|
|
||||||
const chatWindow = document.getElementById("chat-window");
|
|
||||||
const sendBtn = document.getElementById("send-btn");
|
|
||||||
|
|
||||||
if (!input || !chatWindow) return;
|
|
||||||
const text = input.value.trim();
|
|
||||||
|
|
||||||
if (!text && !selectedFile) return;
|
|
||||||
|
|
||||||
// Деактивируем предыдущие интерактивные кнопки
|
|
||||||
disableAllActionButtons();
|
|
||||||
|
|
||||||
let userDisplayHtml = escapeHtml(text);
|
|
||||||
if (selectedFile) {
|
|
||||||
userDisplayHtml = `<div class="font-bold border-b border-indigo-400/40 pb-1 mb-1 text-[11px] flex items-center gap-1.5">
|
|
||||||
<i class="fa-solid fa-file"></i> ${escapeHtml(selectedFile.name)}
|
|
||||||
</div>` + userDisplayHtml;
|
|
||||||
}
|
|
||||||
|
|
||||||
const userMsgHtml = `
|
|
||||||
<div class="flex justify-end mb-3">
|
|
||||||
<div class="bg-indigo-600 text-white rounded-2xl px-4 py-2.5 max-w-2xl text-xs sm:text-sm shadow-sm">
|
|
||||||
${userDisplayHtml}
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
`;
|
|
||||||
chatWindow.insertAdjacentHTML("beforeend", userMsgHtml);
|
|
||||||
|
|
||||||
input.value = "";
|
|
||||||
updateInputHeight(input);
|
|
||||||
chatWindow.scrollTop = chatWindow.scrollHeight;
|
|
||||||
|
|
||||||
if (sendBtn) {
|
|
||||||
sendBtn.disabled = true;
|
|
||||||
sendBtn.classList.add("opacity-50");
|
|
||||||
}
|
|
||||||
|
|
||||||
const token = typeof API_TOKEN !== 'undefined' && API_TOKEN ? API_TOKEN : localStorage.getItem("scud_api_auth_token");
|
|
||||||
const isGuest = typeof IS_GUEST !== 'undefined' ? IS_GUEST : (localStorage.getItem("scud_is_guest") === "true");
|
|
||||||
|
|
||||||
const endpoint = isGuest ? "/api/v1/chat/guest" : "/api/v1/chat";
|
|
||||||
|
|
||||||
const formData = new FormData();
|
|
||||||
formData.append("session_id", "web_session_main");
|
|
||||||
formData.append("message", text || "Проанализируй прикрепленный файл");
|
|
||||||
|
|
||||||
if (selectedFile instanceof File) {
|
|
||||||
formData.append("file", selectedFile, selectedFile.name);
|
|
||||||
}
|
|
||||||
|
|
||||||
const headers = {};
|
|
||||||
if (!isGuest && token) {
|
|
||||||
headers["Authorization"] = "Bearer " + token;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const res = await fetch(endpoint, {
|
|
||||||
method: "POST",
|
|
||||||
headers: headers,
|
|
||||||
body: formData
|
|
||||||
});
|
|
||||||
|
|
||||||
if (res.status === 401 && !isGuest) {
|
|
||||||
if (typeof logout === 'function') logout();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
const data = await res.json();
|
|
||||||
const assistantTitle = isGuest ? "Локальная нейросеть (Гость)" : "ИИ-Ассистент SCUD Orion AI";
|
|
||||||
const replyText = data.reply || "Пустой ответ от нейросети";
|
|
||||||
|
|
||||||
// Генерация блока кнопок подтверждения при необходимости
|
|
||||||
let actionButtonsHtml = "";
|
|
||||||
const actionData = data.action_type || data.action;
|
|
||||||
|
|
||||||
if (actionData && actionData.buttons && actionData.buttons.length > 0) {
|
|
||||||
const buttonsMarkup = actionData.buttons.map(btn => {
|
|
||||||
let btnClasses = "bg-slate-100 hover:bg-slate-200 active:bg-slate-300 text-slate-700 border border-slate-300";
|
|
||||||
let iconMarkup = '<i class="fa-solid fa-arrow-right text-[10px] opacity-60"></i>';
|
|
||||||
|
|
||||||
const labelLower = (btn.label || "").toLowerCase();
|
|
||||||
const isPrimary = btn.style === "primary" || labelLower.includes("подтверд") || labelLower.startsWith("да");
|
|
||||||
const isDanger = btn.style === "danger" || labelLower.includes("отмен") || labelLower.startsWith("нет") || labelLower.includes("законч") || labelLower.includes("заверш");
|
|
||||||
|
|
||||||
if (isPrimary) {
|
|
||||||
btnClasses = "bg-emerald-600 hover:bg-emerald-700 active:bg-emerald-800 text-white shadow-sm";
|
|
||||||
iconMarkup = '<i class="fa-solid fa-check"></i>';
|
|
||||||
} else if (isDanger) {
|
|
||||||
btnClasses = "bg-rose-50 hover:bg-rose-100 active:bg-rose-200 text-rose-700 border border-rose-300";
|
|
||||||
iconMarkup = '<i class="fa-solid fa-xmark"></i>';
|
|
||||||
}
|
|
||||||
|
|
||||||
return `
|
|
||||||
<button type="button" onclick="handleActionButtonClick('${escapeHtml(btn.value)}')"
|
|
||||||
class="${btnClasses} font-semibold px-3 py-1.5 rounded-xl text-xs flex items-center gap-1.5 transition cursor-pointer">
|
|
||||||
${iconMarkup}
|
|
||||||
<span>${escapeHtml(btn.label)}</span>
|
|
||||||
</button>
|
|
||||||
`;
|
|
||||||
}).join("");
|
|
||||||
|
|
||||||
actionButtonsHtml = `
|
|
||||||
<div class="action-buttons-container flex flex-wrap items-center gap-2 mt-3 pt-2.5 border-t border-slate-100">
|
|
||||||
${buttonsMarkup}
|
|
||||||
</div>
|
|
||||||
`;
|
|
||||||
}
|
|
||||||
|
|
||||||
const botMsgHtml = `
|
|
||||||
<div class="bg-white border border-slate-200 rounded-2xl p-3.5 shadow-sm max-w-2xl mb-3">
|
|
||||||
<p class="text-[11px] font-bold text-indigo-600 uppercase tracking-wider mb-1">
|
|
||||||
<i class="fa-solid fa-robot mr-1"></i> ${assistantTitle}
|
|
||||||
</p>
|
|
||||||
<p class="text-slate-800 text-xs sm:text-sm whitespace-pre-wrap leading-relaxed">${escapeHtml(replyText)}</p>
|
|
||||||
${actionButtonsHtml}
|
|
||||||
</div>
|
|
||||||
`;
|
|
||||||
chatWindow.insertAdjacentHTML("beforeend", botMsgHtml);
|
|
||||||
chatWindow.scrollTop = chatWindow.scrollHeight;
|
|
||||||
|
|
||||||
clearAttachedFile();
|
|
||||||
|
|
||||||
if (!isGuest && typeof loadTasks === 'function') {
|
|
||||||
loadTasks();
|
|
||||||
}
|
|
||||||
|
|
||||||
} catch (err) {
|
|
||||||
console.error("[Chat Error]", err);
|
|
||||||
const errorHtml = `
|
|
||||||
<div class="bg-red-50 border border-red-200 rounded-2xl p-3.5 max-w-2xl text-red-700 text-xs sm:text-sm mb-3">
|
|
||||||
Ошибка связи с сервером.
|
|
||||||
</div>
|
|
||||||
`;
|
|
||||||
chatWindow.insertAdjacentHTML("beforeend", errorHtml);
|
|
||||||
chatWindow.scrollTop = chatWindow.scrollHeight;
|
|
||||||
} finally {
|
|
||||||
if (sendBtn) {
|
|
||||||
sendBtn.disabled = false;
|
|
||||||
sendBtn.classList.remove("opacity-50");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function escapeHtml(text) {
|
|
||||||
if (!text) return "";
|
|
||||||
return text
|
|
||||||
.replace(/&/g, "&")
|
|
||||||
.replace(/</g, "<")
|
|
||||||
.replace(/>/g, ">")
|
|
||||||
.replace(/"/g, """)
|
|
||||||
.replace(/'/g, "'");
|
|
||||||
}
|
|
||||||
|
|
||||||
document.addEventListener("DOMContentLoaded", () => {
|
|
||||||
const input = document.getElementById("user-input");
|
|
||||||
const dropZone = document.getElementById("chat-window")?.parentElement;
|
|
||||||
const dropOverlay = document.getElementById("drop-overlay");
|
|
||||||
|
|
||||||
if (input) {
|
|
||||||
let historyIndex = -1;
|
|
||||||
let localHistory = JSON.parse(localStorage.getItem("scud_chat_input_history") || "[]");
|
|
||||||
|
|
||||||
input.addEventListener("keydown", (e) => {
|
|
||||||
if (e.key === "Enter" && !e.shiftKey) {
|
|
||||||
e.preventDefault();
|
|
||||||
const text = input.value.trim();
|
|
||||||
if (text) {
|
|
||||||
if (localHistory.length === 0 || localHistory[0] !== text) {
|
|
||||||
localHistory.unshift(text);
|
|
||||||
if (localHistory.length > 50) localHistory.pop();
|
|
||||||
localStorage.setItem("scud_chat_input_history", JSON.stringify(localHistory));
|
|
||||||
}
|
|
||||||
historyIndex = -1;
|
|
||||||
}
|
|
||||||
sendMessage(e);
|
|
||||||
updateInputHeight(input);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (e.key === "ArrowUp") {
|
|
||||||
const textBeforeCursor = input.value.substring(0, input.selectionStart);
|
|
||||||
const isFirstLine = !textBeforeCursor.includes("\n");
|
|
||||||
|
|
||||||
if (isFirstLine && input.selectionStart === 0 && localHistory.length > 0) {
|
|
||||||
if (historyIndex < localHistory.length - 1) {
|
|
||||||
e.preventDefault();
|
|
||||||
if (historyIndex === -1) {
|
|
||||||
input.dataset.draft = input.value;
|
|
||||||
}
|
|
||||||
historyIndex++;
|
|
||||||
input.value = localHistory[historyIndex];
|
|
||||||
updateInputHeight(input);
|
|
||||||
input.setSelectionRange(input.value.length, input.value.length);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (e.key === "ArrowDown") {
|
|
||||||
const textAfterCursor = input.value.substring(input.selectionEnd);
|
|
||||||
const isLastLine = !textAfterCursor.includes("\n");
|
|
||||||
|
|
||||||
if (isLastLine && input.selectionEnd === input.value.length && historyIndex >= 0) {
|
|
||||||
e.preventDefault();
|
|
||||||
if (historyIndex > 0) {
|
|
||||||
historyIndex--;
|
|
||||||
input.value = localHistory[historyIndex];
|
|
||||||
} else {
|
|
||||||
historyIndex = -1;
|
|
||||||
input.value = input.dataset.draft || "";
|
|
||||||
}
|
|
||||||
updateInputHeight(input);
|
|
||||||
input.setSelectionRange(input.value.length, input.value.length);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
if (dropZone && dropOverlay) {
|
|
||||||
["dragenter", "dragover", "dragleave", "drop"].forEach(eventName => {
|
|
||||||
dropZone.addEventListener(eventName, (e) => {
|
|
||||||
e.preventDefault();
|
|
||||||
e.stopPropagation();
|
|
||||||
}, false);
|
|
||||||
});
|
|
||||||
|
|
||||||
["dragenter", "dragover"].forEach(eventName => {
|
|
||||||
dropZone.addEventListener(eventName, () => {
|
|
||||||
dropOverlay.classList.remove("hidden");
|
|
||||||
dropOverlay.classList.add("flex");
|
|
||||||
}, false);
|
|
||||||
});
|
|
||||||
|
|
||||||
["dragleave", "drop"].forEach(eventName => {
|
|
||||||
dropZone.addEventListener(eventName, (e) => {
|
|
||||||
if (eventName === "drop" || e.target === dropZone || !dropZone.contains(e.relatedTarget)) {
|
|
||||||
dropOverlay.classList.add("hidden");
|
|
||||||
dropOverlay.classList.remove("flex");
|
|
||||||
}
|
|
||||||
}, false);
|
|
||||||
});
|
|
||||||
|
|
||||||
dropZone.addEventListener("drop", (e) => {
|
|
||||||
const dt = e.dataTransfer;
|
|
||||||
const files = dt.files;
|
|
||||||
|
|
||||||
if (files && files.length > 0) {
|
|
||||||
const file = files[0];
|
|
||||||
handleFileSelect({ target: { files: [file] } });
|
|
||||||
|
|
||||||
const fileInput = document.getElementById("file-input");
|
|
||||||
if (fileInput) {
|
|
||||||
const dataTransfer = new DataTransfer();
|
|
||||||
dataTransfer.items.add(file);
|
|
||||||
fileInput.files = dataTransfer.files;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}, false);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
@@ -0,0 +1,67 @@
|
|||||||
|
/**
|
||||||
|
* ===============================================================================
|
||||||
|
* FILE: static/js/chat/core.js
|
||||||
|
* ROLE: Отправка сообщений в API с токеном и обработка ошибок.
|
||||||
|
* ===============================================================================
|
||||||
|
*/
|
||||||
|
|
||||||
|
async function sendMessage(userMessageText) {
|
||||||
|
const chatInput = document.getElementById("chat-input");
|
||||||
|
const message = userMessageText || (chatInput ? chatInput.value.trim() : "");
|
||||||
|
if (!message) return;
|
||||||
|
|
||||||
|
if (chatInput && !userMessageText) {
|
||||||
|
chatInput.value = "";
|
||||||
|
}
|
||||||
|
|
||||||
|
// Отображаем сообщение пользователя в чате
|
||||||
|
if (typeof appendMessageToUI === "function") {
|
||||||
|
appendMessageToUI("user", message);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Подготовка заголовков с гарантированным токеном
|
||||||
|
const token = (typeof AuthManager !== "undefined") ? AuthManager.getToken() : (localStorage.getItem("auth_token") || "dev_token_1");
|
||||||
|
const userId = (typeof AuthManager !== "undefined") ? AuthManager.getUserId() : parseInt(localStorage.getItem("user_id") || "1", 10);
|
||||||
|
const sessionId = localStorage.getItem("chat_session_id") || "web_session_main";
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await fetch("/api/v1/chat", {
|
||||||
|
method: "POST",
|
||||||
|
headers: {
|
||||||
|
"Content-Type": "application/json",
|
||||||
|
"Authorization": `Bearer ${token}`
|
||||||
|
},
|
||||||
|
body: JSON.stringify({
|
||||||
|
message: message,
|
||||||
|
session_id: sessionId,
|
||||||
|
user_id: userId
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
if (response.status === 403) {
|
||||||
|
console.warn("[Chat] Получен 403 Forbidden. Сбрасываем сессию и повторяем...");
|
||||||
|
localStorage.removeItem("auth_token");
|
||||||
|
if (typeof appendMessageToUI === "function") {
|
||||||
|
appendMessageToUI("assistant", "⚠️ Сессия была обновлена. Пожалуйста, отправьте сообщение повторно.");
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!response.ok) {
|
||||||
|
throw new Error(`Ошибка сервера (HTTP ${response.status})`);
|
||||||
|
}
|
||||||
|
|
||||||
|
const data = await response.json();
|
||||||
|
const reply = data.response || data.message || "Ответ получен без текста.";
|
||||||
|
|
||||||
|
if (typeof appendMessageToUI === "function") {
|
||||||
|
appendMessageToUI("assistant", reply);
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (err) {
|
||||||
|
console.error("[Chat Error]:", err);
|
||||||
|
if (typeof appendMessageToUI === "function") {
|
||||||
|
appendMessageToUI("assistant", `⚠️ Не удалось связаться с сервером: ${err.message}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,445 @@
|
|||||||
|
/*
|
||||||
|
===============================================================================
|
||||||
|
FILE: modules/web_api/static/js/chat/task_widget.js
|
||||||
|
ROLE: Интерактивные виджеты задач и срезов СКУД (SNAPSHOTS_CARD) с чекбоксами.
|
||||||
|
===============================================================================
|
||||||
|
*/
|
||||||
|
|
||||||
|
window.activeTaskFilter = window.activeTaskFilter || 'IN_PROGRESS';
|
||||||
|
window.currentTasksCache = window.currentTasksCache || [];
|
||||||
|
|
||||||
|
function getAuthHeaders() {
|
||||||
|
const token = typeof API_TOKEN !== 'undefined' && API_TOKEN ? API_TOKEN : localStorage.getItem("scud_api_auth_token");
|
||||||
|
const headers = { 'Content-Type': 'application/json' };
|
||||||
|
if (token) {
|
||||||
|
headers['Authorization'] = 'Bearer ' + token;
|
||||||
|
}
|
||||||
|
return headers;
|
||||||
|
}
|
||||||
|
|
||||||
|
window.sendChatAction = function(actionText) {
|
||||||
|
const input = document.getElementById("user-input");
|
||||||
|
if (input && typeof sendMessage === "function") {
|
||||||
|
input.value = actionText;
|
||||||
|
if (typeof setInputLocked === "function") setInputLocked(false);
|
||||||
|
sendMessage();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// РЕНДЕРИНГ КАРТОЧКИ СРЕЗОВ СКУД (SNAPSHOTS_CARD) С ЧЕКБОКСАМИ
|
||||||
|
// ============================================================================
|
||||||
|
function renderSnapshotsCard(data) {
|
||||||
|
if (!data || !data.snapshots || !Array.isArray(data.snapshots)) return '';
|
||||||
|
const queryDate = data.query_date || 'выбранную дату';
|
||||||
|
const snapshots = data.snapshots;
|
||||||
|
|
||||||
|
if (snapshots.length === 0) {
|
||||||
|
return `
|
||||||
|
<div class="p-6 bg-slate-50 border border-slate-200 rounded-xl text-center text-xs text-slate-500 my-2">
|
||||||
|
📸 За дату <b>${queryDate}</b> сохраненных снапшотов не найдено.
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
|
||||||
|
const rowsHtml = snapshots.map((s, idx) => {
|
||||||
|
const snapId = s.snapshot_id || `ID-${idx}`;
|
||||||
|
const snapTime = s.snapshot_time ? s.snapshot_time.split(' ')[1] || s.snapshot_time : '—';
|
||||||
|
const count = s.record_count || 0;
|
||||||
|
const isFinal = snapId.startsWith('Y');
|
||||||
|
|
||||||
|
if (isFinal) {
|
||||||
|
return `
|
||||||
|
<div class="p-2.5 bg-purple-50/60 rounded-lg border border-purple-200 shadow-sm flex items-center justify-between gap-3">
|
||||||
|
<div class="flex items-center gap-2.5 min-w-0">
|
||||||
|
<span class="w-4 flex justify-center text-purple-400" title="Итоговый срез защищен">
|
||||||
|
<i class="fa-solid fa-lock text-[11px]"></i>
|
||||||
|
</span>
|
||||||
|
<span class="font-mono text-xs font-bold px-2 py-0.5 rounded bg-purple-100 text-purple-800 border border-purple-300">
|
||||||
|
#${snapId}
|
||||||
|
</span>
|
||||||
|
<div class="flex items-center gap-3 text-xs text-slate-600">
|
||||||
|
<span class="flex items-center gap-1 font-semibold text-purple-900">
|
||||||
|
<i class="fa-regular fa-clock text-purple-600 text-[11px]"></i> ${snapTime}
|
||||||
|
</span>
|
||||||
|
<span class="flex items-center gap-1 text-slate-500">
|
||||||
|
<i class="fa-solid fa-users text-slate-400 text-[11px]"></i> ${count} записей
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<span class="text-[10px] font-bold text-purple-700 bg-purple-100 border border-purple-200 px-2 py-0.5 rounded">
|
||||||
|
Итоговый Y-срез
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
|
||||||
|
return `
|
||||||
|
<div class="p-2.5 bg-white rounded-lg border border-slate-200 shadow-sm hover:border-slate-300 transition flex items-center justify-between gap-3">
|
||||||
|
<div class="flex items-center gap-2.5 min-w-0">
|
||||||
|
<input type="checkbox" value="${snapId}" onchange="window.updateSelectedSnapshots(this)"
|
||||||
|
class="snapshot-item-checkbox rounded border-slate-300 text-indigo-600 focus:ring-indigo-500 w-4 h-4 cursor-pointer">
|
||||||
|
<span class="font-mono text-xs font-bold px-2 py-0.5 rounded bg-slate-100 text-slate-700 border border-slate-200">
|
||||||
|
#${snapId}
|
||||||
|
</span>
|
||||||
|
<div class="flex items-center gap-3 text-xs text-slate-600">
|
||||||
|
<span class="flex items-center gap-1 font-semibold text-slate-800">
|
||||||
|
<i class="fa-regular fa-clock text-indigo-500 text-[11px]"></i> ${snapTime}
|
||||||
|
</span>
|
||||||
|
<span class="flex items-center gap-1 text-slate-500">
|
||||||
|
<i class="fa-solid fa-users text-slate-400 text-[11px]"></i> ${count} записей
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<button type="button" onclick="window.sendChatAction('удали снапшот ${snapId}')"
|
||||||
|
class="p-1 text-slate-400 hover:text-rose-600 hover:bg-rose-50 rounded border border-transparent hover:border-rose-200 transition"
|
||||||
|
title="Удалить срез">
|
||||||
|
<i class="fa-solid fa-trash-can text-xs"></i>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}).join('');
|
||||||
|
|
||||||
|
return `
|
||||||
|
<div class="snapshots-widget-root w-full max-w-4xl mx-auto my-2 bg-slate-50 border border-slate-300 rounded-xl shadow-md overflow-hidden flex flex-col">
|
||||||
|
<div class="px-4 py-2.5 bg-white border-b border-slate-200 flex items-center justify-between gap-2 flex-wrap">
|
||||||
|
<div class="flex items-center gap-2">
|
||||||
|
<div class="bg-indigo-600 text-white p-1.5 rounded-lg flex items-center justify-center">
|
||||||
|
<i class="fa-solid fa-camera text-xs"></i>
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<span class="text-xs font-bold text-slate-800">Реестр срезов СКУД</span>
|
||||||
|
<span class="text-xs text-slate-500 ml-1">за ${queryDate}</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="flex items-center gap-3">
|
||||||
|
<label class="flex items-center gap-1.5 text-xs text-slate-600 cursor-pointer select-none">
|
||||||
|
<input type="checkbox" onchange="window.toggleSelectAllSnapshots(this)" class="select-all-snapshots-cb rounded border-slate-300 text-indigo-600 focus:ring-indigo-500 w-3.5 h-3.5">
|
||||||
|
<span>Выбрать все</span>
|
||||||
|
</label>
|
||||||
|
<span class="text-[11px] font-semibold bg-indigo-50 text-indigo-700 border border-indigo-200 px-2 py-0.5 rounded-full">
|
||||||
|
Срезов: ${snapshots.length}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="p-3 flex flex-col gap-2">
|
||||||
|
${rowsHtml}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- ПОДВАЛ С КНОПКОЙ УДАЛЕНИЯ ВЫБРАННЫХ -->
|
||||||
|
<div class="snapshot-bulk-actions-footer hidden px-4 py-2 bg-rose-50/70 border-t border-rose-200 flex items-center justify-between">
|
||||||
|
<span class="text-xs text-rose-800 font-medium bulk-selected-counter">Выбрано: 0</span>
|
||||||
|
<button type="button" onclick="window.submitBulkDeleteSnapshots(this)"
|
||||||
|
class="px-3 py-1.5 bg-rose-600 hover:bg-rose-700 active:bg-rose-800 text-white font-bold rounded-lg text-xs flex items-center gap-1.5 transition shadow-sm">
|
||||||
|
<i class="fa-solid fa-trash-can"></i>
|
||||||
|
<span>Удалить выбранные</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ⭐️ Обработчики чекбоксов
|
||||||
|
window.toggleSelectAllSnapshots = function(masterCb) {
|
||||||
|
const root = masterCb.closest('.snapshots-widget-root');
|
||||||
|
if (!root) return;
|
||||||
|
const checkboxes = root.querySelectorAll('.snapshot-item-checkbox');
|
||||||
|
checkboxes.forEach(cb => cb.checked = masterCb.checked);
|
||||||
|
window.syncBulkDeleteFooter(root);
|
||||||
|
};
|
||||||
|
|
||||||
|
window.updateSelectedSnapshots = function(itemCb) {
|
||||||
|
const root = itemCb.closest('.snapshots-widget-root');
|
||||||
|
if (!root) return;
|
||||||
|
window.syncBulkDeleteFooter(root);
|
||||||
|
};
|
||||||
|
|
||||||
|
window.syncBulkDeleteFooter = function(root) {
|
||||||
|
const checkboxes = root.querySelectorAll('.snapshot-item-checkbox:checked');
|
||||||
|
const footer = root.querySelector('.snapshot-bulk-actions-footer');
|
||||||
|
const counter = root.querySelector('.bulk-selected-counter');
|
||||||
|
const masterCb = root.querySelector('.select-all-snapshots-cb');
|
||||||
|
const allCheckboxes = root.querySelectorAll('.snapshot-item-checkbox');
|
||||||
|
|
||||||
|
if (masterCb) {
|
||||||
|
masterCb.checked = allCheckboxes.length > 0 && checkboxes.length === allCheckboxes.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (checkboxes.length > 0) {
|
||||||
|
if (footer) footer.classList.remove('hidden');
|
||||||
|
if (counter) counter.innerText = `Выбрано дневных срезов: ${checkboxes.length}`;
|
||||||
|
} else {
|
||||||
|
if (footer) footer.classList.add('hidden');
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Одиночная корзина в строке снапшота:
|
||||||
|
// onclick="window.sendChatAction('удали снапшот ${snapId}')"
|
||||||
|
|
||||||
|
// Кнопка пакетного удаления в подвале карточки:
|
||||||
|
window.submitBulkDeleteSnapshots = function(btnEl) {
|
||||||
|
const root = btnEl.closest('.snapshots-widget-root');
|
||||||
|
if (!root) return;
|
||||||
|
const selected = Array.from(root.querySelectorAll('.snapshot-item-checkbox:checked')).map(cb => cb.value);
|
||||||
|
if (selected.length === 0) return;
|
||||||
|
window.sendChatAction(`удали снапшоты ${selected.join(', ')}`);
|
||||||
|
};
|
||||||
|
|
||||||
|
// ============================================================================
|
||||||
|
// РЕНДЕРИНГ КАРТОЧКИ ЗАДАЧ (TASK_INTERACTIVE_CARD)
|
||||||
|
// ============================================================================
|
||||||
|
function renderInteractiveTaskCard(tasks) {
|
||||||
|
if (!tasks || !Array.isArray(tasks)) return '';
|
||||||
|
window.currentTasksCache = tasks;
|
||||||
|
|
||||||
|
const counts = {
|
||||||
|
ALL: tasks.length,
|
||||||
|
IN_PROGRESS: tasks.filter(t => t.status === 'IN_PROGRESS' || t.status === 'PROGRESS').length,
|
||||||
|
PLANNED: tasks.filter(t => t.status === 'BACKLOG' || t.status === 'PLANNED').length,
|
||||||
|
COMPLETED: tasks.filter(t => t.status === 'COMPLETED' || t.status === 'DONE').length
|
||||||
|
};
|
||||||
|
|
||||||
|
const currentFilter = window.activeTaskFilter || 'IN_PROGRESS';
|
||||||
|
|
||||||
|
const filteredTasks = tasks.filter(t => {
|
||||||
|
const s = (t.status || 'BACKLOG').toUpperCase();
|
||||||
|
if (currentFilter === 'ALL') return true;
|
||||||
|
if (currentFilter === 'IN_PROGRESS') return s === 'IN_PROGRESS' || s === 'PROGRESS';
|
||||||
|
if (currentFilter === 'PLANNED') return s === 'BACKLOG' || s === 'PLANNED';
|
||||||
|
if (currentFilter === 'COMPLETED') return s === 'COMPLETED' || s === 'DONE';
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
|
||||||
|
const getPrioBadge = (prio) => {
|
||||||
|
const p = (prio || 'MEDIUM').toUpperCase();
|
||||||
|
if (p === 'HIGH' || p === 'CRITICAL') return '<span class="text-[10px] px-2 py-0.5 rounded font-bold bg-rose-100 text-rose-700 border border-rose-200">🔥 HIGH</span>';
|
||||||
|
if (p === 'LOW') return '<span class="text-[10px] px-2 py-0.5 rounded font-semibold bg-slate-100 text-slate-600 border border-slate-200">☕ LOW</span>';
|
||||||
|
return '<span class="text-[10px] px-2 py-0.5 rounded font-semibold bg-amber-100 text-amber-700 border border-amber-200">⚡ MEDIUM</span>';
|
||||||
|
};
|
||||||
|
|
||||||
|
const getStatusBadge = (status) => {
|
||||||
|
const s = (status || 'BACKLOG').toUpperCase();
|
||||||
|
if (s === 'IN_PROGRESS' || s === 'PROGRESS') return '<span class="text-[10px] px-2 py-0.5 rounded font-bold bg-blue-50 text-blue-700 border border-blue-200">⚙️ В работе</span>';
|
||||||
|
if (s === 'COMPLETED' || s === 'DONE') return '<span class="text-[10px] px-2 py-0.5 rounded font-semibold bg-emerald-50 text-emerald-700 border border-emerald-200">✓ Готово</span>';
|
||||||
|
return '<span class="text-[10px] px-2 py-0.5 rounded font-semibold bg-slate-50 text-slate-600 border border-slate-200">📋 В планах</span>';
|
||||||
|
};
|
||||||
|
|
||||||
|
const taskRows = filteredTasks.map(t => {
|
||||||
|
const id = t.id;
|
||||||
|
const title = t.title || 'Без названия';
|
||||||
|
const isDone = t.status === 'COMPLETED' || t.status === 'DONE';
|
||||||
|
const isInProgress = t.status === 'IN_PROGRESS' || t.status === 'PROGRESS';
|
||||||
|
|
||||||
|
const actionBtn = isInProgress
|
||||||
|
? `<button type="button" onclick="window.sendChatAction('заверши задачу ${id}')" class="px-2.5 py-1 text-xs font-semibold rounded bg-emerald-50 text-emerald-700 border border-emerald-300 hover:bg-emerald-100 transition-colors shadow-sm" title="Завершить задачу">✓ Готово</button>`
|
||||||
|
: (!isDone
|
||||||
|
? `<button type="button" onclick="window.sendChatAction('возьми в работу задачу ${id}')" class="px-2.5 py-1 text-xs font-semibold rounded bg-blue-50 text-blue-700 border border-blue-300 hover:bg-blue-100 transition-colors shadow-sm" title="Взять в работу">⚙️ В работу</button>`
|
||||||
|
: '');
|
||||||
|
|
||||||
|
return `
|
||||||
|
<div id="task-card-${id}" class="p-3 bg-white rounded-lg border border-slate-200 shadow-sm hover:border-slate-300 transition-all flex flex-col gap-2">
|
||||||
|
<div class="task-view-mode flex items-center justify-between gap-3">
|
||||||
|
<div class="flex items-center gap-2 flex-wrap flex-1 min-w-0">
|
||||||
|
<span class="text-xs font-bold px-1.5 py-0.5 rounded bg-slate-100 text-slate-700 border border-slate-200">#${id}</span>
|
||||||
|
<span class="text-sm font-medium text-slate-900 truncate" title="${title}">${title}</span>
|
||||||
|
</div>
|
||||||
|
<div class="flex items-center gap-1.5 shrink-0">
|
||||||
|
${actionBtn}
|
||||||
|
<button type="button" onclick="window.openTaskInlineEditor(${id})" class="p-1 text-xs text-slate-500 hover:text-indigo-600 hover:bg-slate-50 rounded border border-slate-200 transition-colors" title="Редактировать">✏️</button>
|
||||||
|
<button type="button" onclick="window.sendChatAction('удали задачу ${id}')" class="p-1 text-xs text-slate-400 hover:text-rose-600 hover:bg-rose-50 rounded border border-slate-200 transition-colors" title="Удалить">🗑️</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="task-view-mode flex items-center gap-2 text-xs text-slate-500 flex-wrap">
|
||||||
|
${getPrioBadge(t.priority)}
|
||||||
|
${getStatusBadge(t.status)}
|
||||||
|
<span class="px-1.5 py-0.5 rounded bg-slate-50 border border-slate-200 text-slate-600 font-mono text-[11px]">${t.module || 'general'}</span>
|
||||||
|
${t.due_date ? `<span class="text-slate-500">📅 срок: <b>${t.due_date}</b></span>` : ''}
|
||||||
|
${t.created_at ? `<span class="text-slate-400">создана: ${t.created_at.split(' ')[0]}</span>` : ''}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- ФОРМА ИНЛАЙН РЕДАКТИРОВАНИЯ -->
|
||||||
|
<div id="task-editor-${id}" class="hidden flex flex-col gap-2 pt-2 border-t border-slate-100">
|
||||||
|
<input type="text" id="task-edit-title-${id}" value="${title.replace(/"/g, '"')}" class="w-full text-xs px-2.5 py-1.5 border border-slate-300 rounded focus:border-indigo-500 focus:outline-none bg-slate-50" placeholder="Описание задачи..." />
|
||||||
|
<div class="flex items-center gap-2 flex-wrap">
|
||||||
|
<input type="date" id="task-edit-date-${id}" value="${t.due_date || ''}" class="text-xs px-2 py-1 border border-slate-300 rounded focus:border-indigo-500 focus:outline-none bg-slate-50" />
|
||||||
|
<select id="task-edit-prio-${id}" class="text-xs px-2 py-1 border border-slate-300 rounded focus:border-indigo-500 focus:outline-none bg-slate-50">
|
||||||
|
<option value="LOW" ${t.priority === 'LOW' ? 'selected' : ''}>☕ LOW</option>
|
||||||
|
<option value="MEDIUM" ${t.priority === 'MEDIUM' || !t.priority ? 'selected' : ''}>⚡ MEDIUM</option>
|
||||||
|
<option value="HIGH" ${t.priority === 'HIGH' ? 'selected' : ''}>🔥 HIGH</option>
|
||||||
|
<option value="CRITICAL" ${t.priority === 'CRITICAL' ? 'selected' : ''}>🚨 CRITICAL</option>
|
||||||
|
</select>
|
||||||
|
<button type="button" onclick="window.saveTaskInlineEdit(${id})" class="px-2.5 py-1 text-xs font-semibold rounded bg-indigo-600 text-white hover:bg-indigo-700 transition-colors shadow-sm">Сохранить</button>
|
||||||
|
<button type="button" onclick="window.closeTaskInlineEditor(${id})" class="px-2 py-1 text-xs font-medium rounded text-slate-500 hover:bg-slate-100 transition-colors">Отмена</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}).join('');
|
||||||
|
|
||||||
|
const emptyState = `
|
||||||
|
<div class="p-8 text-center text-slate-400">
|
||||||
|
<div class="text-3xl mb-2">📭</div>
|
||||||
|
<div class="text-sm font-medium">Нет задач в категории «${currentFilter}»</div>
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
|
||||||
|
return `
|
||||||
|
<div class="task-widget-root w-full max-w-4xl mx-auto my-2 bg-slate-50 border border-slate-300 rounded-xl shadow-md overflow-hidden flex flex-col">
|
||||||
|
<div class="px-4 py-3 bg-white border-b border-slate-200 flex items-center justify-between gap-2 flex-wrap">
|
||||||
|
<div class="flex items-center gap-1.5 flex-wrap">
|
||||||
|
<button type="button" onclick="window.switchTaskFilter('IN_PROGRESS', this)" class="px-2.5 py-1 text-xs font-bold rounded-md transition-colors ${currentFilter === 'IN_PROGRESS' ? 'bg-blue-600 text-white shadow-sm' : 'bg-slate-100 text-slate-600 hover:bg-slate-200'}">
|
||||||
|
⚙️ В работе (${counts.IN_PROGRESS})
|
||||||
|
</button>
|
||||||
|
<button type="button" onclick="window.switchTaskFilter('PLANNED', this)" class="px-2.5 py-1 text-xs font-bold rounded-md transition-colors ${currentFilter === 'PLANNED' ? 'bg-indigo-600 text-white shadow-sm' : 'bg-slate-100 text-slate-600 hover:bg-slate-200'}">
|
||||||
|
📋 В планах (${counts.PLANNED})
|
||||||
|
</button>
|
||||||
|
<button type="button" onclick="window.switchTaskFilter('COMPLETED', this)" class="px-2.5 py-1 text-xs font-bold rounded-md transition-colors ${currentFilter === 'COMPLETED' ? 'bg-emerald-600 text-white shadow-sm' : 'bg-slate-100 text-slate-600 hover:bg-slate-200'}">
|
||||||
|
✓ Готово (${counts.COMPLETED})
|
||||||
|
</button>
|
||||||
|
<button type="button" onclick="window.switchTaskFilter('ALL', this)" class="px-2.5 py-1 text-xs font-bold rounded-md transition-colors ${currentFilter === 'ALL' ? 'bg-slate-800 text-white shadow-sm' : 'bg-slate-100 text-slate-600 hover:bg-slate-200'}">
|
||||||
|
Все (${counts.ALL})
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button type="button" onclick="window.toggleCreateTaskForm(this)" class="px-3 py-1.5 text-xs font-bold rounded-md bg-indigo-600 hover:bg-indigo-700 text-white transition-all shadow-sm flex items-center gap-1">
|
||||||
|
➕ Добавить задачу
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="new-task-creation-form hidden p-3 bg-indigo-50/70 border-b border-indigo-100 flex flex-col gap-2">
|
||||||
|
<div class="text-xs font-bold text-indigo-900">Новая задача:</div>
|
||||||
|
<input type="text" class="new-task-title w-full text-xs px-2.5 py-1.5 border border-indigo-200 rounded focus:border-indigo-500 focus:outline-none bg-white" placeholder="Что необходимо сделать?..." />
|
||||||
|
<div class="flex items-center gap-2 flex-wrap">
|
||||||
|
<input type="date" class="new-task-date text-xs px-2 py-1 border border-indigo-200 rounded focus:border-indigo-500 focus:outline-none bg-white" />
|
||||||
|
<select class="new-task-prio text-xs px-2 py-1 border border-indigo-200 rounded focus:border-indigo-500 focus:outline-none bg-white">
|
||||||
|
<option value="LOW">☕ LOW (Низкий)</option>
|
||||||
|
<option value="MEDIUM" selected>⚡ MEDIUM (Средний)</option>
|
||||||
|
<option value="HIGH">🔥 HIGH (Высокий)</option>
|
||||||
|
<option value="CRITICAL">🚨 CRITICAL (Критический)</option>
|
||||||
|
</select>
|
||||||
|
<select class="new-task-status text-xs px-2 py-1 border border-indigo-200 rounded focus:border-indigo-500 focus:outline-none bg-white">
|
||||||
|
<option value="BACKLOG" selected>📋 В планы (Бэклог)</option>
|
||||||
|
<option value="IN_PROGRESS">⚙️ Сразу в работу</option>
|
||||||
|
</select>
|
||||||
|
<button type="button" onclick="window.submitCreateTask(this)" class="px-3 py-1 text-xs font-bold rounded bg-indigo-600 text-white hover:bg-indigo-700 transition-colors shadow-sm">Создать</button>
|
||||||
|
<button type="button" onclick="window.toggleCreateTaskForm(this)" class="px-2 py-1 text-xs font-medium rounded text-slate-500 hover:bg-slate-200 transition-colors">Отмена</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="p-3 overflow-y-auto max-h-[70vh] flex flex-col gap-2 task-rows-container">
|
||||||
|
${filteredTasks.length > 0 ? taskRows : emptyState}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
|
||||||
|
window.switchTaskFilter = function(filterName, btnEl) {
|
||||||
|
window.activeTaskFilter = filterName;
|
||||||
|
const root = btnEl ? btnEl.closest('.task-widget-root') : document.querySelector('.task-widget-root');
|
||||||
|
if (root && window.currentTasksCache && window.currentTasksCache.length > 0) {
|
||||||
|
root.outerHTML = renderInteractiveTaskCard(window.currentTasksCache);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
window.toggleCreateTaskForm = function(btnEl) {
|
||||||
|
const root = btnEl ? btnEl.closest('.task-widget-root') : document.querySelector('.task-widget-root');
|
||||||
|
if (!root) return;
|
||||||
|
const form = root.querySelector('.new-task-creation-form');
|
||||||
|
if (form) {
|
||||||
|
form.classList.toggle('hidden');
|
||||||
|
if (!form.classList.contains('hidden')) {
|
||||||
|
const input = form.querySelector('.new-task-title');
|
||||||
|
if (input) input.focus();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
window.submitCreateTask = async function(btnEl) {
|
||||||
|
const root = btnEl ? btnEl.closest('.task-widget-root') : document.querySelector('.task-widget-root');
|
||||||
|
if (!root) return;
|
||||||
|
|
||||||
|
const titleInput = root.querySelector('.new-task-title');
|
||||||
|
const dateInput = root.querySelector('.new-task-date');
|
||||||
|
const prioInput = root.querySelector('.new-task-prio');
|
||||||
|
const statusInput = root.querySelector('.new-task-status');
|
||||||
|
|
||||||
|
if (!titleInput || !titleInput.value.trim()) {
|
||||||
|
alert('Введите описание задачи');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const res = await fetch('/api/v1/tasks', {
|
||||||
|
method: 'POST',
|
||||||
|
headers: getAuthHeaders(),
|
||||||
|
body: JSON.stringify({
|
||||||
|
title: titleInput.value.trim(),
|
||||||
|
due_date: dateInput ? (dateInput.value || null) : null,
|
||||||
|
priority: prioInput ? prioInput.value : 'MEDIUM',
|
||||||
|
status: statusInput ? statusInput.value : 'BACKLOG'
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
if (res.ok) {
|
||||||
|
window.activeTaskFilter = (statusInput && statusInput.value === 'IN_PROGRESS') ? 'IN_PROGRESS' : 'PLANNED';
|
||||||
|
window.sendChatAction('покажи задачи');
|
||||||
|
} else {
|
||||||
|
const err = await res.json();
|
||||||
|
alert('Ошибка создания задачи: ' + (err.detail || 'Неизвестная ошибка'));
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('Ошибка создания задачи:', e);
|
||||||
|
alert('Сетевая ошибка при создании задачи');
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
window.openTaskInlineEditor = function(id) {
|
||||||
|
const card = document.getElementById(`task-card-${id}`);
|
||||||
|
if (!card) return;
|
||||||
|
card.querySelectorAll('.task-view-mode').forEach(el => el.classList.add('hidden'));
|
||||||
|
const editor = document.getElementById(`task-editor-${id}`);
|
||||||
|
if (editor) editor.classList.remove('hidden');
|
||||||
|
};
|
||||||
|
|
||||||
|
window.closeTaskInlineEditor = function(id) {
|
||||||
|
const card = document.getElementById(`task-card-${id}`);
|
||||||
|
if (!card) return;
|
||||||
|
card.querySelectorAll('.task-view-mode').forEach(el => el.classList.remove('hidden'));
|
||||||
|
const editor = document.getElementById(`task-editor-${id}`);
|
||||||
|
if (editor) editor.classList.add('hidden');
|
||||||
|
};
|
||||||
|
|
||||||
|
window.saveTaskInlineEdit = async function(id) {
|
||||||
|
const titleInput = document.getElementById(`task-edit-title-${id}`);
|
||||||
|
const dateInput = document.getElementById(`task-edit-date-${id}`);
|
||||||
|
const prioInput = document.getElementById(`task-edit-prio-${id}`);
|
||||||
|
|
||||||
|
if (!titleInput || !titleInput.value.trim()) {
|
||||||
|
alert('Описание задачи не может быть пустым');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const res = await fetch(`/api/v1/tasks/${id}`, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: getAuthHeaders(),
|
||||||
|
body: JSON.stringify({
|
||||||
|
title: titleInput.value.trim(),
|
||||||
|
due_date: dateInput ? (dateInput.value || null) : null,
|
||||||
|
priority: prioInput ? prioInput.value : 'MEDIUM'
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
if (res.ok) {
|
||||||
|
window.sendChatAction('покажи задачи');
|
||||||
|
} else {
|
||||||
|
const err = await res.json();
|
||||||
|
alert('Ошибка обновления задачи: ' + (err.detail || 'Неизвестная ошибка'));
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error('Ошибка сохранения задачи:', e);
|
||||||
|
alert('Сетевая ошибка при обновлении задачи');
|
||||||
|
}
|
||||||
|
};
|
||||||
@@ -1,145 +1,117 @@
|
|||||||
let currentFilter = 'ALL';
|
/**
|
||||||
let allTasks = [];
|
===============================================================================
|
||||||
|
FILE: modules/web_api/static/js/tasks.js
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: web_api / static / js
|
||||||
|
ROLE: Загрузка, фильтрация и рендеринг списка задач в боковой панели (Drawer).
|
||||||
|
===============================================================================
|
||||||
|
*/
|
||||||
|
|
||||||
function toggleDrawer() {
|
let currentTaskFilter = 'IN_PROGRESS';
|
||||||
if (typeof IS_GUEST !== 'undefined' && IS_GUEST) return;
|
|
||||||
const drawer = document.getElementById("task-drawer");
|
|
||||||
const backdrop = document.getElementById("drawer-backdrop");
|
|
||||||
if (!drawer) return;
|
|
||||||
|
|
||||||
const isHidden = drawer.classList.contains("translate-x-full");
|
|
||||||
if (isHidden) {
|
|
||||||
drawer.classList.remove("translate-x-full");
|
|
||||||
if (backdrop) backdrop.classList.remove("hidden");
|
|
||||||
loadTasks();
|
|
||||||
} else {
|
|
||||||
drawer.classList.add("translate-x-full");
|
|
||||||
if (backdrop) backdrop.classList.add("hidden");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function setFilter(status) {
|
|
||||||
currentFilter = status;
|
|
||||||
["ALL", "IN_PROGRESS", "BACKLOG", "COMPLETED"].forEach(f => {
|
|
||||||
const btn = document.getElementById(`filter-${f}`);
|
|
||||||
if (btn) {
|
|
||||||
btn.className = (f === status)
|
|
||||||
? "px-3 py-1.5 rounded-t-lg border-b-2 border-indigo-600 text-indigo-600 font-bold whitespace-nowrap"
|
|
||||||
: "px-3 py-1.5 rounded-t-lg border-b-2 border-transparent whitespace-nowrap";
|
|
||||||
}
|
|
||||||
});
|
|
||||||
renderTasks();
|
|
||||||
}
|
|
||||||
|
|
||||||
async function loadTasks() {
|
async function loadTasks() {
|
||||||
const badge = document.getElementById("task-count-badge");
|
|
||||||
const container = document.getElementById("tasks-container");
|
|
||||||
|
|
||||||
const token = typeof API_TOKEN !== 'undefined' && API_TOKEN ? API_TOKEN : localStorage.getItem("scud_api_auth_token");
|
const token = typeof API_TOKEN !== 'undefined' && API_TOKEN ? API_TOKEN : localStorage.getItem("scud_api_auth_token");
|
||||||
const isGuest = typeof IS_GUEST !== 'undefined' ? IS_GUEST : (localStorage.getItem("scud_is_guest") === "true");
|
if (!token) return;
|
||||||
|
|
||||||
if (isGuest || !token) {
|
|
||||||
if (badge) badge.innerText = "0";
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const res = await fetch("/api/v1/tasks", {
|
const res = await fetch("/api/v1/tasks", {
|
||||||
headers: {
|
headers: { "Authorization": "Bearer " + token }
|
||||||
"Authorization": "Bearer " + token,
|
|
||||||
"Content-Type": "application/json"
|
|
||||||
}
|
|
||||||
});
|
});
|
||||||
|
if (res.ok) {
|
||||||
if (res.status === 401) {
|
const data = await res.json();
|
||||||
if (typeof logout === 'function') logout();
|
renderSidebarTasks(data.tasks || []);
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!res.ok) {
|
|
||||||
throw new Error(`HTTP error! status: ${res.status}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
const data = await res.json();
|
|
||||||
|
|
||||||
// Гибкое определение структуры данных (массив или объект с ключом tasks)
|
|
||||||
if (Array.isArray(data)) {
|
|
||||||
allTasks = data;
|
|
||||||
} else if (data && Array.isArray(data.tasks)) {
|
|
||||||
allTasks = data.tasks;
|
|
||||||
} else if (data && typeof data === 'object') {
|
|
||||||
allTasks = Object.values(data).find(val => Array.isArray(val)) || [];
|
|
||||||
} else {
|
} else {
|
||||||
allTasks = [];
|
renderSidebarError("Ошибка доступа. Авторизуйтесь снова.");
|
||||||
}
|
|
||||||
|
|
||||||
if (badge) {
|
|
||||||
badge.innerText = allTasks.length.toString();
|
|
||||||
}
|
|
||||||
|
|
||||||
renderTasks();
|
|
||||||
|
|
||||||
} catch (err) {
|
|
||||||
console.error("[Tasks Error]", err);
|
|
||||||
if (badge) badge.innerText = "0";
|
|
||||||
if (container) {
|
|
||||||
container.innerHTML = `<div class="text-red-500 text-xs py-8 text-center font-medium">Ошибка обработки списка задач</div>`;
|
|
||||||
}
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.error("Ошибка загрузки задач:", e);
|
||||||
|
renderSidebarError("Ошибка сети. Сервер недоступен.");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function renderTasks() {
|
function renderSidebarError(msg) {
|
||||||
const container = document.getElementById("tasks-container");
|
// Поддерживаем оба варианта ID (новый и старый) для обратной совместимости
|
||||||
|
const container = document.getElementById("tasks-list") || document.getElementById("tasks-list-container");
|
||||||
|
if (container) {
|
||||||
|
container.innerHTML = `<div class="text-center py-8 text-xs text-rose-500 font-semibold">${msg}</div>`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function filterTasksByTab(status) {
|
||||||
|
currentTaskFilter = status;
|
||||||
|
|
||||||
|
// Сброс стилей всех кнопок-вкладок
|
||||||
|
document.querySelectorAll('.task-tab-btn').forEach(btn => {
|
||||||
|
btn.classList.remove('text-indigo-600', 'bg-indigo-50');
|
||||||
|
btn.classList.add('text-slate-600');
|
||||||
|
});
|
||||||
|
|
||||||
|
// Установка активного стиля для выбранной вкладки
|
||||||
|
const activeBtnId = {
|
||||||
|
'IN_PROGRESS': 'tab-in-progress',
|
||||||
|
'BACKLOG': 'tab-backlog',
|
||||||
|
'COMPLETED': 'tab-completed',
|
||||||
|
'ALL': 'tab-all'
|
||||||
|
}[status];
|
||||||
|
|
||||||
|
if (activeBtnId) {
|
||||||
|
const btn = document.getElementById(activeBtnId);
|
||||||
|
if (btn) {
|
||||||
|
btn.classList.remove('text-slate-600', 'hover:bg-slate-100');
|
||||||
|
btn.classList.add('text-indigo-600', 'bg-indigo-50');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
loadTasks();
|
||||||
|
}
|
||||||
|
|
||||||
|
function renderSidebarTasks(tasks) {
|
||||||
|
const container = document.getElementById("tasks-list") || document.getElementById("tasks-list-container");
|
||||||
if (!container) return;
|
if (!container) return;
|
||||||
|
|
||||||
if (!Array.isArray(allTasks)) {
|
let filtered = tasks;
|
||||||
allTasks = [];
|
if (currentTaskFilter !== 'ALL') {
|
||||||
|
filtered = tasks.filter(t => t.status === currentTaskFilter);
|
||||||
}
|
}
|
||||||
|
|
||||||
const filtered = allTasks.filter(t => currentFilter === "ALL" || t.status === currentFilter);
|
|
||||||
|
|
||||||
if (filtered.length === 0) {
|
if (filtered.length === 0) {
|
||||||
container.innerHTML = `<div class="text-slate-400 text-xs py-8 text-center">Нет задач с выбранным фильтром</div>`;
|
container.innerHTML = `<div class="text-center py-8 text-[11px] font-medium text-slate-400 bg-slate-50 rounded-xl border border-dashed border-slate-200">Нет задач в этой категории</div>`;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
container.innerHTML = filtered.map(t => {
|
container.innerHTML = filtered.map(t => {
|
||||||
let statusBadge = "bg-slate-100 text-slate-600 border-slate-200";
|
const isCompleted = t.status === 'COMPLETED';
|
||||||
let cardBg = "bg-white";
|
const priorityColor = t.priority === 'HIGH' || t.priority === 'CRITICAL'
|
||||||
|
? 'text-rose-600 bg-rose-50 border-rose-200'
|
||||||
|
: t.priority === 'MEDIUM'
|
||||||
|
? 'text-amber-600 bg-amber-50 border-amber-200'
|
||||||
|
: 'text-slate-600 bg-slate-50 border-slate-200';
|
||||||
|
|
||||||
if (t.status === "COMPLETED") {
|
|
||||||
statusBadge = "bg-emerald-50 text-emerald-700 border-emerald-300 font-semibold";
|
|
||||||
cardBg = "bg-emerald-50/20";
|
|
||||||
} else if (t.status === "IN_PROGRESS") {
|
|
||||||
statusBadge = "bg-amber-50 text-amber-700 border-amber-300 font-bold";
|
|
||||||
cardBg = "bg-amber-50/20 border-amber-200";
|
|
||||||
}
|
|
||||||
|
|
||||||
let priorityBadge = "text-slate-500 bg-slate-100 border-slate-200";
|
|
||||||
if (t.priority === "HIGH") priorityBadge = "text-red-700 bg-red-50 border-red-200 font-bold";
|
|
||||||
|
|
||||||
let dueDateHtml = t.due_date ? `
|
|
||||||
<div class="mt-2 text-[11px] text-amber-800 bg-amber-50 border border-amber-200 px-2 py-0.5 rounded-md flex items-center gap-1.5 w-fit font-medium">
|
|
||||||
<i class="fa-solid fa-clock text-amber-600"></i>
|
|
||||||
<span>Срок: ${t.due_date}</span>
|
|
||||||
</div>` : "";
|
|
||||||
|
|
||||||
return `
|
return `
|
||||||
<div class="${cardBg} border border-slate-200 rounded-xl p-3.5 shadow-sm hover:shadow-md transition">
|
<div class="p-3 bg-white border border-slate-200 rounded-xl shadow-sm hover:border-indigo-400 hover:shadow-md transition cursor-pointer flex flex-col gap-2 group"
|
||||||
<div class="flex justify-between items-center mb-1.5">
|
onclick="handleActionButtonClick('покажи задачу ${t.id}')">
|
||||||
<div class="flex items-center gap-1.5">
|
|
||||||
<span class="font-mono text-xs font-bold text-slate-900 bg-slate-100 px-2 py-0.5 rounded border border-slate-200">${t.task_id || t.id || 'TASK'}</span>
|
<div class="flex items-start justify-between gap-2">
|
||||||
<span class="text-[10px] uppercase px-1.5 py-0.5 rounded border ${priorityBadge}">${t.priority || 'MEDIUM'}</span>
|
<span class="text-[10px] font-bold text-slate-400 uppercase tracking-wider group-hover:text-indigo-500 transition">#${t.id}</span>
|
||||||
</div>
|
${isCompleted
|
||||||
<span class="text-[10px] uppercase px-2 py-0.5 rounded border ${statusBadge}">${t.status || 'BACKLOG'}</span>
|
? `<span class="text-[10px] font-bold px-1.5 py-0.5 rounded-md bg-emerald-50 text-emerald-600 border border-emerald-200 shadow-sm"><i class="fa-solid fa-check mr-0.5"></i> Готово</span>`
|
||||||
|
: `<span class="text-[10px] font-bold px-1.5 py-0.5 rounded-md ${priorityColor} shadow-sm">${t.priority || 'LOW'}</span>`
|
||||||
|
}
|
||||||
</div>
|
</div>
|
||||||
<h3 class="text-xs font-semibold text-slate-800 mb-1 leading-snug">${t.title || t.description || ''}</h3>
|
|
||||||
<div class="text-[10px] text-slate-400 font-mono flex items-center gap-1">
|
<div class="text-xs font-semibold text-slate-700 leading-snug line-clamp-3">${escapeHtml(t.title || 'Без названия')}</div>
|
||||||
<i class="fa-solid fa-folder-closed text-slate-300"></i>
|
|
||||||
<span>${t.module || 'General'}</span>
|
<div class="flex items-center justify-between text-[10px] text-slate-400 mt-1">
|
||||||
|
<span class="bg-slate-100 px-1.5 py-0.5 rounded font-mono truncate max-w-[120px]">${escapeHtml(t.module || 'general')}</span>
|
||||||
|
${t.due_date ? `<span class="shrink-0 font-medium text-slate-500"><i class="fa-regular fa-calendar mr-1"></i>${escapeHtml(t.due_date)}</span>` : ''}
|
||||||
</div>
|
</div>
|
||||||
${dueDateHtml}
|
|
||||||
</div>
|
</div>
|
||||||
`;
|
`;
|
||||||
}).join("");
|
}).join('');
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Глобальная инициализация при загрузке DOM
|
||||||
|
document.addEventListener("DOMContentLoaded", () => {
|
||||||
|
// Небольшая задержка, чтобы гарантировать применение токена
|
||||||
|
setTimeout(loadTasks, 200);
|
||||||
|
});
|
||||||
+6325
-1588
File diff suppressed because it is too large
Load Diff
Executable
+20
@@ -0,0 +1,20 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
# Переход в папку проекта
|
||||||
|
cd /home/puh/projects/scud_ai
|
||||||
|
|
||||||
|
# Создание папки для логов
|
||||||
|
mkdir -p /home/puh/projects/scud_ai/logs
|
||||||
|
|
||||||
|
# Фиксация старта
|
||||||
|
echo "==================================================" >> /home/puh/projects/scud_ai/logs/cron_etl.log
|
||||||
|
echo "[CRON START] $(date '+%Y-%m-%d %H:%M:%S')" >> /home/puh/projects/scud_ai/logs/cron_etl.log
|
||||||
|
echo "==================================================" >> /home/puh/projects/scud_ai/logs/cron_etl.log
|
||||||
|
|
||||||
|
# Запуск main_etl.py через интерпретатор окружения
|
||||||
|
/home/puh/scud_orion_ai_v2/venv/bin/python /home/puh/projects/scud_ai/main_etl.py >> /home/puh/projects/scud_ai/logs/cron_etl.log 2>&1
|
||||||
|
|
||||||
|
# Фиксация окончания
|
||||||
|
echo "[CRON FINISH] $(date '+%Y-%m-%d %H:%M:%S')" >> /home/puh/projects/scud_ai/logs/cron_etl.log
|
||||||
|
echo "" >> /home/puh/projects/scud_ai/logs/cron_etl.log
|
||||||
+292
-64
@@ -7,7 +7,7 @@ from datetime import datetime
|
|||||||
|
|
||||||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||||
|
|
||||||
from config import DATA_DIR, DATE_TODAY, OUTPUT_DIR
|
from config import DATA_DIR, DATE_TODAY, OUTPUT_DIR, EXCEPTIONS_PATH, normalize_fio
|
||||||
from core.database import (
|
from core.database import (
|
||||||
get_connection,
|
get_connection,
|
||||||
get_available_snapshots,
|
get_available_snapshots,
|
||||||
@@ -15,9 +15,54 @@ from core.database import (
|
|||||||
load_scud_from_db_by_snapshot,
|
load_scud_from_db_by_snapshot,
|
||||||
get_latest_snapshot_time
|
get_latest_snapshot_time
|
||||||
)
|
)
|
||||||
|
from services.exceptions_repo import (
|
||||||
|
get_all_exceptions_from_db,
|
||||||
|
add_exception_to_db,
|
||||||
|
remove_exception_from_db,
|
||||||
|
sync_json_to_db
|
||||||
|
)
|
||||||
|
|
||||||
DB_PATH = os.path.join(DATA_DIR, "scud_orion_ai.db")
|
DB_PATH = os.path.join(DATA_DIR, "scud_orion_ai.db")
|
||||||
|
|
||||||
|
def cmd_mapping(args_list):
|
||||||
|
"""Управление подтвержденными сопоставлениями ФИО (СКУД <-> 1С:ЗУП)."""
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
|
||||||
|
if not args_list or args_list[0] in ["list", "show"]:
|
||||||
|
cursor.execute("SELECT id, scud_fio, zup_fio, match_source, status FROM person_identity_mapping ORDER BY id DESC")
|
||||||
|
rows = cursor.fetchall()
|
||||||
|
print("\n🔗 СОХРАНЕННЫЕ СОПОСТАВЛЕНИЯ ФИО (person_identity_mapping):")
|
||||||
|
print("=" * 80)
|
||||||
|
if rows:
|
||||||
|
for r in rows:
|
||||||
|
print(f" #{r[0]} [{r[4]}] СКУД: '{r[1]}' ⟷ 1С: '{r[2]}' ({r[3]})")
|
||||||
|
else:
|
||||||
|
print(" — сопоставлений пока нет")
|
||||||
|
print("=" * 80 + "\n")
|
||||||
|
return
|
||||||
|
|
||||||
|
subcmd = args_list[0]
|
||||||
|
if subcmd == "add":
|
||||||
|
if len(args_list) < 3:
|
||||||
|
print("Использование: python scripts/db_cli.py mapping add 'ФИО в СКУД' 'ФИО в 1С'")
|
||||||
|
return
|
||||||
|
scud_f, zup_f = args_list[1], args_list[2]
|
||||||
|
cursor.execute("""
|
||||||
|
INSERT OR REPLACE INTO person_identity_mapping (scud_fio, zup_fio, match_source, status)
|
||||||
|
VALUES (?, ?, 'MANUAL', 'ACTIVE')
|
||||||
|
""", (scud_f.strip(), zup_f.strip()))
|
||||||
|
conn.commit()
|
||||||
|
print(f"✅ Успешно добавлена связка: '{scud_f}' ⟷ '{zup_f}'")
|
||||||
|
|
||||||
|
elif subcmd in ["del", "delete", "remove"]:
|
||||||
|
if len(args_list) < 2:
|
||||||
|
print("Использование: python scripts/db_cli.py mapping del 'ФИО в СКУД'")
|
||||||
|
return
|
||||||
|
cursor.execute("DELETE FROM person_identity_mapping WHERE scud_fio = ?", (args_list[1].strip(),))
|
||||||
|
conn.commit()
|
||||||
|
print(f"✅ Связка для '{args_list[1]}' удалена.")
|
||||||
|
|
||||||
def print_tool_actions():
|
def print_tool_actions():
|
||||||
"""Выводит реестр декларативных действий инструментов и шаблоны кнопок."""
|
"""Выводит реестр декларативных действий инструментов и шаблоны кнопок."""
|
||||||
print("\n" + "=" * 110)
|
print("\n" + "=" * 110)
|
||||||
@@ -25,25 +70,29 @@ def print_tool_actions():
|
|||||||
print("=" * 110)
|
print("=" * 110)
|
||||||
with get_connection() as conn:
|
with get_connection() as conn:
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
cursor.execute("""
|
try:
|
||||||
SELECT id, tool_name, category, bypass_llm, success_template, follow_up_question, buttons_json
|
cursor.execute("""
|
||||||
FROM tool_action_registry
|
SELECT id, tool_name, category, bypass_llm, success_template, follow_up_question, buttons_json
|
||||||
WHERE is_active = 1
|
FROM tool_action_registry
|
||||||
ORDER BY id ASC
|
WHERE is_active = 1
|
||||||
""")
|
ORDER BY id ASC
|
||||||
rows = cursor.fetchall()
|
""")
|
||||||
if not rows:
|
rows = cursor.fetchall()
|
||||||
print("Таблица tool_action_registry пуста.")
|
if not rows:
|
||||||
else:
|
print("Таблица tool_action_registry пуста.")
|
||||||
for r in rows:
|
else:
|
||||||
print(f"ID: {r[0]} | Tool: [{r[1]}] | Категория: {r[2]} | Bypass LLM: {'ДА (0.05с)' if r[3] else 'НЕТ'}")
|
for r in rows:
|
||||||
print(f" • Сообщение: {r[4]}")
|
print(f"ID: {r[0]} | Tool: [{r[1]}] | Категория: {r[2]} | Bypass LLM: {'ДА (0.05с)' if r[3] else 'НЕТ'}")
|
||||||
if r[5]:
|
print(f" • Сообщение: {r[4]}")
|
||||||
print(f" • Вопрос: {r[5]}")
|
if r[5]:
|
||||||
print(f" • Кнопки: {r[6]}")
|
print(f" • Вопрос: {r[5]}")
|
||||||
print("-" * 110)
|
print(f" • Кнопки: {r[6]}")
|
||||||
|
print("-" * 110)
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Таблица tool_action_registry недоступна: {e}")
|
||||||
print("=" * 110 + "\n")
|
print("=" * 110 + "\n")
|
||||||
|
|
||||||
|
|
||||||
def print_stats():
|
def print_stats():
|
||||||
"""Выводит общую статистику по записям в таблицах БД."""
|
"""Выводит общую статистику по записям в таблицах БД."""
|
||||||
print("\n" + "=" * 60)
|
print("\n" + "=" * 60)
|
||||||
@@ -51,11 +100,18 @@ def print_stats():
|
|||||||
print("=" * 60)
|
print("=" * 60)
|
||||||
with get_connection() as conn:
|
with get_connection() as conn:
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
tables = ['scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history', 'ai_knowledge_base']
|
tables = [
|
||||||
|
'scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history',
|
||||||
|
'ai_knowledge_base', 'chat_messages', 'session_states',
|
||||||
|
'system_prompt_nodes', 'tasks', 'exceptions_registry'
|
||||||
|
]
|
||||||
for t in tables:
|
for t in tables:
|
||||||
cursor.execute(f"SELECT COUNT(*) FROM {t}")
|
try:
|
||||||
cnt = cursor.fetchone()[0]
|
cursor.execute(f"SELECT COUNT(*) FROM {t}")
|
||||||
print(f" • Таблица [{t:<18}]: {cnt:>6} записей")
|
cnt = cursor.fetchone()[0]
|
||||||
|
print(f" • Таблица [{t:<22}]: {cnt:>6} записей")
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
print("=" * 60 + "\n")
|
print("=" * 60 + "\n")
|
||||||
|
|
||||||
|
|
||||||
@@ -79,13 +135,11 @@ def print_snapshots_list(date_str=None):
|
|||||||
def snapshot_sort_key(row):
|
def snapshot_sort_key(row):
|
||||||
snap_id = row[0] or ""
|
snap_id = row[0] or ""
|
||||||
snap_time = row[2] or ""
|
snap_time = row[2] or ""
|
||||||
|
|
||||||
seq_num = 0
|
seq_num = 0
|
||||||
if "-" in snap_id:
|
if "-" in snap_id:
|
||||||
parts = snap_id.replace("Y", "").split("-")
|
parts = snap_id.replace("Y", "").split("-")
|
||||||
if len(parts) > 1 and parts[1].isdigit():
|
if len(parts) > 1 and parts[1].isdigit():
|
||||||
seq_num = int(parts[1])
|
seq_num = int(parts[1])
|
||||||
|
|
||||||
return (snap_time, seq_num)
|
return (snap_time, seq_num)
|
||||||
|
|
||||||
sorted_rows = sorted(rows, key=snapshot_sort_key, reverse=True)
|
sorted_rows = sorted(rows, key=snapshot_sort_key, reverse=True)
|
||||||
@@ -101,11 +155,7 @@ def print_snapshots_list(date_str=None):
|
|||||||
time_part = snap_time.split(" ")[1]
|
time_part = snap_time.split(" ")[1]
|
||||||
|
|
||||||
slice_datetime_str = f"{log_date} {time_part}" if time_part != "—" else log_date
|
slice_datetime_str = f"{log_date} {time_part}" if time_part != "—" else log_date
|
||||||
|
formatted_snap_id = f" {snap_id}" if not snap_id.startswith("Y") else snap_id
|
||||||
if not snap_id.startswith("Y"):
|
|
||||||
formatted_snap_id = f" {snap_id}"
|
|
||||||
else:
|
|
||||||
formatted_snap_id = snap_id
|
|
||||||
|
|
||||||
print(f"{formatted_snap_id:<16} | {snap_time:<24} | {slice_datetime_str:<20} | {count:<8}")
|
print(f"{formatted_snap_id:<16} | {snap_time:<24} | {slice_datetime_str:<20} | {count:<8}")
|
||||||
|
|
||||||
@@ -138,31 +188,44 @@ def inspect_scud(snapshot_id=None, date_str=None, export_xlsx=None):
|
|||||||
"""Инспектирует логи СКУД за выбранный снапшот или дату и опционально сохраняет XLSX."""
|
"""Инспектирует логи СКУД за выбранный снапшот или дату и опционально сохраняет XLSX."""
|
||||||
target_date = date_str if date_str else DATE_TODAY
|
target_date = date_str if date_str else DATE_TODAY
|
||||||
|
|
||||||
print("\n" + "=" * 90)
|
print("\n" + "=" * 115)
|
||||||
if snapshot_id:
|
if snapshot_id:
|
||||||
print(f"🔍 ИНСПЕКЦИЯ СКУД ПО СНАПШОТУ [{snapshot_id}] (Дата: {target_date}):")
|
print(f"🔍 ИНСПЕКЦИЯ СКУД ПО СНАПШОТУ [{snapshot_id}] (Дата среза: {target_date}):")
|
||||||
else:
|
else:
|
||||||
print(f"🔍 ИНСПЕКЦИЯ СКУД ЗА ТЕКУЩУЮ ДАТУ [{target_date}] (ПОСЛЕДНИЙ СРЕЗ):")
|
print(f"🔍 ИНСПЕКЦИЯ СКУД ЗА ТЕКУЩУЮ ДАТУ [{target_date}] (ПОСЛЕДНИЙ СРЕЗ):")
|
||||||
print("=" * 90)
|
print("=" * 115)
|
||||||
|
|
||||||
df = load_scud_from_db_by_snapshot(target_date, snapshot_param=snapshot_id)
|
df = load_scud_from_db_by_snapshot(target_date, snapshot_param=snapshot_id)
|
||||||
|
|
||||||
if df.empty:
|
if df is None or df.empty:
|
||||||
print(f"Записи СКУД не найдены.")
|
print("Записи СКУД не найдены.")
|
||||||
print("=" * 90 + "\n")
|
print("=" * 115 + "\n")
|
||||||
return
|
return
|
||||||
|
|
||||||
|
fio_col = next((c for c in ['Сотрудник', 'fio', 'fio_clean'] if c in df.columns), None)
|
||||||
|
dept_col = next((c for c in ['Подразделение', 'department_scud', 'department'] if c in df.columns), None)
|
||||||
|
pos_col = next((c for c in ['Должность', 'position'] if c in df.columns), None)
|
||||||
|
in_col = next((c for c in ['Начало_дня', 'time_in'] if c in df.columns), None)
|
||||||
|
first_act_col = next((c for c in ['Первая_активность', 'first_activity'] if c in df.columns), None)
|
||||||
|
out_col = next((c for c in ['Конец_дня', 'time_out'] if c in df.columns), None)
|
||||||
|
dur_col = next((c for c in ['Находился_в_здании', 'time_in_building', 'duration'] if c in df.columns), None)
|
||||||
|
present_col = next((c for c in ['Пришел', 'is_present'] if c in df.columns), None)
|
||||||
|
anom_col = 'anomaly_flag' if 'anomaly_flag' in df.columns else None
|
||||||
|
snap_col = 'snapshot_id' if 'snapshot_id' in df.columns else None
|
||||||
|
|
||||||
total = len(df)
|
total = len(df)
|
||||||
present_cnt = len(df[df['Пришел'] == True]) if 'Пришел' in df.columns else 0
|
if present_col:
|
||||||
|
present_cnt = len(df[df[present_col].astype(str).str.lower().isin(['true', '1'])])
|
||||||
|
else:
|
||||||
|
present_cnt = 0
|
||||||
absent_cnt = total - present_cnt
|
absent_cnt = total - present_cnt
|
||||||
|
|
||||||
print(f"Всего записей: {total} | Пришли: {present_cnt} | Не пришли: {absent_cnt}")
|
print(f"Всего записей: {total} | Присутствовали: {present_cnt} | Отсутствовали: {absent_cnt}")
|
||||||
print("-" * 90)
|
print("-" * 115)
|
||||||
|
|
||||||
cols_to_show = ['fio', 'department', 'position', 'time_in', 'first_activity', 'time_out', 'is_present', 'anomaly_flag', 'snapshot_id']
|
display_cols = [c for c in [fio_col, dept_col, in_col, first_act_col, out_col, dur_col, present_col, anom_col, snap_col] if c]
|
||||||
existing_cols = [c for c in cols_to_show if c in df.columns]
|
print(df[display_cols].head(30).to_string(index=False))
|
||||||
|
|
||||||
print(df[existing_cols].head(30).to_string(index=False))
|
|
||||||
if len(df) > 30:
|
if len(df) > 30:
|
||||||
print(f"\n... и ещё {len(df) - 30} строк.")
|
print(f"\n... и ещё {len(df) - 30} строк.")
|
||||||
|
|
||||||
@@ -172,11 +235,11 @@ def inspect_scud(snapshot_id=None, date_str=None, export_xlsx=None):
|
|||||||
out_path = os.path.join(OUTPUT_DIR, out_path)
|
out_path = os.path.join(OUTPUT_DIR, out_path)
|
||||||
|
|
||||||
df.to_excel(out_path, index=False)
|
df.to_excel(out_path, index=False)
|
||||||
print("\n" + "*" * 90)
|
print("\n" + "*" * 115)
|
||||||
print(f"[✓] УСПЕШНЫЙ ЭКСПОРТ ДЕБАГ-ФАЙЛА В EXCEL: {out_path}")
|
print(f"[✓] УСПЕШНЫЙ ЭКСПОРТ ДЕБАГ-ФАЙЛА В EXCEL: {out_path}")
|
||||||
print("*" * 90)
|
print("*" * 115)
|
||||||
|
|
||||||
print("=" * 90 + "\n")
|
print("=" * 115 + "\n")
|
||||||
|
|
||||||
|
|
||||||
def print_absences(date_str=None):
|
def print_absences(date_str=None):
|
||||||
@@ -236,28 +299,35 @@ def dump_all_to_excel(out_filename="db_dump_full.xlsx"):
|
|||||||
out_path = os.path.join(OUTPUT_DIR, out_filename)
|
out_path = os.path.join(OUTPUT_DIR, out_filename)
|
||||||
print(f"\n[🔄] Создание полного дампа БД в файл: {out_path} ...")
|
print(f"\n[🔄] Создание полного дампа БД в файл: {out_path} ...")
|
||||||
with get_connection() as conn, pd.ExcelWriter(out_path, engine='openpyxl') as writer:
|
with get_connection() as conn, pd.ExcelWriter(out_path, engine='openpyxl') as writer:
|
||||||
for table in ['scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history', 'ai_knowledge_base']:
|
for table in ['scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history', 'ai_knowledge_base', 'chat_messages', 'session_states', 'system_prompt_nodes', 'tasks', 'exceptions_registry']:
|
||||||
df = pd.read_sql_query(f"SELECT * FROM {table}", conn)
|
try:
|
||||||
df.to_excel(writer, sheet_name=table[:31], index=False)
|
df = pd.read_sql_query(f"SELECT * FROM {table}", conn)
|
||||||
|
df.to_excel(writer, sheet_name=table[:31], index=False)
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
print(f"[✓] Дамп успешно сохранен: {out_path}\n")
|
print(f"[✓] Дамп успешно сохранен: {out_path}\n")
|
||||||
|
|
||||||
|
|
||||||
def print_system_prompts():
|
def print_system_prompts():
|
||||||
"""Выводит все системные промпты из базы данных."""
|
"""Выводит все системные промпты из базы данных."""
|
||||||
print("\n" + "=" * 80)
|
print("\n" + "=" * 80)
|
||||||
print("📝 СИСТЕМНЫЕ ПРОМПТЫ (system_prompts):")
|
print("📝 СИСТЕМНЫЕ ПРОМПТЫ (system_prompt_nodes):")
|
||||||
print("=" * 80)
|
print("=" * 80)
|
||||||
with get_connection() as conn:
|
with get_connection() as conn:
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
cursor.execute("SELECT id, name, is_active, updated_at, prompt_text FROM system_prompts ORDER BY id DESC")
|
try:
|
||||||
rows = cursor.fetchall()
|
cursor.execute("SELECT section_id, item_id, content, is_active, updated_at FROM system_prompt_nodes ORDER BY section_id ASC, item_id ASC")
|
||||||
|
rows = cursor.fetchall()
|
||||||
|
except Exception:
|
||||||
|
rows = []
|
||||||
|
|
||||||
if not rows:
|
if not rows:
|
||||||
print("Таблица system_prompts пуста.")
|
print("Таблица system_prompt_nodes пуста.")
|
||||||
else:
|
else:
|
||||||
for r in rows:
|
for r in rows:
|
||||||
print(f"ID: {r[0]} | Name: {r[1]} | Active: {r[2]} | Updated: {r[3]}")
|
print(f"Раздел {r[0]}.{r[1]} | Active: {r[3]} | Updated: {r[4]}")
|
||||||
|
print(f" {r[2]}")
|
||||||
print("-" * 80)
|
print("-" * 80)
|
||||||
print(f"{r[4]}\n")
|
|
||||||
print("=" * 80 + "\n")
|
print("=" * 80 + "\n")
|
||||||
|
|
||||||
|
|
||||||
@@ -280,6 +350,107 @@ def print_session_states():
|
|||||||
print("=" * 80 + "\n")
|
print("=" * 80 + "\n")
|
||||||
|
|
||||||
|
|
||||||
|
def print_chat_messages(session_id=None, limit=50):
|
||||||
|
"""Выводит таблицу истории сообщений чата (контекст)."""
|
||||||
|
print("\n" + "=" * 105)
|
||||||
|
print(f"💬 ИСТОРИЯ СООБЩЕНИЙ ЧАТА (chat_messages) {f'для сессии: {session_id}' if session_id else 'все сессии'}:")
|
||||||
|
print("=" * 105)
|
||||||
|
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
if session_id:
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT id, session_id, role, content, is_ephemeral, created_at
|
||||||
|
FROM chat_messages
|
||||||
|
WHERE session_id = ?
|
||||||
|
ORDER BY id ASC
|
||||||
|
LIMIT ?
|
||||||
|
""", (session_id, limit))
|
||||||
|
else:
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT id, session_id, role, content, is_ephemeral, created_at
|
||||||
|
FROM chat_messages
|
||||||
|
ORDER BY id DESC
|
||||||
|
LIMIT ?
|
||||||
|
""", (limit,))
|
||||||
|
|
||||||
|
rows = cursor.fetchall()
|
||||||
|
if not rows:
|
||||||
|
print("Таблица chat_messages пуста.")
|
||||||
|
else:
|
||||||
|
if not session_id:
|
||||||
|
rows = list(reversed(rows))
|
||||||
|
|
||||||
|
for r in rows:
|
||||||
|
msg_id, sess, role, content, ephemeral, created = r
|
||||||
|
eph_marker = " [ЭФЕМЕРНОЕ]" if ephemeral else ""
|
||||||
|
print(f"[{msg_id}] {created} | Сессия: {sess} | Роль: {role.upper()}{eph_marker}")
|
||||||
|
print("-" * 105)
|
||||||
|
content_preview = content if content else ""
|
||||||
|
print(f"{content_preview}")
|
||||||
|
print("=" * 105)
|
||||||
|
print("\n")
|
||||||
|
|
||||||
|
|
||||||
|
def purge_chat_context(session_id=None, purge_all=False):
|
||||||
|
"""
|
||||||
|
Очистка контекста сообщений:
|
||||||
|
- По умолчанию: удаляет эфемерные сообщения, осиротевшие превью и сбрасывает стейты сессий.
|
||||||
|
- purge_all=True (--all): полностью очищает всю таблицу chat_messages и сбрасывает сессии.
|
||||||
|
"""
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
|
||||||
|
if purge_all:
|
||||||
|
if session_id:
|
||||||
|
cursor.execute("DELETE FROM chat_messages WHERE session_id = ?", (session_id,))
|
||||||
|
cursor.execute("DELETE FROM session_states WHERE session_id = ?", (session_id,))
|
||||||
|
else:
|
||||||
|
cursor.execute("DELETE FROM chat_messages")
|
||||||
|
cursor.execute("DELETE FROM session_states")
|
||||||
|
deleted_msgs = cursor.rowcount
|
||||||
|
conn.commit()
|
||||||
|
print(f"\n[✓] Полная очистка истории выполнена! Удалено сообщений: {deleted_msgs}\n")
|
||||||
|
return
|
||||||
|
|
||||||
|
query = """
|
||||||
|
DELETE FROM chat_messages
|
||||||
|
WHERE is_ephemeral = 1
|
||||||
|
OR content LIKE '%Предпросмотр изменений%'
|
||||||
|
OR content LIKE '%Удален пункт:%'
|
||||||
|
OR content LIKE '%добавлен пункт:%'
|
||||||
|
"""
|
||||||
|
if session_id:
|
||||||
|
cursor.execute(query + " AND session_id = ?", (session_id,))
|
||||||
|
cursor.execute("DELETE FROM session_states WHERE session_id = ?", (session_id,))
|
||||||
|
else:
|
||||||
|
cursor.execute(query)
|
||||||
|
cursor.execute("DELETE FROM session_states")
|
||||||
|
|
||||||
|
deleted_msgs = cursor.rowcount
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
print(f"\n[✓] Умная зачистка контекста выполнена! Удалено сообщений: {deleted_msgs}\n")
|
||||||
|
|
||||||
|
|
||||||
|
# ⭐️ Новые функции управления исключениями (Exceptions & Whitelist)
|
||||||
|
def print_exceptions():
|
||||||
|
"""Выводит реестр исключений и белый список сотрудников из базы SQLite."""
|
||||||
|
exc = get_all_exceptions_from_db()
|
||||||
|
print("\n" + "=" * 80)
|
||||||
|
print("📋 РЕЕСТР ИСКЛЮЧЕНИЙ И БЕЛЫЙ СПИСОК (exceptions_registry):")
|
||||||
|
print("=" * 80)
|
||||||
|
for cat, items in exc.items():
|
||||||
|
print(f"[{cat.upper()}] ({len(items)} шт.):")
|
||||||
|
if items:
|
||||||
|
for it in items:
|
||||||
|
print(f" • {it}")
|
||||||
|
else:
|
||||||
|
print(" — пусто")
|
||||||
|
print("-" * 80)
|
||||||
|
print("=" * 80 + "\n")
|
||||||
|
|
||||||
|
|
||||||
HELP_TEXT = """
|
HELP_TEXT = """
|
||||||
CLI-утилита инспекции и управления SQLite базой данных СКУД (scud_orion_ai.db)
|
CLI-утилита инспекции и управления SQLite базой данных СКУД (scud_orion_ai.db)
|
||||||
|
|
||||||
@@ -290,24 +461,38 @@ CLI-утилита инспекции и управления SQLite базой
|
|||||||
absences [ДД.ММ.ГГГГ] -- Посмотреть список официально отсутствующих из 1С:ЗУП
|
absences [ДД.ММ.ГГГГ] -- Посмотреть список официально отсутствующих из 1С:ЗУП
|
||||||
anomalies -- Посмотреть историю найденных аномалий СКУД ⟷ 1С
|
anomalies -- Посмотреть историю найденных аномалий СКУД ⟷ 1С
|
||||||
rules -- Посмотреть правила Базы Знаний ИИ из SQLite
|
rules -- Посмотреть правила Базы Знаний ИИ из SQLite
|
||||||
prompts -- Посмотреть системные промпты (system_prompts)
|
prompts -- Посмотреть узлы системного промпта (system_prompt_nodes)
|
||||||
tools -- Посмотреть реестр действий инструментов и кнопок (tool_action_registry)
|
tools -- Посмотреть реестр действий инструментов и кнопок (tool_action_registry)
|
||||||
sessions -- Посмотреть активные сессии и превью (session_states)
|
sessions -- Посмотреть активные сессии и превью (session_states)
|
||||||
|
context [session_id] [--limit N] -- Посмотреть таблицу контекста сообщений чата (chat_messages)
|
||||||
|
context purge [session_id] [--all] -- Очистить контекст (умная зачистка или полная с флагом --all)
|
||||||
dump [output.xlsx] -- Полный дамп всех таблиц БД в многостраничный Excel
|
dump [output.xlsx] -- Полный дамп всех таблиц БД в многостраничный Excel
|
||||||
snapshot del [ID] или [--day ДД.ММ.ГГГГ] -- Удаление снапшота по ID или всех за выбранный день
|
snapshot del [ID] или [--day ДД.ММ.ГГГГ] -- Удаление снапшота по ID или всех за выбранный день
|
||||||
|
|
||||||
|
exceptions [list] -- Посмотреть реестр исключений и белый список (SQLite)
|
||||||
|
exceptions add -c CATEGORY -v VALUE [-m COMMENT] -- Добавить исключение (fio, include_fio, departments, positions, position_keywords)
|
||||||
|
exceptions del -c CATEGORY -v VALUE -- Удалить исключение из БД
|
||||||
|
exceptions sync -- Синхронизировать exceptions.json -> SQLite
|
||||||
|
|
||||||
ПРИМЕРЫ ЗАПУСКА:
|
ПРИМЕРЫ ЗАПУСКА:
|
||||||
python scripts/db_cli.py stats
|
python scripts/db_cli.py stats
|
||||||
python scripts/db_cli.py snapshots 06.08.2026
|
python scripts/db_cli.py snapshots 06.08.2026
|
||||||
python scripts/db_cli.py scud 06.08.2026 --export-xlsx срез_четверг
|
python scripts/db_cli.py scud 06.08.2026 --export-xlsx срез_четверг
|
||||||
python scripts/db_cli.py scud --snapshot Y20260805-007
|
|
||||||
python scripts/db_cli.py absences 07.08.2026
|
python scripts/db_cli.py absences 07.08.2026
|
||||||
python scripts/db_cli.py prompts
|
python scripts/db_cli.py prompts
|
||||||
python scripts/db_cli.py tools
|
python scripts/db_cli.py tools
|
||||||
python scripts/db_cli.py sessions
|
python scripts/db_cli.py sessions
|
||||||
|
python scripts/db_cli.py context web_session_main --limit 20
|
||||||
|
python scripts/db_cli.py context purge
|
||||||
|
python scripts/db_cli.py context purge --all
|
||||||
|
python scripts/db_cli.py context purge web_session_main --all
|
||||||
python scripts/db_cli.py snapshot del Y20260805-007
|
python scripts/db_cli.py snapshot del Y20260805-007
|
||||||
python scripts/db_cli.py snapshot del --day 04.08.2026
|
|
||||||
python scripts/db_cli.py dump my_dump.xlsx
|
python scripts/db_cli.py dump my_dump.xlsx
|
||||||
|
python scripts/db_cli.py exceptions
|
||||||
|
python scripts/db_cli.py exceptions add -c include_fio -v "Тарасенко Александр Александрович"
|
||||||
|
python scripts/db_cli.py exceptions add -c fio -v "Михалев Сергей Геннадьевич" -m "Уборщик"
|
||||||
|
python scripts/db_cli.py exceptions del -c fio -v "Михалев Сергей Геннадьевич"
|
||||||
|
python scripts/db_cli.py exceptions sync
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
@@ -321,12 +506,21 @@ def main():
|
|||||||
formatter_class=argparse.RawDescriptionHelpFormatter,
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
||||||
add_help=False
|
add_help=False
|
||||||
)
|
)
|
||||||
parser.add_argument('command', nargs='?', default=None, choices=['stats', 'snapshots', 'scud', 'absences', 'anomalies', 'rules', 'prompts', 'sessions', 'dump', 'snapshot', 'tools'], help="Основная команда")
|
parser.add_argument('command', nargs='?', default=None, choices=[
|
||||||
parser.add_argument('action', nargs='?', default=None, help="Дополнительное действие (например, 'del')")
|
'stats', 'snapshots', 'scud', 'absences', 'anomalies',
|
||||||
parser.add_argument('param', nargs='?', default=None, help="Параметр (дата, ID снапшота или имя файла)")
|
'rules', 'prompts', 'sessions', 'dump', 'snapshot',
|
||||||
|
'tools', 'context', 'exceptions'
|
||||||
|
], help="Основная команда")
|
||||||
|
parser.add_argument('action', nargs='?', default=None, help="Действие ('del', 'purge', 'add', 'sync') или дата/сессия")
|
||||||
|
parser.add_argument('param', nargs='?', default=None, help="Параметр (дата, ID снапшота, session_id, имя файла)")
|
||||||
|
parser.add_argument('-c', '--category', type=str, default=None, choices=['departments', 'positions', 'fio', 'position_keywords', 'include_fio'], help="Категория исключения")
|
||||||
|
parser.add_argument('-v', '--value', type=str, default=None, help="Значение исключения (ФИО, отдел, должность)")
|
||||||
|
parser.add_argument('-m', '--comment', type=str, default="", help="Комментарий к исключению")
|
||||||
parser.add_argument('--snapshot', type=str, default=None, help="ID конкретного снапшота для инспекции")
|
parser.add_argument('--snapshot', type=str, default=None, help="ID конкретного снапшота для инспекции")
|
||||||
parser.add_argument('--export-xlsx', type=str, default=None, help="Экспорт среза СКУД в Excel-файл")
|
parser.add_argument('--export-xlsx', type=str, default=None, help="Экспорт среза СКУД в Excel-файл")
|
||||||
parser.add_argument('--day', type=str, default=None, help="Удалить снапшоты за конкретный день (ДД.ММ.ГГГГ)")
|
parser.add_argument('--day', type=str, default=None, help="Удалить снапшоты за конкретный день (ДД.ММ.ГГГГ)")
|
||||||
|
parser.add_argument('--limit', type=int, default=50, help="Лимит выводимых сообщений чата (для команды context)")
|
||||||
|
parser.add_argument('--all', action='store_true', help="Полная очистка всех сообщений (для команды context purge)")
|
||||||
|
|
||||||
if len(sys.argv) == 1:
|
if len(sys.argv) == 1:
|
||||||
print_stats()
|
print_stats()
|
||||||
@@ -337,12 +531,13 @@ def main():
|
|||||||
if args.command == 'stats':
|
if args.command == 'stats':
|
||||||
print_stats()
|
print_stats()
|
||||||
elif args.command == 'snapshots':
|
elif args.command == 'snapshots':
|
||||||
date_val = args.param or args.action
|
date_val = args.action or args.param
|
||||||
print_snapshots_list(date_str=date_val)
|
print_snapshots_list(date_str=date_val)
|
||||||
elif args.command == 'scud':
|
elif args.command == 'scud':
|
||||||
inspect_scud(snapshot_id=args.snapshot, date_str=args.param, export_xlsx=args.export_xlsx)
|
date_val = args.action or args.param
|
||||||
|
inspect_scud(snapshot_id=args.snapshot, date_str=date_val, export_xlsx=args.export_xlsx)
|
||||||
elif args.command == 'absences':
|
elif args.command == 'absences':
|
||||||
date_val = args.param or args.action
|
date_val = args.action or args.param
|
||||||
print_absences(date_str=date_val)
|
print_absences(date_str=date_val)
|
||||||
elif args.command == 'anomalies':
|
elif args.command == 'anomalies':
|
||||||
print_anomalies()
|
print_anomalies()
|
||||||
@@ -354,8 +549,16 @@ def main():
|
|||||||
print_session_states()
|
print_session_states()
|
||||||
elif args.command == 'tools':
|
elif args.command == 'tools':
|
||||||
print_tool_actions()
|
print_tool_actions()
|
||||||
|
elif args.command == 'context':
|
||||||
|
if args.action in ['purge', 'clear']:
|
||||||
|
is_all = args.all or (args.param == '--all')
|
||||||
|
sess_id = None if (args.param == '--all' or not args.param) else args.param
|
||||||
|
purge_chat_context(session_id=sess_id, purge_all=is_all)
|
||||||
|
else:
|
||||||
|
sess_id = args.action if args.action else None
|
||||||
|
print_chat_messages(session_id=sess_id, limit=args.limit)
|
||||||
elif args.command == 'dump':
|
elif args.command == 'dump':
|
||||||
filename = args.param if args.param else "db_dump_full.xlsx"
|
filename = args.action or args.param or "db_dump_full.xlsx"
|
||||||
dump_all_to_excel(filename)
|
dump_all_to_excel(filename)
|
||||||
elif args.command == 'snapshot':
|
elif args.command == 'snapshot':
|
||||||
if args.action == 'del':
|
if args.action == 'del':
|
||||||
@@ -369,9 +572,34 @@ def main():
|
|||||||
else:
|
else:
|
||||||
print(f"\n[❌] Ошибка: Неизвестное действие '{args.action}' для команды snapshot.")
|
print(f"\n[❌] Ошибка: Неизвестное действие '{args.action}' для команды snapshot.")
|
||||||
print("Используйте: python scripts/db_cli.py snapshot del [ID или --day 'ДД.ММ.ГГГГ']\n")
|
print("Используйте: python scripts/db_cli.py snapshot del [ID или --day 'ДД.ММ.ГГГГ']\n")
|
||||||
|
elif args.command == 'exceptions':
|
||||||
|
if args.action == 'add':
|
||||||
|
if not args.category or not args.value:
|
||||||
|
print("\n[❌] Ошибка: Для добавления исключения укажите флаги -c/--category и -v/--value")
|
||||||
|
print("Пример: python scripts/db_cli.py exceptions add -c include_fio -v \"Тарасенко Александр Александрович\"\n")
|
||||||
|
return
|
||||||
|
if add_exception_to_db(args.category, args.value, args.comment):
|
||||||
|
print(f"\n[✓] Успешно добавлено исключение: [{args.category}] {args.value}\n")
|
||||||
|
else:
|
||||||
|
print(f"\n[❌] Ошибка добавления исключения [{args.category}] {args.value}\n")
|
||||||
|
elif args.action in ['del', 'delete', 'remove']:
|
||||||
|
if not args.category or not args.value:
|
||||||
|
print("\n[❌] Ошибка: Для удаления исключения укажите флаги -c/--category и -v/--value")
|
||||||
|
print("Пример: python scripts/db_cli.py exceptions del -c fio -v \"Михалев Сергей Геннадьевич\"\n")
|
||||||
|
return
|
||||||
|
if remove_exception_from_db(args.category, args.value):
|
||||||
|
print(f"\n[✓] Успешно удалено исключение: [{args.category}] {args.value}\n")
|
||||||
|
else:
|
||||||
|
print(f"\n[⚠️] Запись не найдена в базе: [{args.category}] {args.value}\n")
|
||||||
|
elif args.action == 'sync':
|
||||||
|
sync_json_to_db()
|
||||||
|
print("\n[✓] Синхронизация exceptions.json -> SQLite успешно завершена.\n")
|
||||||
|
else:
|
||||||
|
print_exceptions()
|
||||||
else:
|
else:
|
||||||
print(f"\n[❌] Ошибка: Неизвестная команда.")
|
print("\n[❌] Ошибка: Неизвестная команда.")
|
||||||
print(HELP_TEXT)
|
print(HELP_TEXT)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
@@ -0,0 +1,58 @@
|
|||||||
|
import os
|
||||||
|
import sqlite3
|
||||||
|
|
||||||
|
DB_PATH = os.path.join(os.path.dirname(__file__), "../../data/scud_orion_ai.db")
|
||||||
|
if not os.path.exists(DB_PATH):
|
||||||
|
DB_PATH = "data/scud_orion_ai.db"
|
||||||
|
|
||||||
|
conn = sqlite3.connect(DB_PATH)
|
||||||
|
cursor = conn.cursor()
|
||||||
|
|
||||||
|
print("=" * 90)
|
||||||
|
print(f"🔍 ИНСПЕКЦИЯ СООБЩЕНИЙ В БАЗЕ: {DB_PATH}")
|
||||||
|
print("=" * 90)
|
||||||
|
|
||||||
|
# 1. Проверяем наличие колонки is_ephemeral
|
||||||
|
cursor.execute("PRAGMA table_info(chat_messages);")
|
||||||
|
cols = [c[1] for c in cursor.fetchall()]
|
||||||
|
|
||||||
|
if "is_ephemeral" not in cols:
|
||||||
|
print("❌ Колонка is_ephemeral отсутствует в таблице chat_messages!")
|
||||||
|
conn.close()
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
# 2. Считаем общее количество и распределение
|
||||||
|
cursor.execute("SELECT COUNT(*) FROM chat_messages")
|
||||||
|
total_count = cursor.fetchone()[0]
|
||||||
|
|
||||||
|
cursor.execute("SELECT COUNT(*) FROM chat_messages WHERE is_ephemeral = 1")
|
||||||
|
ephemeral_count = cursor.fetchone()[0]
|
||||||
|
|
||||||
|
cursor.execute("SELECT COUNT(*) FROM chat_messages WHERE is_ephemeral = 0")
|
||||||
|
permanent_count = cursor.fetchone()[0]
|
||||||
|
|
||||||
|
print(f"Всего записей в истории: {total_count}")
|
||||||
|
print(f" • Постоянные (is_ephemeral = 0): {permanent_count}")
|
||||||
|
print(f" • Фантомные / Эфемерные (is_ephemeral = 1): {ephemeral_count}")
|
||||||
|
print("-" * 90)
|
||||||
|
|
||||||
|
# 3. Выводим все фантомные записи
|
||||||
|
if ephemeral_count == 0:
|
||||||
|
print("В базе нет фантомных записей (is_ephemeral = 1 пуст).")
|
||||||
|
else:
|
||||||
|
print(f"📋 СПИСОК ВСЕХ ФАНТОМНЫХ СООБЩЕНИЙ ({ephemeral_count} шт.):")
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT id, session_id, role, created_at, content
|
||||||
|
FROM chat_messages
|
||||||
|
WHERE is_ephemeral = 1
|
||||||
|
ORDER BY id ASC
|
||||||
|
""")
|
||||||
|
for row in cursor.fetchall():
|
||||||
|
msg_id, session_id, role, created_at, content = row
|
||||||
|
print(f"\n[ID: {msg_id}] | Сессия: {session_id} | Роль: {role} | Время: {created_at}")
|
||||||
|
print("Содержимое:")
|
||||||
|
print(content.strip())
|
||||||
|
print("-" * 70)
|
||||||
|
|
||||||
|
conn.close()
|
||||||
|
print("=" * 90)
|
||||||
@@ -1,31 +0,0 @@
|
|||||||
import os
|
|
||||||
|
|
||||||
OUTPUT_SNAPSHOT = "project_code_snapshot.md"
|
|
||||||
|
|
||||||
# Расширения файлов для включения в снимок
|
|
||||||
ALLOWED_EXTENSIONS = {'.py', '.json', '.md', '.sh', '.ini'}
|
|
||||||
EXCLUDE_DIRS = {'.git', '__pycache__', 'venv', '.venv', 'output', 'logs', 'extracted_project'}
|
|
||||||
EXCLUDE_FILES = {OUTPUT_SNAPSHOT, 'scud_orion_ai_v2.tar.gz', 'context_memory.db'}
|
|
||||||
|
|
||||||
print(f"🔄 Сборка полного контекстного слепка проекта в {OUTPUT_SNAPSHOT}...")
|
|
||||||
|
|
||||||
with open(OUTPUT_SNAPSHOT, 'w', encoding='utf-8') as out:
|
|
||||||
out.write("# 📦 ПОЛНЫЙ ИСХОДНЫЙ КОД И КОНФИГУРАЦИЯ ПРОЕКТА scud_orion_ai_v2\n\n")
|
|
||||||
|
|
||||||
for root, dirs, files in os.walk('.'):
|
|
||||||
dirs[:] = [d for d in dirs if d not in EXCLUDE_DIRS]
|
|
||||||
|
|
||||||
for file in sorted(files):
|
|
||||||
ext = os.path.splitext(file)[1].lower()
|
|
||||||
if ext in ALLOWED_EXTENSIONS and file not in EXCLUDE_FILES:
|
|
||||||
filepath = os.path.join(root, file)
|
|
||||||
out.write(f"## File: `{filepath}`\n")
|
|
||||||
out.write("```" + (ext.replace('.', '') if ext != '.md' else '') + "\n")
|
|
||||||
try:
|
|
||||||
with open(filepath, 'r', encoding='utf-8', errors='replace') as f:
|
|
||||||
out.write(f.read())
|
|
||||||
except Exception as e:
|
|
||||||
out.write(f"// Ошибка чтения файла: {e}\n")
|
|
||||||
out.write("\n```\n\n")
|
|
||||||
|
|
||||||
print(f"✓ Успешно создан слепок проекта: {OUTPUT_SNAPSHOT} ({os.path.getsize(OUTPUT_SNAPSHOT):,} bytes)")
|
|
||||||
@@ -0,0 +1,73 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: scripts/diagnostics/make_etl_snapshot.py
|
||||||
|
ROLE: Генерация компактного слепка ETL-конвейера, БД и сервисов СКУД.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
|
||||||
|
ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
|
||||||
|
OUTPUT_FILE = os.path.join(ROOT_DIR, "etl_code_snapshot.md")
|
||||||
|
|
||||||
|
# Строгий целевой список файлов для ETL-слепка
|
||||||
|
TARGET_FILES = [
|
||||||
|
"config.py",
|
||||||
|
"exceptions.json",
|
||||||
|
"main_etl.py",
|
||||||
|
"run_cron_etl.sh",
|
||||||
|
"scripts/db_cli.py", # ⭐️ Гарантированно включен
|
||||||
|
"core/connection.py",
|
||||||
|
"core/database.py",
|
||||||
|
"core/schema.py",
|
||||||
|
"core/repositories/scud_repo.py",
|
||||||
|
"core/repositories/zup_repo.py",
|
||||||
|
"services/data_loader.py",
|
||||||
|
"services/scud_export.py",
|
||||||
|
"services/share_copier.py",
|
||||||
|
"services/excel_exporter.py",
|
||||||
|
"services/text_reporter.py",
|
||||||
|
"services/exceptions_repo.py",
|
||||||
|
"services/zup_extractor.py",
|
||||||
|
"services/ai_verifier.py",
|
||||||
|
"services/knowledge_base.py",
|
||||||
|
"services/knowledge/service.py",
|
||||||
|
"services/scud_etl/pipeline.py",
|
||||||
|
"services/scud_etl/merger.py",
|
||||||
|
"services/scud_etl/anomaly_detector.py",
|
||||||
|
"services/snapshots/service.py",
|
||||||
|
"services/tasks/repository.py",
|
||||||
|
"services/tasks/service.py"
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
def create_etl_snapshot():
|
||||||
|
content = ["# 📦 ETL-СЛЕПОК ИСХОДНОГО КОДА (СКУД ⟷ 1С & DB CORE)\n"]
|
||||||
|
included_count = 0
|
||||||
|
|
||||||
|
for rel_path in TARGET_FILES:
|
||||||
|
full_path = os.path.join(ROOT_DIR, rel_path)
|
||||||
|
if os.path.exists(full_path):
|
||||||
|
ext = os.path.splitext(rel_path)[1].replace(".", "")
|
||||||
|
lang = "py" if ext == "py" else ("json" if ext == "json" else "bash")
|
||||||
|
try:
|
||||||
|
with open(full_path, "r", encoding="utf-8") as f:
|
||||||
|
file_text = f.read()
|
||||||
|
content.append(f"## File: `./{rel_path}`\n```{lang}\n{file_text}\n```\n")
|
||||||
|
included_count += 1
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[⚠️] Ошибка чтения {rel_path}: {e}")
|
||||||
|
else:
|
||||||
|
print(f"[ℹ️] Пропущен отсутствующий файл: {rel_path}")
|
||||||
|
|
||||||
|
with open(OUTPUT_FILE, "w", encoding="utf-8") as f:
|
||||||
|
f.write("\n".join(content))
|
||||||
|
|
||||||
|
size_kb = os.path.getsize(OUTPUT_FILE) / 1024
|
||||||
|
print(f"\n[✓] ETL-слепок создан: {OUTPUT_FILE}")
|
||||||
|
print(f" Включено файлов: {included_count} | Размер: {size_kb:.1f} KB\n")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
create_etl_snapshot()
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: scripts/diagnostics/make_web_snapshot.py
|
||||||
|
ROLE: Генерация слепка Web API, LLM-движка и клиентских скриптов.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
|
||||||
|
ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
|
||||||
|
OUTPUT_FILE = os.path.join(ROOT_DIR, "web_api_code_snapshot.md")
|
||||||
|
|
||||||
|
WEB_TARGET_FILES = [
|
||||||
|
"modules/web_api/main.py",
|
||||||
|
"modules/web_api/routers/chat.py",
|
||||||
|
"modules/web_api/routers/auth.py",
|
||||||
|
"modules/web_api/routers/tasks.py",
|
||||||
|
"modules/web_api/routers/exceptions.py",
|
||||||
|
"modules/web_api/routers/admin.py",
|
||||||
|
"modules/web_api/routers/files.py",
|
||||||
|
"modules/web_api/llm/agent.py",
|
||||||
|
"modules/web_api/llm/db_tools.py",
|
||||||
|
"modules/web_api/llm/schemas.py",
|
||||||
|
"modules/web_api/llm/core/context_manager.py",
|
||||||
|
"modules/web_api/llm/core/tool_injector.py",
|
||||||
|
"modules/web_api/llm/core/ollama_client.py",
|
||||||
|
"modules/web_api/llm/core/fast_path.py",
|
||||||
|
"modules/web_api/static/js/app.js",
|
||||||
|
"modules/web_api/static/js/auth.js",
|
||||||
|
"modules/web_api/static/js/tasks.js",
|
||||||
|
"modules/web_api/static/js/chat/core.js",
|
||||||
|
"modules/web_api/static/js/chat/task_widget.js"
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
def create_web_snapshot():
|
||||||
|
content = ["# 🌐 WEB API & LLM AGENT CODE SNAPSHOT\n"]
|
||||||
|
included_count = 0
|
||||||
|
|
||||||
|
for rel_path in WEB_TARGET_FILES:
|
||||||
|
full_path = os.path.join(ROOT_DIR, rel_path)
|
||||||
|
if os.path.exists(full_path):
|
||||||
|
ext = os.path.splitext(rel_path)[1].replace(".", "")
|
||||||
|
lang = "js" if ext == "js" else ("py" if ext == "py" else "text")
|
||||||
|
try:
|
||||||
|
with open(full_path, "r", encoding="utf-8") as f:
|
||||||
|
file_text = f.read()
|
||||||
|
content.append(f"## File: `./{rel_path}`\n```{lang}\n{file_text}\n```\n")
|
||||||
|
included_count += 1
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[⚠️] Ошибка чтения {rel_path}: {e}")
|
||||||
|
|
||||||
|
with open(OUTPUT_FILE, "w", encoding="utf-8") as f:
|
||||||
|
f.write("\n".join(content))
|
||||||
|
|
||||||
|
size_kb = os.path.getsize(OUTPUT_FILE) / 1024
|
||||||
|
print(f"\n[✓] Web API слепок создан: {OUTPUT_FILE}")
|
||||||
|
print(f" Включено файлов: {included_count} | Размер: {size_kb:.1f} KB\n")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
create_web_snapshot()
|
||||||
@@ -0,0 +1,22 @@
|
|||||||
|
import os
|
||||||
|
|
||||||
|
OUTPUT_SNAPSHOT = "web_api_code_snapshot.md"
|
||||||
|
ALLOWED_EXTENSIONS = {'.py', '.html', '.css', '.js'}
|
||||||
|
EXCLUDE_DIRS = {'.git', '__pycache__', 'venv', '.venv', 'output', 'logs', 'extracted_project', 'docs', 'data'}
|
||||||
|
|
||||||
|
print(f"🔄 Сборка Web API слепка кода в {OUTPUT_SNAPSHOT}...")
|
||||||
|
with open(OUTPUT_SNAPSHOT, 'w', encoding='utf-8') as out:
|
||||||
|
out.write("# 📦 WEB API & AI СЛЕПОК ИСХОДНОГО КОДА\n\n")
|
||||||
|
if os.path.exists('modules'):
|
||||||
|
for r, d, files in os.walk('modules'):
|
||||||
|
d[:] = [sub for sub in d if sub not in EXCLUDE_DIRS]
|
||||||
|
for file in sorted(files):
|
||||||
|
ext = os.path.splitext(file)[1].lower()
|
||||||
|
if ext in ALLOWED_EXTENSIONS:
|
||||||
|
filepath = os.path.join(r, file)
|
||||||
|
out.write(f"## File: `./{filepath}`\n```" + ext.replace('.', '') + "\n")
|
||||||
|
with open(filepath, 'r', encoding='utf-8', errors='replace') as f:
|
||||||
|
out.write(f.read())
|
||||||
|
out.write("\n```\n\n")
|
||||||
|
|
||||||
|
print(f"✓ Web API слепок готов: {OUTPUT_SNAPSHOT} ({os.path.getsize(OUTPUT_SNAPSHOT) / 1024:.1f} KB)")
|
||||||
+30
-11
@@ -151,7 +151,7 @@ def ai_verify_scud_against_staff(unrecognized_scud_fios, staff_fios):
|
|||||||
if not unrecognized_scud_fios or not staff_fios:
|
if not unrecognized_scud_fios or not staff_fios:
|
||||||
return {}
|
return {}
|
||||||
|
|
||||||
# 🛡 Исключаем точные совпадения (чтобы ИИ не совершал ложных замен)
|
# Исключаем точные совпадения
|
||||||
staff_fios_set = set(staff_fios)
|
staff_fios_set = set(staff_fios)
|
||||||
real_unrecognized = [f for f in unrecognized_scud_fios if f not in staff_fios_set]
|
real_unrecognized = [f for f in unrecognized_scud_fios if f not in staff_fios_set]
|
||||||
|
|
||||||
@@ -163,7 +163,7 @@ def ai_verify_scud_against_staff(unrecognized_scud_fios, staff_fios):
|
|||||||
prompt = f"""
|
prompt = f"""
|
||||||
Ты — кадровый аудитор безопасности СКУД.
|
Ты — кадровый аудитор безопасности СКУД.
|
||||||
{rules_context}
|
{rules_context}
|
||||||
В СКУД записаны неопознанные ФИО: {json.dumps(unrecognized_scud_fios, ensure_ascii=False)}
|
В СКУД записаны неопознанные ФИО: {json.dumps(real_unrecognized, ensure_ascii=False)}
|
||||||
В официальном Штатном расписании 1С записаны ЭТАЛОНЫ: {json.dumps(staff_fios, ensure_ascii=False)}
|
В официальном Штатном расписании 1С записаны ЭТАЛОНЫ: {json.dumps(staff_fios, ensure_ascii=False)}
|
||||||
|
|
||||||
СТРОГИЕ ПРАВИЛА:
|
СТРОГИЕ ПРАВИЛА:
|
||||||
@@ -173,40 +173,59 @@ def ai_verify_scud_against_staff(unrecognized_scud_fios, staff_fios):
|
|||||||
4. В поле "warning" опиши обнаруженную опечатку.
|
4. В поле "warning" опиши обнаруженную опечатку.
|
||||||
5. Запрещено выдумывать опечатки и объединять разных людей/однофамильцев!
|
5. Запрещено выдумывать опечатки и объединять разных людей/однофамильцев!
|
||||||
|
|
||||||
ОТВЕЧАЙ ТОЛЬКО ИСКЛЮЧИТЕЛЬНО В ФОРМАТЕ JSON!
|
ОТВЕЧАЙ ТОЛЬКО ИСКЛЮЧИТЕЛЬНО В ФОРМАТЕ ВАЛИДНОГО JSON:
|
||||||
|
|
||||||
Формат ответа:
|
|
||||||
{{
|
{{
|
||||||
"verified_matches": [
|
"verified_matches": [
|
||||||
{{
|
{{
|
||||||
"scud_fio": "ФИО из СКУД",
|
"scud_fio": "ФИО из СКУД",
|
||||||
"staff_fio": "эталон ФИО из Штат 1С",
|
"staff_fio": "эталон ФИО из Штат 1С",
|
||||||
"warning": "Описание опечатки в СКУД или 'Точное совпадение (без опечаток)'"
|
"warning": "Описание опечатки в СКУД"
|
||||||
}}
|
}}
|
||||||
]
|
]
|
||||||
}}
|
}}
|
||||||
"""
|
"""
|
||||||
raw_response = ask_ollama(
|
raw_response = ask_ollama(
|
||||||
prompt,
|
prompt,
|
||||||
system_prompt="Ты — JSON API. Выдавай ТОЛЬКО валидный JSON без markdown-разметки."
|
system_prompt="Ты — строгий JSON API генератор. Отвечай только валидным JSON объектом без пояснительного текста."
|
||||||
)
|
)
|
||||||
|
|
||||||
mapping = {}
|
mapping = {}
|
||||||
|
if not raw_response:
|
||||||
|
return mapping
|
||||||
|
|
||||||
|
# Фаза 1: Попытка прямого разбора с санитарной очисткой
|
||||||
try:
|
try:
|
||||||
match = re.search(r'\{.*\}', raw_response, re.DOTALL)
|
match = re.search(r'\{.*\}', raw_response, re.DOTALL)
|
||||||
if match:
|
if match:
|
||||||
json_str = match.group(0)
|
json_str = match.group(0)
|
||||||
json_str = re.sub(r'\}\s*[^}\]]*$', '}', json_str)
|
# Убираем висячие запятые: {"a": 1,} -> {"a": 1}
|
||||||
|
json_str = re.sub(r',\s*([\}\]])', r'\1', json_str)
|
||||||
|
# Заменяем одинарные кавычки в ключах/значениях на двойные при необходимости
|
||||||
|
json_str = re.sub(r"(?<=\{|\,)\s*'([^']+)'\s*:", r'"\1":', json_str)
|
||||||
|
|
||||||
data = json.loads(json_str)
|
data = json.loads(json_str)
|
||||||
for item in data.get("verified_matches", []):
|
for item in data.get("verified_matches", []):
|
||||||
scud_f = item.get("scud_fio")
|
scud_f = item.get("scud_fio")
|
||||||
staff_f = item.get("staff_fio")
|
staff_f = item.get("staff_fio")
|
||||||
warn = item.get("warning", "Точное совпадение (без опечаток)")
|
warn = item.get("warning", "Точное совпадение (без опечаток)")
|
||||||
if scud_f and staff_f:
|
if scud_f and staff_f and staff_f in staff_fios_set:
|
||||||
mapping[scud_f] = {"staff_fio": staff_f, "warning": warn}
|
mapping[scud_f] = {"staff_fio": staff_f, "warning": warn}
|
||||||
|
return mapping
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Фаза 2: Резервный Regex-парсер (если JSON синтаксически сломан, но пары ключ-значение есть)
|
||||||
|
try:
|
||||||
|
pattern = r'["\']scud_fio["\']\s*:\s*["\']([^"\']+)["\'].*?["\']staff_fio["\']\s*:\s*["\']([^"\']+)["\']'
|
||||||
|
matches = re.findall(pattern, raw_response, re.DOTALL)
|
||||||
|
for scud_f, staff_f in matches:
|
||||||
|
scud_clean = scud_f.strip()
|
||||||
|
staff_clean = staff_f.strip()
|
||||||
|
if staff_clean in staff_fios_set:
|
||||||
|
mapping[scud_clean] = {"staff_fio": staff_clean, "warning": "Восстановлено парсером опечаток"}
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"[!] Ошибка разбора JSON от ИИ при сверке опечаток: {e}")
|
print(f"[!] Ошибка резервного парсинга опечаток: {e}")
|
||||||
|
|
||||||
return mapping
|
return mapping
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
+55
-156
@@ -1,11 +1,17 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/data_loader.py
|
||||||
|
ROLE: Надежная загрузка штата и отсутствий (MS SQL ЗУП -> Резервный Excel).
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import pandas as pd
|
|
||||||
import warnings
|
import warnings
|
||||||
|
import pandas as pd
|
||||||
from config import (
|
from config import (
|
||||||
normalize_fio, clean_scud_fio_light, load_exceptions,
|
normalize_fio, clean_scud_fio_light, load_exceptions,
|
||||||
DATE_TODAY, DATE_YESTERDAY, find_dated_file, ZUP_1C_DIR, SCUD_DIR
|
DATE_TODAY, DATE_YESTERDAY, find_dated_file, ZUP_1C_DIR, SCUD_DIR, DATA_DIR
|
||||||
)
|
)
|
||||||
# Импортируем прямую SQL-выгрузку отсутствий из 1С и функции работы с SQLite
|
|
||||||
from services.zup_extractor import fetch_zup_absences_from_sql
|
from services.zup_extractor import fetch_zup_absences_from_sql
|
||||||
from core.database import (
|
from core.database import (
|
||||||
load_scud_from_db_by_snapshot,
|
load_scud_from_db_by_snapshot,
|
||||||
@@ -16,23 +22,8 @@ from core.database import (
|
|||||||
warnings.filterwarnings('ignore', category=UserWarning, module='pandas')
|
warnings.filterwarnings('ignore', category=UserWarning, module='pandas')
|
||||||
|
|
||||||
|
|
||||||
def is_excluded(fio, dept, pos, exceptions):
|
|
||||||
"""Проверяет, входит ли сотрудник или должность в список исключений."""
|
|
||||||
fio_clean = normalize_fio(fio)
|
|
||||||
dept_str = str(dept).strip().upper() if pd.notna(dept) else ""
|
|
||||||
pos_str = str(pos).strip().lower() if pd.notna(pos) else ""
|
|
||||||
|
|
||||||
if fio_clean in exceptions["fio"] or dept_str in exceptions["departments"] or pos_str in exceptions["positions"]:
|
|
||||||
return True
|
|
||||||
if any(keyword in pos_str for keyword in exceptions["position_keywords"]):
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
def load_staff_data(date_str):
|
def load_staff_data(date_str):
|
||||||
"""
|
"""Загружает файл Штатного расписания 1С."""
|
||||||
Загружает файл Штатного расписания 1С за конкретную дату из папки 1c.
|
|
||||||
"""
|
|
||||||
filepath = find_dated_file("Штат", date_str)
|
filepath = find_dated_file("Штат", date_str)
|
||||||
if not filepath:
|
if not filepath:
|
||||||
fallback_path = os.path.join(ZUP_1C_DIR, "штат.xlsx")
|
fallback_path = os.path.join(ZUP_1C_DIR, "штат.xlsx")
|
||||||
@@ -43,8 +34,6 @@ def load_staff_data(date_str):
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
df_raw = pd.read_excel(filepath, skiprows=8)
|
df_raw = pd.read_excel(filepath, skiprows=8)
|
||||||
|
|
||||||
# Индексы столбцов в файле 1С: 1 — ФИО, 5 — Подразделение, 12 — Должность
|
|
||||||
df_staff = df_raw.iloc[:, [1, 5, 12]].copy()
|
df_staff = df_raw.iloc[:, [1, 5, 12]].copy()
|
||||||
df_staff.columns = ['ФИО', 'Подразделение', 'Должность']
|
df_staff.columns = ['ФИО', 'Подразделение', 'Должность']
|
||||||
|
|
||||||
@@ -57,30 +46,63 @@ def load_staff_data(date_str):
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def load_absent_from_excel(date_str):
|
||||||
|
"""Резервное чтение файла Отсутствия_ДД_ММ_ГГГГ.xlsx."""
|
||||||
|
filepath = find_dated_file("Отсутствия", date_str)
|
||||||
|
if not filepath:
|
||||||
|
return None
|
||||||
|
|
||||||
|
try:
|
||||||
|
for skip in [0, 1, 2, 3, 4, 5, 8]:
|
||||||
|
df_try = pd.read_excel(filepath, skiprows=skip)
|
||||||
|
fio_col = None
|
||||||
|
reason_col = None
|
||||||
|
for col in df_try.columns:
|
||||||
|
c_str = str(col).lower()
|
||||||
|
if ('фио' in c_str or 'сотрудник' in c_str) and fio_col is None:
|
||||||
|
fio_col = col
|
||||||
|
if ('вид' in c_str or 'причина' in c_str or 'отсутств' in c_str) and reason_col is None:
|
||||||
|
reason_col = col
|
||||||
|
|
||||||
|
if fio_col and reason_col:
|
||||||
|
df_res = df_try[[fio_col, reason_col]].copy()
|
||||||
|
df_res.columns = ['ФИО', 'Вид_отсутствия']
|
||||||
|
df_res = df_res.dropna(subset=['ФИО', 'Вид_отсутствия'])
|
||||||
|
df_res = df_res[~df_res['ФИО'].astype(str).str.contains('Всего|Организация|Сотрудник|ФИО|ЛЕНМОРНИИПРОЕКТ', case=False, na=False)]
|
||||||
|
df_res['fio_clean'] = df_res['ФИО'].apply(normalize_fio)
|
||||||
|
print(f" [✓] Резервный Excel: загружено {len(df_res)} записей отсутствий из {os.path.basename(filepath)}")
|
||||||
|
return df_res[['fio_clean', 'Вид_отсутствия']].dropna(subset=['fio_clean'])
|
||||||
|
except Exception as e:
|
||||||
|
print(f" [⚠️] Ошибка чтения резервного Excel отсутствий {filepath}: {e}")
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
def load_absent_data(date_str):
|
def load_absent_data(date_str):
|
||||||
"""
|
"""
|
||||||
Загружает документально подтвержденные отсутствия сотрудников
|
Загружает отсутствия из MS SQL 1С:ЗУП, а при сбое — из локального Excel.
|
||||||
НАПРЯМУЮ из базы MS SQL 1С:ЗУП за указанную дату,
|
|
||||||
а также автоматически обогащает их удаленщиками из static_reason_workers.csv.
|
|
||||||
"""
|
"""
|
||||||
df_absent = None
|
df_absent = None
|
||||||
try:
|
try:
|
||||||
df_absent = fetch_zup_absences_from_sql(date_str)
|
df_absent = fetch_zup_absences_from_sql(date_str)
|
||||||
|
|
||||||
if df_absent is not None and not df_absent.empty:
|
if df_absent is not None and not df_absent.empty:
|
||||||
|
print(f" [✓] MS SQL ЗУП: получено {len(df_absent)} записей отсутствий за {date_str}")
|
||||||
df_absent = df_absent.dropna(subset=['ФИО', 'Вид_отсутствия'])
|
df_absent = df_absent.dropna(subset=['ФИО', 'Вид_отсутствия'])
|
||||||
df_absent = df_absent[~df_absent['ФИО'].astype(str).str.contains('АО "ЛЕНМОРНИИПРОЕКТ"|Сотрудник', case=False, na=False)]
|
df_absent = df_absent[~df_absent['ФИО'].astype(str).str.contains('АО "ЛЕНМОРНИИПРОЕКТ"|Сотрудник', case=False, na=False)]
|
||||||
df_absent['fio_clean'] = df_absent['ФИО'].apply(normalize_fio)
|
df_absent['fio_clean'] = df_absent['ФИО'].apply(normalize_fio)
|
||||||
df_absent = df_absent[['fio_clean', 'Вид_отсутствия']].dropna(subset=['fio_clean'])
|
df_absent = df_absent[['fio_clean', 'Вид_отсутствия']].dropna(subset=['fio_clean'])
|
||||||
else:
|
|
||||||
df_absent = pd.DataFrame(columns=['fio_clean', 'Вид_отсутствия'])
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"[⚠️] Ошибка SQL-выгрузки отсутствий за {date_str}: {e}")
|
print(f" [⚠️] Ошибка подключения к MS SQL ЗУП за {date_str}: {e}")
|
||||||
|
|
||||||
|
# Резервный источник: Excel файл из 1С с сетевой шары
|
||||||
|
if df_absent is None or df_absent.empty:
|
||||||
|
df_absent = load_absent_from_excel(date_str)
|
||||||
|
|
||||||
|
if df_absent is None:
|
||||||
df_absent = pd.DataFrame(columns=['fio_clean', 'Вид_отсутствия'])
|
df_absent = pd.DataFrame(columns=['fio_clean', 'Вид_отсутствия'])
|
||||||
|
|
||||||
# ⭐️ ОБОГАЩЕНИЕ УДАЛЕНЩИКАМИ ИЗ static_reason_workers.csv
|
# Обогащение удаленщиками из static_reason_workers.csv
|
||||||
try:
|
try:
|
||||||
from config import DATA_DIR
|
|
||||||
static_path = os.path.join(DATA_DIR, "static_reason_workers.csv")
|
static_path = os.path.join(DATA_DIR, "static_reason_workers.csv")
|
||||||
if os.path.exists(static_path):
|
if os.path.exists(static_path):
|
||||||
df_static = pd.read_csv(static_path, encoding='utf-8')
|
df_static = pd.read_csv(static_path, encoding='utf-8')
|
||||||
@@ -97,18 +119,14 @@ def load_absent_data(date_str):
|
|||||||
})
|
})
|
||||||
if new_rows:
|
if new_rows:
|
||||||
df_absent = pd.concat([df_absent, pd.DataFrame(new_rows)], ignore_index=True)
|
df_absent = pd.concat([df_absent, pd.DataFrame(new_rows)], ignore_index=True)
|
||||||
|
print(f" [✓] Реестр удаленщиков: добавлено {len(new_rows)} чел. из static_reason_workers.csv")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"[⚠️] Ошибка обогащения удаленщиками в load_absent_data: {e}")
|
print(f" [⚠️] Ошибка чтения static_reason_workers.csv: {e}")
|
||||||
|
|
||||||
return df_absent if not df_absent.empty else None
|
return df_absent if not df_absent.empty else None
|
||||||
|
|
||||||
|
|
||||||
def load_1c_data_smart(date_str, use_db=False):
|
def load_1c_data_smart(date_str, use_db=False):
|
||||||
"""
|
|
||||||
Универсальный загрузчик данных 1С (Штат и Отсутствия).
|
|
||||||
Если use_db=True, приоритетно извлекает сохраненные данные из SQLite БД
|
|
||||||
за целевую дату указанного снапшота.
|
|
||||||
"""
|
|
||||||
df_staff = None
|
df_staff = None
|
||||||
df_absent = None
|
df_absent = None
|
||||||
|
|
||||||
@@ -116,128 +134,9 @@ def load_1c_data_smart(date_str, use_db=False):
|
|||||||
df_staff = load_staff_from_db(date_str)
|
df_staff = load_staff_from_db(date_str)
|
||||||
df_absent = load_absences_from_db(date_str)
|
df_absent = load_absences_from_db(date_str)
|
||||||
|
|
||||||
# Если в базе нет записей за эту дату или use_db=False — загружаем из файлов/SQL
|
|
||||||
if df_staff is None:
|
if df_staff is None:
|
||||||
df_staff = load_staff_data(date_str)
|
df_staff = load_staff_data(date_str)
|
||||||
if df_absent is None:
|
if df_absent is None:
|
||||||
df_absent = load_absent_data(date_str)
|
df_absent = load_absent_data(date_str)
|
||||||
|
|
||||||
return df_staff, df_absent
|
return df_staff, df_absent
|
||||||
|
|
||||||
|
|
||||||
def process_scud_anomalies(df_scud, exceptions=None):
|
|
||||||
"""
|
|
||||||
Вычисляет 'Первую активность' и маркирует аномалии СКУД.
|
|
||||||
"""
|
|
||||||
if exceptions is None:
|
|
||||||
exceptions = load_exceptions()
|
|
||||||
|
|
||||||
start_col = 'Начало_дня' if 'Начало_дня' in df_scud.columns else ('Начало дня' if 'Начало дня' in df_scud.columns else None)
|
|
||||||
end_col = 'Конец_дня' if 'Конец_дня' in df_scud.columns else ('Конец дня' if 'Конец дня' in df_scud.columns else None)
|
|
||||||
status_col = 'Статус' if 'Статус' in df_scud.columns else None
|
|
||||||
|
|
||||||
first_activities = []
|
|
||||||
anomaly_flags = []
|
|
||||||
is_present_flags = []
|
|
||||||
|
|
||||||
for _, row in df_scud.iterrows():
|
|
||||||
fio = str(row.get('Сотрудник', row.get('fio_raw', ''))).strip()
|
|
||||||
fio_clean = normalize_fio(fio)
|
|
||||||
dept = str(row.get('Подразделение', '')).strip()
|
|
||||||
pos = str(row.get('Должность', '')).strip()
|
|
||||||
|
|
||||||
val_start = str(row.get(start_col, '')).strip() if start_col else ''
|
|
||||||
val_end = str(row.get(end_col, '')).strip() if end_col else ''
|
|
||||||
val_status = str(row.get(status_col, '')).strip().lower() if status_col else ''
|
|
||||||
|
|
||||||
raw_first_event = row.get('raw_first_event', None)
|
|
||||||
if pd.isna(raw_first_event) or str(raw_first_event).strip() in ['', 'None', 'nan', '00:00:00']:
|
|
||||||
raw_first_event = None
|
|
||||||
|
|
||||||
has_no_in = (val_start in ['нет входа', 'none', 'nan', '', '00:00:00', '00:00'])
|
|
||||||
has_no_out = (val_end in ['нет выхода', 'none', 'nan', '', '00:00:00', '00:00'])
|
|
||||||
|
|
||||||
first_act = '—'
|
|
||||||
anom_flag = 'NONE'
|
|
||||||
is_present = not ('отсутствовал' in val_status or has_no_in)
|
|
||||||
|
|
||||||
if has_no_in:
|
|
||||||
if raw_first_event is not None or not has_no_out:
|
|
||||||
first_act = str(raw_first_event) if raw_first_event else val_end
|
|
||||||
anom_flag = 'ANOMALY_NO_IN_HAS_ACTIVITY'
|
|
||||||
|
|
||||||
if not is_excluded(fio, dept, pos, exceptions):
|
|
||||||
is_present = False
|
|
||||||
else:
|
|
||||||
first_act = '—'
|
|
||||||
is_present = False
|
|
||||||
|
|
||||||
first_activities.append(first_act)
|
|
||||||
anomaly_flags.append(anom_flag)
|
|
||||||
is_present_flags.append(is_present)
|
|
||||||
|
|
||||||
df_scud['Первая_активность'] = first_activities
|
|
||||||
df_scud['anomaly_flag'] = anomaly_flags
|
|
||||||
df_scud['Пришел'] = is_present_flags
|
|
||||||
|
|
||||||
return df_scud
|
|
||||||
|
|
||||||
|
|
||||||
def load_scud_data(target_date_type="today", snapshot_param=None):
|
|
||||||
"""
|
|
||||||
Загружает логи СКУД за 'today' или 'yesterday' НАПРЯМУЮ из локальной базы SQLite.
|
|
||||||
"""
|
|
||||||
target_date = DATE_TODAY if target_date_type == "today" else DATE_YESTERDAY
|
|
||||||
|
|
||||||
df_scud = load_scud_from_db_by_snapshot(target_date, snapshot_param=snapshot_param)
|
|
||||||
|
|
||||||
if df_scud is None or df_scud.empty:
|
|
||||||
df_scud = load_scud_from_db_by_snapshot(DATE_TODAY, snapshot_param=snapshot_param)
|
|
||||||
if df_scud is None or df_scud.empty:
|
|
||||||
print(f"[ℹ️] В базе SQLite / файлах отсутствует СКУД за {target_date}. Возвращаем пустой набор.")
|
|
||||||
return pd.DataFrame()
|
|
||||||
|
|
||||||
exceptions = load_exceptions()
|
|
||||||
|
|
||||||
if 'fio_clean' not in df_scud.columns:
|
|
||||||
fio_raw_col = 'fio' if 'fio' in df_scud.columns else ('Сотрудник' if 'Сотрудник' in df_scud.columns else df_scud.columns[0])
|
|
||||||
df_scud['fio_raw'] = df_scud[fio_raw_col].astype(str)
|
|
||||||
df_scud['fio_clean'] = df_scud['fio_raw'].apply(clean_scud_fio_light)
|
|
||||||
|
|
||||||
if 'Начало_дня' not in df_scud.columns and 'time_in' in df_scud.columns:
|
|
||||||
df_scud['Начало_дня'] = df_scud['time_in']
|
|
||||||
if 'Конец_дня' not in df_scud.columns and 'time_out' in df_scud.columns:
|
|
||||||
df_scud['Конец_дня'] = df_scud['time_out']
|
|
||||||
if 'Первая_активность' not in df_scud.columns and 'first_activity' in df_scud.columns:
|
|
||||||
df_scud['Первая_активность'] = df_scud['first_activity']
|
|
||||||
if 'Пришел' not in df_scud.columns and 'is_present' in df_scud.columns:
|
|
||||||
df_scud['Пришел'] = df_scud['is_present'].astype(bool)
|
|
||||||
|
|
||||||
df_scud['is_excluded'] = df_scud.apply(
|
|
||||||
lambda r: is_excluded(r.get('fio', r.get('Сотрудник', '')), r.get('department', r.get('Подразделение', '')), r.get('position', r.get('Должность', '')), exceptions),
|
|
||||||
axis=1
|
|
||||||
)
|
|
||||||
return df_scud
|
|
||||||
|
|
||||||
|
|
||||||
def load_all_data():
|
|
||||||
"""
|
|
||||||
Загружает слитные датасеты за Сегодня и за Вчера напрямую из базы SQLite и 1С.
|
|
||||||
"""
|
|
||||||
df_scud_today = load_scud_data(target_date_type="today")
|
|
||||||
try:
|
|
||||||
df_scud_yesterday = load_scud_data(target_date_type="yesterday")
|
|
||||||
except FileNotFoundError:
|
|
||||||
df_scud_yesterday = df_scud_today.copy()
|
|
||||||
|
|
||||||
df_staff_today, df_absent_today = load_1c_data_smart(DATE_TODAY)
|
|
||||||
df_staff_yesterday, df_absent_yesterday = load_1c_data_smart(DATE_YESTERDAY)
|
|
||||||
|
|
||||||
return (
|
|
||||||
df_scud_today,
|
|
||||||
df_scud_yesterday,
|
|
||||||
df_staff_today,
|
|
||||||
df_absent_today,
|
|
||||||
df_staff_yesterday,
|
|
||||||
df_absent_yesterday
|
|
||||||
)
|
|
||||||
+133
-85
@@ -3,52 +3,87 @@ import os
|
|||||||
import time
|
import time
|
||||||
import openpyxl
|
import openpyxl
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
from datetime import datetime
|
from datetime import datetime, timedelta
|
||||||
from openpyxl import Workbook
|
from openpyxl import Workbook
|
||||||
from openpyxl.styles import Font, Alignment, PatternFill, Border, Side
|
from openpyxl.styles import Font, Alignment, PatternFill, Border, Side
|
||||||
from openpyxl.utils import get_column_letter
|
from openpyxl.utils import get_column_letter
|
||||||
from config import OUTPUT_DIR
|
from config import REPORTS_DIR
|
||||||
|
|
||||||
# Словарь месяцев для текстового формата в названиях файлов
|
MONTHS_RU_GENITIVE = {
|
||||||
MONTHS_RU = {
|
|
||||||
1: "января", 2: "февраля", 3: "марта", 4: "апреля",
|
1: "января", 2: "февраля", 3: "марта", 4: "апреля",
|
||||||
5: "мая", 6: "июня", 7: "июля", 8: "августа",
|
5: "мая", 6: "июня", 7: "июля", 8: "августа",
|
||||||
9: "сентября", 10: "октября", 11: "ноября", 12: "декабря"
|
9: "сентября", 10: "октября", 11: "ноября", 12: "декабря"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
MONTHS_RU_NOMINATIVE = {
|
||||||
|
1: "январь", 2: "февраль", 3: "март", 4: "апрель",
|
||||||
|
5: "май", 6: "июнь", 7: "июль", 8: "август",
|
||||||
|
9: "сентябрь", 10: "октябрь", 11: "ноябрь", 12: "декабрь"
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
def format_date_ru(date_str):
|
def format_date_ru(date_str):
|
||||||
"""Преобразует дату формата '27.07.2026' в '27 июля 2026'"""
|
date_clean = str(date_str).replace('_', '.')
|
||||||
try:
|
try:
|
||||||
dt = datetime.strptime(date_str, "%d.%m.%Y")
|
dt = datetime.strptime(date_clean, "%d.%m.%Y")
|
||||||
return f"{dt.day} {MONTHS_RU[dt.month]} {dt.year}"
|
return f"{dt.day} {MONTHS_RU_GENITIVE[dt.month]} {dt.year}"
|
||||||
except Exception:
|
except Exception:
|
||||||
return date_str
|
return date_str
|
||||||
|
|
||||||
|
|
||||||
# Границы ячеек
|
def get_dated_reports_dir(date_str):
|
||||||
|
date_clean = str(date_str).replace('_', '.')
|
||||||
|
try:
|
||||||
|
dt = datetime.strptime(date_clean, "%d.%m.%Y")
|
||||||
|
year_str = str(dt.year)
|
||||||
|
month_name = MONTHS_RU_NOMINATIVE[dt.month]
|
||||||
|
except Exception:
|
||||||
|
now = datetime.now()
|
||||||
|
year_str = str(now.year)
|
||||||
|
month_name = MONTHS_RU_NOMINATIVE[now.month]
|
||||||
|
|
||||||
|
target_dir = os.path.join(REPORTS_DIR, year_str, month_name)
|
||||||
|
os.makedirs(target_dir, exist_ok=True)
|
||||||
|
return target_dir
|
||||||
|
|
||||||
|
|
||||||
THIN_SIDE = Side(border_style="thin", color="D3D3D3")
|
THIN_SIDE = Side(border_style="thin", color="D3D3D3")
|
||||||
THIN_BORDER = Border(left=THIN_SIDE, right=THIN_SIDE, top=THIN_SIDE, bottom=THIN_SIDE)
|
THIN_BORDER = Border(left=THIN_SIDE, right=THIN_SIDE, top=THIN_SIDE, bottom=THIN_SIDE)
|
||||||
|
|
||||||
# Палитра заливки для Сводки
|
|
||||||
FILL_HEADER = PatternFill(start_color="D9E1F2", end_color="D9E1F2", fill_type="solid")
|
FILL_HEADER = PatternFill(start_color="D9E1F2", end_color="D9E1F2", fill_type="solid")
|
||||||
FILL_TOTAL_LIST = PatternFill(start_color="F2F2F2", end_color="F2F2F2", fill_type="solid")
|
FILL_TOTAL_LIST = PatternFill(start_color="F2F2F2", end_color="F2F2F2", fill_type="solid")
|
||||||
FILL_UNEXPLAINED = PatternFill(start_color="FCE4D6", end_color="FCE4D6", fill_type="solid")
|
FILL_UNEXPLAINED = PatternFill(start_color="FCE4D6", end_color="FCE4D6", fill_type="solid")
|
||||||
FILL_PRESENT = PatternFill(start_color="E2EFDA", end_color="E2EFDA", fill_type="solid")
|
FILL_PRESENT = PatternFill(start_color="E2EFDA", end_color="E2EFDA", fill_type="solid")
|
||||||
|
FILL_REMOTE = PatternFill(start_color="E8F8F5", end_color="E8F8F5", fill_type="solid")
|
||||||
FILL_ANOMALY = PatternFill(start_color="FCE4D6", end_color="FCE4D6", fill_type="solid")
|
FILL_ANOMALY = PatternFill(start_color="FCE4D6", end_color="FCE4D6", fill_type="solid")
|
||||||
FILL_NO_PASS = PatternFill(start_color="E1F5FE", end_color="E1F5FE", fill_type="solid") # Мягкий пастельно-голубой для "Нет пропуска"
|
FILL_NO_PASS = PatternFill(start_color="E1F5FE", end_color="E1F5FE", fill_type="solid")
|
||||||
|
FILL_EXCEPTIONS = PatternFill(start_color="EFEBE9", end_color="EFEBE9", fill_type="solid")
|
||||||
|
|
||||||
CATEGORY_PASTEL_COLORS = ["FFF2CC", "E1D5E7", "E1F5FE", "FFF0F5", "E8F8F5", "FCF3CF"]
|
CATEGORY_PASTEL_COLORS = ["FFF2CC", "E1D5E7", "E1F5FE", "FFF0F5", "FCF3CF"]
|
||||||
|
YELLOW_FILL = PatternFill(start_color="FFF2CC", end_color="FFF2CC", fill_type="solid")
|
||||||
|
LIGHT_RED_FILL = PatternFill(start_color="FCE4D6", end_color="FCE4D6", fill_type="solid")
|
||||||
|
GREEN_FILL = PatternFill(start_color="E2EFDA", end_color="E2EFDA", fill_type="solid")
|
||||||
|
|
||||||
# Палитра для Детального отчета
|
|
||||||
YELLOW_FILL = PatternFill(start_color="FFF2CC", end_color="FFF2CC", fill_type="solid") # Обычные отсутствия
|
def calculate_autoclose_time(time_in_str: str) -> tuple[str, str, str]:
|
||||||
LIGHT_RED_FILL = PatternFill(start_color="FCE4D6", end_color="FCE4D6", fill_type="solid") # Потенциальные прогулы
|
"""
|
||||||
GREEN_FILL = PatternFill(start_color="E2EFDA", end_color="E2EFDA", fill_type="solid") # Аномалии (пришел в отпуске)
|
⭐️ Правило 8.5ч: при наличии входа и отсутствии выхода
|
||||||
LIGHT_BLUE_FILL = PatternFill(start_color="E1F5FE", end_color="E1F5FE", fill_type="solid") # Бледно-голубой для "Нет пропуска"
|
рассчитывает время выхода (Вход + 8ч 30мин) с нормой 8:00 и отклонением 0:00.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
parts = time_in_str.strip().split(':')
|
||||||
|
hh = int(parts[0])
|
||||||
|
mm = int(parts[1]) if len(parts) > 1 else 0
|
||||||
|
ss = int(parts[2]) if len(parts) > 2 else 0
|
||||||
|
|
||||||
|
dt_in = datetime(2000, 1, 1, hh, mm, ss)
|
||||||
|
dt_out = dt_in + timedelta(hours=8, minutes=30)
|
||||||
|
return dt_out.strftime("%H:%M:%S"), "08:30", "0:00"
|
||||||
|
except Exception:
|
||||||
|
return "17:00:00", "08:30", "0:00"
|
||||||
|
|
||||||
|
|
||||||
def calculate_deviation(time_in_building_str, reason="", norm_hours=8, lunch_minutes=30):
|
def calculate_deviation(time_in_building_str, reason="", norm_hours=8, lunch_minutes=30):
|
||||||
"""Рассчитывает точное отклонение от нормы с учетом обеда 30 мин и уважительных причин"""
|
|
||||||
if pd.notna(reason) and isinstance(reason, str) and reason.strip() != "":
|
if pd.notna(reason) and isinstance(reason, str) and reason.strip() != "":
|
||||||
return "0:00"
|
return "0:00"
|
||||||
|
|
||||||
@@ -101,11 +136,13 @@ def format_row_cells(ws, r_num, fill_obj, is_bold=False, align_b="right", bold_f
|
|||||||
|
|
||||||
|
|
||||||
# --- 1. СВОДКА НА СЕГОДНЯ ---
|
# --- 1. СВОДКА НА СЕГОДНЯ ---
|
||||||
def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
def generate_summary_excel(merged_df, date_str="21.08.2026", filename=None):
|
||||||
|
date_clean = str(date_str).replace('_', '.')
|
||||||
if not filename:
|
if not filename:
|
||||||
filename = f"{format_date_ru(date_str)} сводка.xlsx"
|
filename = f"{format_date_ru(date_clean)} сводка.xlsx"
|
||||||
|
|
||||||
output_path = os.path.join(OUTPUT_DIR, filename)
|
target_dir = get_dated_reports_dir(date_clean)
|
||||||
|
output_path = os.path.join(target_dir, filename)
|
||||||
wb = Workbook()
|
wb = Workbook()
|
||||||
ws = wb.active
|
ws = wb.active
|
||||||
ws.title = "Лист_1"
|
ws.title = "Лист_1"
|
||||||
@@ -117,26 +154,22 @@ def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
|||||||
|
|
||||||
bold_font = Font(name="Calibri", size=11, bold=True)
|
bold_font = Font(name="Calibri", size=11, bold=True)
|
||||||
|
|
||||||
# 1. Шапка
|
|
||||||
ws.cell(row=1, column=1, value="Сводка на")
|
ws.cell(row=1, column=1, value="Сводка на")
|
||||||
ws.cell(row=1, column=2, value=date_str)
|
ws.cell(row=1, column=2, value=date_clean)
|
||||||
format_row_cells(ws, 1, FILL_HEADER, is_bold=True, bold_font=bold_font)
|
format_row_cells(ws, 1, FILL_HEADER, is_bold=True, bold_font=bold_font)
|
||||||
|
|
||||||
apply_borders_to_cell(ws.cell(row=2, column=1))
|
apply_borders_to_cell(ws.cell(row=2, column=1))
|
||||||
apply_borders_to_cell(ws.cell(row=2, column=2))
|
apply_borders_to_cell(ws.cell(row=2, column=2))
|
||||||
|
|
||||||
# 2. По списку
|
|
||||||
ws.cell(row=3, column=1, value="По списку")
|
ws.cell(row=3, column=1, value="По списку")
|
||||||
ws.cell(row=3, column=2, value=len(merged_df))
|
ws.cell(row=3, column=2, value=len(merged_df))
|
||||||
format_row_cells(ws, 3, FILL_TOTAL_LIST, is_bold=True, bold_font=bold_font)
|
format_row_cells(ws, 3, FILL_TOTAL_LIST, is_bold=True, bold_font=bold_font)
|
||||||
|
|
||||||
current_row = 4
|
current_row = 4
|
||||||
|
|
||||||
# Подготовка флагов для точной фильтрации
|
|
||||||
is_no_pass = merged_df['no_scud_pass'] == True if 'no_scud_pass' in merged_df.columns else False
|
is_no_pass = merged_df['no_scud_pass'] == True if 'no_scud_pass' in merged_df.columns else False
|
||||||
is_exc = merged_df.get('is_excluded', False) == True
|
is_exc = merged_df.get('is_excluded', False) == True
|
||||||
|
|
||||||
# 3. НЕИЗВЕСТНО (Исключаем категорию "Нет пропуска" и "Исключения ОВК/Охрана")
|
|
||||||
unexplained = merged_df[
|
unexplained = merged_df[
|
||||||
(merged_df['Пришел'] == False) &
|
(merged_df['Пришел'] == False) &
|
||||||
(merged_df['Вид_отсутствия'].isna() | (merged_df['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
(merged_df['Вид_отсутствия'].isna() | (merged_df['Вид_отсутствия'].astype(str).str.strip() == '')) &
|
||||||
@@ -155,9 +188,7 @@ def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
|||||||
ws.row_dimensions[current_row].hidden = False
|
ws.row_dimensions[current_row].hidden = False
|
||||||
current_row += 1
|
current_row += 1
|
||||||
|
|
||||||
# 4. РАЗДЕЛ: НЕТ ПРОПУСКА (Строго один независимый блок)
|
|
||||||
no_pass_df = merged_df[is_no_pass] if 'no_scud_pass' in merged_df.columns else pd.DataFrame()
|
no_pass_df = merged_df[is_no_pass] if 'no_scud_pass' in merged_df.columns else pd.DataFrame()
|
||||||
|
|
||||||
ws.cell(row=current_row, column=1, value="Нет пропуска")
|
ws.cell(row=current_row, column=1, value="Нет пропуска")
|
||||||
ws.cell(row=current_row, column=2, value=len(no_pass_df))
|
ws.cell(row=current_row, column=2, value=len(no_pass_df))
|
||||||
format_row_cells(ws, current_row, FILL_NO_PASS, is_bold=True, bold_font=bold_font)
|
format_row_cells(ws, current_row, FILL_NO_PASS, is_bold=True, bold_font=bold_font)
|
||||||
@@ -171,11 +202,15 @@ def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
|||||||
ws.row_dimensions[current_row].hidden = False
|
ws.row_dimensions[current_row].hidden = False
|
||||||
current_row += 1
|
current_row += 1
|
||||||
|
|
||||||
# 5. КАТЕГОРИИ ОТСУТСТВИЙ (Свернуты hidden=True)
|
reason_clean = merged_df['Вид_отсутствия'].astype(str).str.lower()
|
||||||
|
is_remote_reason = reason_clean.str.contains('удален|дистанцион', regex=True, na=False)
|
||||||
|
|
||||||
absent_only = merged_df[
|
absent_only = merged_df[
|
||||||
(merged_df['Пришел'] == False) &
|
(merged_df['Пришел'] == False) &
|
||||||
(merged_df['Вид_отсутствия'].notna()) &
|
(merged_df['Вид_отсутствия'].notna()) &
|
||||||
(~merged_df['Вид_отсутствия'].astype(str).str.startswith('Исключение'))
|
(~merged_df['Вид_отсутствия'].astype(str).str.startswith('Исключение')) &
|
||||||
|
(~is_remote_reason) &
|
||||||
|
(~is_exc)
|
||||||
]
|
]
|
||||||
absent_groups = absent_only.groupby('Вид_отсутствия')
|
absent_groups = absent_only.groupby('Вид_отсутствия')
|
||||||
|
|
||||||
@@ -195,12 +230,7 @@ def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
|||||||
ws.row_dimensions[current_row].hidden = True
|
ws.row_dimensions[current_row].hidden = True
|
||||||
current_row += 1
|
current_row += 1
|
||||||
|
|
||||||
# 6. ИТОГО НА РАБОТЕ (Свернут hidden=True)
|
present = merged_df[(merged_df['Пришел'] == True) & (~is_exc)]
|
||||||
is_working_mask = (merged_df['Пришел'] == True) | (
|
|
||||||
merged_df['Вид_отсутствия'].astype(str).str.lower().str.contains('командировк|удален|дистанцион|разъездн', regex=True, na=False)
|
|
||||||
)
|
|
||||||
present = merged_df[is_working_mask]
|
|
||||||
|
|
||||||
ws.cell(row=current_row, column=1, value="Итого на работе")
|
ws.cell(row=current_row, column=1, value="Итого на работе")
|
||||||
ws.cell(row=current_row, column=2, value=len(present))
|
ws.cell(row=current_row, column=2, value=len(present))
|
||||||
format_row_cells(ws, current_row, FILL_PRESENT, is_bold=True, bold_font=bold_font)
|
format_row_cells(ws, current_row, FILL_PRESENT, is_bold=True, bold_font=bold_font)
|
||||||
@@ -214,10 +244,26 @@ def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
|||||||
ws.row_dimensions[current_row].hidden = True
|
ws.row_dimensions[current_row].hidden = True
|
||||||
current_row += 1
|
current_row += 1
|
||||||
|
|
||||||
# 7. АНОМАЛИИ СКУД И 1С (ОВК и Подрядчики из исключений СУДА НЕ ПОПАДАЮТ)
|
remote_home = merged_df[(merged_df['Пришел'] == False) & is_remote_reason & (~is_exc)]
|
||||||
|
ws.cell(row=current_row, column=1, value="В том числе на удаленной работе")
|
||||||
|
ws.cell(row=current_row, column=2, value=len(remote_home))
|
||||||
|
format_row_cells(ws, current_row, FILL_REMOTE, is_bold=True, bold_font=bold_font)
|
||||||
|
current_row += 1
|
||||||
|
|
||||||
|
if not remote_home.empty:
|
||||||
|
for fio in sorted(remote_home['Сотрудник'].dropna().unique()):
|
||||||
|
ws.cell(row=current_row, column=1, value=fio)
|
||||||
|
format_row_cells(ws, current_row, FILL_REMOTE, is_bold=False)
|
||||||
|
ws.row_dimensions[current_row].outlineLevel = 1
|
||||||
|
ws.row_dimensions[current_row].hidden = True
|
||||||
|
current_row += 1
|
||||||
|
|
||||||
anomalies = merged_df[
|
anomalies = merged_df[
|
||||||
(~is_exc) & (
|
(~is_exc) & (
|
||||||
((merged_df['Пришел'] == True) & (merged_df['Вид_отсутствия'].notna()) & (~merged_df['Вид_отсутствия'].astype(str).str.startswith('Исключение'))) |
|
((merged_df['Пришел'] == True) & (merged_df['Вид_отсутствия'].notna()) &
|
||||||
|
(~merged_df['Вид_отсутствия'].astype(str).str.startswith('Исключение')) &
|
||||||
|
(~is_remote_reason) &
|
||||||
|
(~reason_clean.str.contains('командировк', na=False))) |
|
||||||
(merged_df.get('anomaly_flag', 'NONE') == 'ANOMALY_NO_IN_HAS_ACTIVITY')
|
(merged_df.get('anomaly_flag', 'NONE') == 'ANOMALY_NO_IN_HAS_ACTIVITY')
|
||||||
)
|
)
|
||||||
]
|
]
|
||||||
@@ -228,7 +274,6 @@ def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
|||||||
current_row += 1
|
current_row += 1
|
||||||
|
|
||||||
chars_per_line_b = 30
|
chars_per_line_b = 30
|
||||||
|
|
||||||
if not anomalies.empty:
|
if not anomalies.empty:
|
||||||
for _, row in anomalies.iterrows():
|
for _, row in anomalies.iterrows():
|
||||||
fio = row.get('Сотрудник', '')
|
fio = row.get('Сотрудник', '')
|
||||||
@@ -243,21 +288,29 @@ def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
|||||||
|
|
||||||
cell_a = ws.cell(row=current_row, column=1, value=f"{fio}")
|
cell_a = ws.cell(row=current_row, column=1, value=f"{fio}")
|
||||||
cell_b = ws.cell(row=current_row, column=2, value=reason_text)
|
cell_b = ws.cell(row=current_row, column=2, value=reason_text)
|
||||||
|
|
||||||
cell_a.fill = FILL_ANOMALY
|
cell_a.fill = FILL_ANOMALY
|
||||||
cell_b.fill = FILL_ANOMALY
|
cell_b.fill = FILL_ANOMALY
|
||||||
apply_borders_to_cell(cell_a)
|
apply_borders_to_cell(cell_a)
|
||||||
apply_borders_to_cell(cell_b)
|
apply_borders_to_cell(cell_b)
|
||||||
|
|
||||||
cell_b.alignment = Alignment(horizontal="left", vertical="center", wrap_text=True)
|
cell_b.alignment = Alignment(horizontal="left", vertical="center", wrap_text=True)
|
||||||
cell_a.alignment = Alignment(horizontal="left", vertical="center")
|
cell_a.alignment = Alignment(horizontal="left", vertical="center")
|
||||||
|
|
||||||
if len(reason_text) > chars_per_line_b:
|
lines_count = math.ceil(len(reason_text) / chars_per_line_b) if len(reason_text) > chars_per_line_b else 1
|
||||||
lines_count = math.ceil(len(reason_text) / chars_per_line_b)
|
ws.row_dimensions[current_row].height = max(lines_count * 18, 20)
|
||||||
ws.row_dimensions[current_row].height = max(lines_count * 18, 22)
|
ws.row_dimensions[current_row].outlineLevel = 1
|
||||||
else:
|
ws.row_dimensions[current_row].hidden = True
|
||||||
ws.row_dimensions[current_row].height = 20
|
current_row += 1
|
||||||
|
|
||||||
|
exceptions_df = merged_df[is_exc]
|
||||||
|
ws.cell(row=current_row, column=1, value="Исключения")
|
||||||
|
ws.cell(row=current_row, column=2, value=len(exceptions_df))
|
||||||
|
format_row_cells(ws, current_row, FILL_EXCEPTIONS, is_bold=True, bold_font=bold_font)
|
||||||
|
current_row += 1
|
||||||
|
|
||||||
|
if not exceptions_df.empty:
|
||||||
|
for fio in sorted(exceptions_df['Сотрудник'].dropna().unique()):
|
||||||
|
ws.cell(row=current_row, column=1, value=fio)
|
||||||
|
format_row_cells(ws, current_row, FILL_EXCEPTIONS, is_bold=False)
|
||||||
ws.row_dimensions[current_row].outlineLevel = 1
|
ws.row_dimensions[current_row].outlineLevel = 1
|
||||||
ws.row_dimensions[current_row].hidden = True
|
ws.row_dimensions[current_row].hidden = True
|
||||||
current_row += 1
|
current_row += 1
|
||||||
@@ -268,25 +321,32 @@ def generate_summary_excel(merged_df, date_str="27.07.2026", filename=None):
|
|||||||
try:
|
try:
|
||||||
wb.save(output_path)
|
wb.save(output_path)
|
||||||
print(f"[✓] Ежедневная сводка сохранена: {output_path}")
|
print(f"[✓] Ежедневная сводка сохранена: {output_path}")
|
||||||
except PermissionError:
|
except (PermissionError, OSError):
|
||||||
alt_filename = filename.replace(".xlsx", f"_{int(time.time())}.xlsx")
|
alt_filename = filename.replace(".xlsx", f"_{int(time.time())}.xlsx")
|
||||||
alt_path = os.path.join(OUTPUT_DIR, alt_filename)
|
alt_path = os.path.join(target_dir, alt_filename)
|
||||||
wb.save(alt_path)
|
wb.save(alt_path)
|
||||||
print(f"[⚠️] Файл открыт в Excel! Сохранено как: {alt_path}")
|
print(f"[⚠️] Файл открыт в Excel! Сохранено как: {alt_path}")
|
||||||
|
|
||||||
|
|
||||||
# --- 2. ДЕТАЛЬНЫЙ ОТЧЕТ ЗА ВЧЕРА ---
|
# --- 2. ДЕТАЛЬНЫЙ ОТЧЕТ ЗА ВЧЕРА (С ПРАВИЛОМ 8.5ч) ---
|
||||||
def generate_detailed_excel(merged_df, date_str="26.07.2026", filename=None):
|
def generate_detailed_excel(merged_df, date_str="20.08.2026", filename=None):
|
||||||
|
date_clean = str(date_str).replace('_', '.')
|
||||||
if not filename:
|
if not filename:
|
||||||
filename = f"{format_date_ru(date_str)} отчет.xlsx"
|
filename = f"{format_date_ru(date_clean)} отчет.xlsx"
|
||||||
|
|
||||||
output_path = os.path.join(OUTPUT_DIR, filename)
|
if merged_df is not None and not merged_df.empty:
|
||||||
|
df_export = merged_df[merged_df.get('is_excluded', False) == False].copy()
|
||||||
|
else:
|
||||||
|
df_export = pd.DataFrame()
|
||||||
|
|
||||||
|
target_dir = get_dated_reports_dir(date_clean)
|
||||||
|
output_path = os.path.join(target_dir, filename)
|
||||||
wb = Workbook()
|
wb = Workbook()
|
||||||
ws = wb.active
|
ws = wb.active
|
||||||
ws.title = "Детальный_отчет"
|
ws.title = "Детальный_отчет"
|
||||||
|
|
||||||
ws["B2"] = "Дата:"
|
ws["B2"] = "Дата:"
|
||||||
ws["D2"] = date_str
|
ws["D2"] = date_clean
|
||||||
ws["B2"].font = Font(name="Arial", size=10, bold=True)
|
ws["B2"].font = Font(name="Arial", size=10, bold=True)
|
||||||
ws["D2"].font = Font(name="Arial", size=10, bold=True)
|
ws["D2"].font = Font(name="Arial", size=10, bold=True)
|
||||||
|
|
||||||
@@ -305,32 +365,38 @@ def generate_detailed_excel(merged_df, date_str="26.07.2026", filename=None):
|
|||||||
cell.alignment = Alignment(horizontal="center", vertical="center", wrap_text=True)
|
cell.alignment = Alignment(horizontal="center", vertical="center", wrap_text=True)
|
||||||
apply_borders_to_cell(cell)
|
apply_borders_to_cell(cell)
|
||||||
|
|
||||||
start_col = 'Начало дня' if 'Начало дня' in merged_df.columns else 'Начало_дня'
|
start_col = 'Начало дня' if 'Начало дня' in df_export.columns else 'Начало_дня'
|
||||||
end_col = 'Конец дня' if 'Конец дня' in merged_df.columns else 'Конец_дня'
|
end_col = 'Конец дня' if 'Конец дня' in df_export.columns else 'Конец_дня'
|
||||||
hours_col = 'Часы' if 'Часы' in merged_df.columns else 'Находился_в_здании'
|
hours_col = 'Часы' if 'Часы' in df_export.columns else 'Находился_в_здании'
|
||||||
|
|
||||||
chars_per_line_h = 24
|
chars_per_line_h = 24
|
||||||
|
|
||||||
for idx, row in merged_df.reset_index(drop=True).iterrows():
|
for idx, row in df_export.reset_index(drop=True).iterrows():
|
||||||
is_present = row.get('Пришел', False)
|
is_present = row.get('Пришел', False)
|
||||||
absence_reason = row.get('Вид_отсутствия', '')
|
absence_reason = row.get('Вид_отсутствия', '')
|
||||||
has_reason = pd.notna(absence_reason) and str(absence_reason).strip() != ''
|
has_reason = pd.notna(absence_reason) and str(absence_reason).strip() != ''
|
||||||
anom_flag = row.get('anomaly_flag', 'NONE')
|
|
||||||
|
|
||||||
|
in_val = str(row.get(start_col, 'Нет входа')).strip()
|
||||||
|
out_val = str(row.get(end_col, 'Нет выхода')).strip()
|
||||||
in_building_str = str(row.get(hours_col, '00:00'))
|
in_building_str = str(row.get(hours_col, '00:00'))
|
||||||
first_act_val = str(row.get('Первая_активность', '—')).strip()
|
first_act_val = str(row.get('Первая_активность', '—')).strip()
|
||||||
has_first_act = first_act_val not in ['—', '', 'None', 'nan']
|
has_first_act = first_act_val not in ['—', '', 'None', 'nan']
|
||||||
|
|
||||||
|
# ⭐️ ПРАВИЛО 8.5ч: если есть вход, но нет выхода (и нет официального документа отсутствия)
|
||||||
|
if in_val not in ['Нет входа', '—', '', 'nan', 'None'] and out_val in ['Нет выхода', '—', '', 'nan', 'None'] and not has_reason:
|
||||||
|
out_val, in_building_str, deviation_val = calculate_autoclose_time(in_val)
|
||||||
|
else:
|
||||||
|
deviation_val = calculate_deviation(in_building_str, reason=absence_reason if has_reason else "", norm_hours=8, lunch_minutes=30)
|
||||||
|
|
||||||
dept_scud_val = row.get('department_scud', row.get('department', row.get('Подразделение', '')))
|
dept_scud_val = row.get('department_scud', row.get('department', row.get('Подразделение', '')))
|
||||||
deviation_val = calculate_deviation(in_building_str, reason=absence_reason if has_reason else "", norm_hours=8, lunch_minutes=30)
|
|
||||||
|
|
||||||
ws.append([
|
ws.append([
|
||||||
idx + 1,
|
idx + 1,
|
||||||
row.get('Сотрудник', ''),
|
row.get('Сотрудник', ''),
|
||||||
dept_scud_val,
|
dept_scud_val,
|
||||||
row.get(start_col, 'Нет входа'),
|
in_val,
|
||||||
first_act_val,
|
first_act_val,
|
||||||
row.get(end_col, 'Нет выхода'),
|
out_val,
|
||||||
in_building_str,
|
in_building_str,
|
||||||
absence_reason if has_reason else '',
|
absence_reason if has_reason else '',
|
||||||
8,
|
8,
|
||||||
@@ -338,31 +404,24 @@ def generate_detailed_excel(merged_df, date_str="26.07.2026", filename=None):
|
|||||||
])
|
])
|
||||||
|
|
||||||
row_num = 5 + idx
|
row_num = 5 + idx
|
||||||
|
|
||||||
# ЗАЛИВКА СТРОК:
|
|
||||||
if is_present and has_reason:
|
if is_present and has_reason:
|
||||||
row_fill = GREEN_FILL
|
row_fill = GREEN_FILL
|
||||||
elif not is_present and has_reason:
|
elif not is_present and has_reason:
|
||||||
row_fill = YELLOW_FILL
|
row_fill = YELLOW_FILL
|
||||||
elif not is_present and not has_reason and not has_first_act:
|
elif not is_present and not has_reason and not has_first_act:
|
||||||
# Розово-красный подсвечивает исключительно неизвестные случаи (потенциальные прогулы)
|
|
||||||
row_fill = LIGHT_RED_FILL
|
row_fill = LIGHT_RED_FILL
|
||||||
else:
|
else:
|
||||||
row_fill = None
|
row_fill = None
|
||||||
|
|
||||||
val_h_str = str(absence_reason) if has_reason else ""
|
val_h_str = str(absence_reason) if has_reason else ""
|
||||||
if len(val_h_str) > chars_per_line_h:
|
lines_count = math.ceil(len(val_h_str) / chars_per_line_h) if len(val_h_str) > chars_per_line_h else 1
|
||||||
needed_lines = math.ceil(len(val_h_str) / chars_per_line_h)
|
ws.row_dimensions[row_num].height = max(lines_count * 18, 20)
|
||||||
ws.row_dimensions[row_num].height = max(needed_lines * 18, 22)
|
|
||||||
else:
|
|
||||||
ws.row_dimensions[row_num].height = 20
|
|
||||||
|
|
||||||
for col_idx in range(1, len(headers) + 1):
|
for col_idx in range(1, len(headers) + 1):
|
||||||
cell = ws.cell(row=row_num, column=col_idx)
|
cell = ws.cell(row=row_num, column=col_idx)
|
||||||
apply_borders_to_cell(cell)
|
apply_borders_to_cell(cell)
|
||||||
if row_fill:
|
if row_fill:
|
||||||
cell.fill = row_fill
|
cell.fill = row_fill
|
||||||
|
|
||||||
if col_idx == 8:
|
if col_idx == 8:
|
||||||
cell.alignment = Alignment(horizontal="left", vertical="center", wrap_text=True)
|
cell.alignment = Alignment(horizontal="left", vertical="center", wrap_text=True)
|
||||||
elif col_idx in [1, 4, 5, 6, 7, 9, 10]:
|
elif col_idx in [1, 4, 5, 6, 7, 9, 10]:
|
||||||
@@ -370,32 +429,21 @@ def generate_detailed_excel(merged_df, date_str="26.07.2026", filename=None):
|
|||||||
else:
|
else:
|
||||||
cell.alignment = Alignment(horizontal="left", vertical="center")
|
cell.alignment = Alignment(horizontal="left", vertical="center")
|
||||||
|
|
||||||
# Динамический компактный автоподгон ширины колонок
|
|
||||||
for col in ws.columns:
|
for col in ws.columns:
|
||||||
col_letter = get_column_letter(col[0].column)
|
col_letter = get_column_letter(col[0].column)
|
||||||
max_len = 0
|
max_len = max((len(str(cell.value or '')) for cell in col), default=8)
|
||||||
for cell in col:
|
ws.column_dimensions[col_letter].width = min(max(max_len + 2, 8), 35)
|
||||||
if cell.value is not None:
|
|
||||||
cell_lines = str(cell.value).split("\n")
|
|
||||||
line_max = max(len(line) for line in cell_lines)
|
|
||||||
if line_max > max_len:
|
|
||||||
max_len = line_max
|
|
||||||
|
|
||||||
optimal_width = max(max_len + 2, 8)
|
|
||||||
if optimal_width > 35:
|
|
||||||
optimal_width = 35
|
|
||||||
ws.column_dimensions[col_letter].width = optimal_width
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
wb.save(output_path)
|
wb.save(output_path)
|
||||||
print(f"[✓] Детальный отчет сохранен: {output_path}")
|
print(f"[✓] Детальный отчет сохранен: {output_path}")
|
||||||
except PermissionError:
|
except (PermissionError, OSError):
|
||||||
alt_filename = filename.replace(".xlsx", f"_{int(time.time())}.xlsx")
|
alt_filename = filename.replace(".xlsx", f"_{int(time.time())}.xlsx")
|
||||||
alt_path = os.path.join(OUTPUT_DIR, alt_filename)
|
alt_path = os.path.join(target_dir, alt_filename)
|
||||||
wb.save(alt_path)
|
wb.save(alt_path)
|
||||||
print(f"[⚠️] Файл открыт в Excel! Сохранено как: {alt_path}")
|
print(f"[⚠️] Файл открыт в Excel! Сохранено как: {alt_path}")
|
||||||
|
|
||||||
|
|
||||||
def export_raw_scud(df_scud, filename="СКУД_Сырые_данные.xlsx"):
|
def export_raw_scud(df_scud, filename="СКУД_Сырые_данные.xlsx"):
|
||||||
output_path = os.path.join(OUTPUT_DIR, filename)
|
output_path = os.path.join(REPORTS_DIR, filename)
|
||||||
df_scud.to_excel(output_path, index=False)
|
df_scud.to_excel(output_path, index=False)
|
||||||
@@ -0,0 +1,84 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/exceptions_repo.py
|
||||||
|
ROLE: Управление исключениями в SQLite с синхронизацией с exceptions.json.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Dict, List, Any
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
from config import EXCEPTIONS_PATH, normalize_fio
|
||||||
|
from core.connection import get_connection
|
||||||
|
|
||||||
|
|
||||||
|
def init_exceptions_table():
|
||||||
|
with get_connection() as conn:
|
||||||
|
conn.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS exceptions_registry (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
category TEXT NOT NULL,
|
||||||
|
value TEXT NOT NULL,
|
||||||
|
comment TEXT DEFAULT '',
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
UNIQUE(category, value)
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def get_all_exceptions_from_db() -> Dict[str, List[str]]:
|
||||||
|
init_exceptions_table()
|
||||||
|
cfg = {"departments": [], "positions": [], "fio": [], "position_keywords": [], "include_fio": []}
|
||||||
|
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT category, value FROM exceptions_registry")
|
||||||
|
rows = cursor.fetchall()
|
||||||
|
|
||||||
|
if not rows and os.path.exists(EXCEPTIONS_PATH):
|
||||||
|
# Первичная миграция из JSON в SQLite
|
||||||
|
sync_json_to_db()
|
||||||
|
return get_all_exceptions_from_db()
|
||||||
|
|
||||||
|
for cat, val in rows:
|
||||||
|
if cat in cfg:
|
||||||
|
cfg[cat].append(val)
|
||||||
|
return cfg
|
||||||
|
|
||||||
|
|
||||||
|
def add_exception_to_db(category: str, value: str, comment: str = "") -> bool:
|
||||||
|
init_exceptions_table()
|
||||||
|
val_clean = normalize_fio(value) if category in ["fio", "include_fio"] else value.strip()
|
||||||
|
if not val_clean:
|
||||||
|
return False
|
||||||
|
with get_connection() as conn:
|
||||||
|
conn.execute(
|
||||||
|
"INSERT OR REPLACE INTO exceptions_registry (category, value, comment) VALUES (?, ?, ?)",
|
||||||
|
(category, val_clean, comment)
|
||||||
|
)
|
||||||
|
conn.commit()
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
def remove_exception_from_db(category: str, value: str) -> bool:
|
||||||
|
init_exceptions_table()
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM exceptions_registry WHERE category = ? AND value = ?", (category, value.strip()))
|
||||||
|
conn.commit()
|
||||||
|
return cursor.rowcount > 0
|
||||||
|
|
||||||
|
|
||||||
|
def sync_json_to_db():
|
||||||
|
"""Переносит данные из exceptions.json в SQLite."""
|
||||||
|
if not os.path.exists(EXCEPTIONS_PATH):
|
||||||
|
return
|
||||||
|
try:
|
||||||
|
with open(EXCEPTIONS_PATH, "r", encoding="utf-8") as f:
|
||||||
|
data = json.load(f)
|
||||||
|
for cat, items in data.items():
|
||||||
|
for item in items:
|
||||||
|
add_exception_to_db(cat, item, comment="Импорт из JSON")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[⚠️] Ошибка синхронизации JSON -> DB: {e}")
|
||||||
@@ -0,0 +1,37 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/knowledge/service.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: services / knowledge
|
||||||
|
ROLE: Доменный сервис базы знаний, правил компании и синонимов подразделений.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import List, Dict, Any
|
||||||
|
from core.repositories.zup_repo import (
|
||||||
|
get_all_rules_from_db,
|
||||||
|
add_rule_to_db,
|
||||||
|
get_department_synonyms_dict,
|
||||||
|
add_department_synonym_to_db
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def get_rules() -> List[Dict[str, Any]]:
|
||||||
|
"""Получить все правила базы знаний в виде списка словарей."""
|
||||||
|
raw_rules = get_all_rules_from_db()
|
||||||
|
return [{"id": idx, "rule_text": r} for idx, r in enumerate(raw_rules, 1)]
|
||||||
|
|
||||||
|
|
||||||
|
def add_rule(rule_text: str, added_by: str = "Human") -> None:
|
||||||
|
"""Добавить новое правило в базу знаний."""
|
||||||
|
add_rule_to_db(rule_text, added_by=added_by)
|
||||||
|
|
||||||
|
|
||||||
|
def get_synonyms() -> Dict[str, str]:
|
||||||
|
"""Получить словарь синонимов отделов."""
|
||||||
|
return get_department_synonyms_dict()
|
||||||
|
|
||||||
|
|
||||||
|
def register_department_synonym(short_name: str, full_name: str) -> None:
|
||||||
|
"""Сохранить новую пару синонимов подразделения."""
|
||||||
|
add_department_synonym_to_db(short_name, full_name)
|
||||||
@@ -0,0 +1,79 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/prompts/diff_engine.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: services / prompts
|
||||||
|
ROLE: Формирование превью изменений промпта и генерация визуального HTML-Diff.
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[PROMPT_DIFF_BUILDER]: Генерация черновика и HTML-разметки изменений.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Tuple, Dict, List, Set
|
||||||
|
from core.connection import get_connection
|
||||||
|
|
||||||
|
def build_prompt_diff(
|
||||||
|
action: str,
|
||||||
|
section_id: int = None,
|
||||||
|
item_id: int = None,
|
||||||
|
content: str = "",
|
||||||
|
delete_nodes: List[Tuple[int, int]] = None,
|
||||||
|
prompt_name: str = "main_agent"
|
||||||
|
) -> Tuple[str, str]:
|
||||||
|
act = (action or "ADD").upper()
|
||||||
|
nodes_to_delete: Set[Tuple[int, int]] = set()
|
||||||
|
|
||||||
|
if delete_nodes:
|
||||||
|
nodes_to_delete.update(delete_nodes)
|
||||||
|
elif act == "DELETE" and section_id is not None and item_id is not None:
|
||||||
|
nodes_to_delete.add((section_id, item_id))
|
||||||
|
|
||||||
|
with get_connection(row_factory=True) as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT section_id, item_id, content
|
||||||
|
FROM system_prompt_nodes
|
||||||
|
WHERE prompt_name = ? AND is_active = 1
|
||||||
|
ORDER BY section_id, item_id
|
||||||
|
""", (prompt_name,))
|
||||||
|
existing_nodes = cursor.fetchall()
|
||||||
|
|
||||||
|
nodes_dict = {(r["section_id"], r["item_id"]): r["content"] for r in existing_nodes}
|
||||||
|
nodes_dict_for_draft = {k: v for k, v in nodes_dict.items() if k not in nodes_to_delete}
|
||||||
|
|
||||||
|
if act not in ["DELETE", "BATCH_DELETE"] and section_id is not None and item_id is not None:
|
||||||
|
nodes_dict_for_draft[(section_id, item_id)] = content
|
||||||
|
|
||||||
|
draft_lines = []
|
||||||
|
curr_sec = None
|
||||||
|
for (s_id, i_id), txt in sorted(nodes_dict_for_draft.items()):
|
||||||
|
if i_id == 0:
|
||||||
|
if curr_sec is not None: draft_lines.append("")
|
||||||
|
draft_lines.append(f"{s_id}. {txt}")
|
||||||
|
curr_sec = s_id
|
||||||
|
else:
|
||||||
|
draft_lines.append(f" {s_id}.{i_id}. {txt}")
|
||||||
|
merged_prompt = "\n".join(draft_lines)
|
||||||
|
|
||||||
|
diff_lines = []
|
||||||
|
curr_sec = None
|
||||||
|
display_nodes = dict(nodes_dict)
|
||||||
|
if act not in ["DELETE", "BATCH_DELETE"] and section_id is not None and item_id is not None:
|
||||||
|
display_nodes[(section_id, item_id)] = content
|
||||||
|
|
||||||
|
for (s_id, i_id), txt in sorted(display_nodes.items()):
|
||||||
|
if i_id == 0:
|
||||||
|
if curr_sec is not None: diff_lines.append("")
|
||||||
|
diff_lines.append(f"{s_id}. {txt}")
|
||||||
|
curr_sec = s_id
|
||||||
|
else:
|
||||||
|
if (s_id, i_id) in nodes_to_delete:
|
||||||
|
line_str = f' <span class="line-through text-rose-600 font-bold bg-rose-50 px-1.5 py-0.5 rounded border border-rose-300 opacity-80">{s_id}.{i_id}. {txt} [УДАЛЕНИЕ]</span>'
|
||||||
|
elif act == "ADD" and s_id == section_id and i_id == item_id:
|
||||||
|
line_str = f' <span class="text-rose-600 font-bold bg-rose-50 px-1.5 py-0.5 rounded border border-rose-300">{s_id}.{i_id}. {txt}</span>'
|
||||||
|
else:
|
||||||
|
line_str = f" {s_id}.{i_id}. {txt}"
|
||||||
|
diff_lines.append(line_str)
|
||||||
|
|
||||||
|
return merged_prompt, "\n".join(diff_lines)
|
||||||
@@ -0,0 +1,114 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/prompts/repository.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: services / prompts
|
||||||
|
ROLE: Реляционное хранилище узлов системного промпта (таблица system_prompt_nodes).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[PROMPT_REPO_GET_ACTIVE]: Сборка активного промпта из узлов БД.
|
||||||
|
- ANCHOR[PROMPT_REPO_APPLY_ACTION]: Точечная вставка / изменение / удаление узла.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import re
|
||||||
|
import logging
|
||||||
|
from typing import List, Tuple, Dict, Any
|
||||||
|
from core.connection import get_connection
|
||||||
|
|
||||||
|
logger = logging.getLogger("PROMPT_REPO")
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[PROMPT_REPO_GET_ACTIVE]
|
||||||
|
def repo_get_active_prompt(prompt_name: str = "main_agent") -> str:
|
||||||
|
"""Собирает структурированный текст системного промпта из активных узлов."""
|
||||||
|
with get_connection(row_factory=True) as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT section_id, item_id, content
|
||||||
|
FROM system_prompt_nodes
|
||||||
|
WHERE prompt_name = ? AND is_active = 1
|
||||||
|
ORDER BY section_id ASC, item_id ASC
|
||||||
|
""", (prompt_name,))
|
||||||
|
rows = cursor.fetchall()
|
||||||
|
|
||||||
|
if not rows:
|
||||||
|
return "Ты — ИИ-ассистент SCUD Orion AI."
|
||||||
|
|
||||||
|
lines = []
|
||||||
|
current_section = None
|
||||||
|
|
||||||
|
for r in rows:
|
||||||
|
sec_id = r["section_id"]
|
||||||
|
itm_id = r["item_id"]
|
||||||
|
content = r["content"]
|
||||||
|
|
||||||
|
if itm_id == 0:
|
||||||
|
if current_section is not None:
|
||||||
|
lines.append("")
|
||||||
|
lines.append(f"{sec_id}. {content}")
|
||||||
|
current_section = sec_id
|
||||||
|
else:
|
||||||
|
lines.append(f" {sec_id}.{itm_id}. {content}")
|
||||||
|
|
||||||
|
return "\n".join(lines)
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[PROMPT_REPO_APPLY_ACTION]
|
||||||
|
def repo_apply_prompt_action(action: str, section_id: int, item_id: int, content: str = "", prompt_name: str = "main_agent") -> None:
|
||||||
|
"""Точечно применяет действие (ADD / UPDATE / DELETE) над узлом промпта."""
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
action_clean = (action or "").upper()
|
||||||
|
if action_clean in ["ADD", "UPDATE", "EDIT"]:
|
||||||
|
cursor.execute("""
|
||||||
|
INSERT INTO system_prompt_nodes (prompt_name, section_id, item_id, content, is_active, updated_at)
|
||||||
|
VALUES (?, ?, ?, ?, 1, CURRENT_TIMESTAMP)
|
||||||
|
ON CONFLICT(prompt_name, section_id, item_id) DO UPDATE SET
|
||||||
|
content = excluded.content,
|
||||||
|
is_active = 1,
|
||||||
|
updated_at = CURRENT_TIMESTAMP
|
||||||
|
""", (prompt_name, section_id, item_id, content))
|
||||||
|
elif action_clean == "DELETE":
|
||||||
|
cursor.execute("""
|
||||||
|
DELETE FROM system_prompt_nodes
|
||||||
|
WHERE prompt_name = ? AND section_id = ? AND item_id = ?
|
||||||
|
""", (prompt_name, section_id, item_id))
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def repo_save_full_prompt(prompt_text: str, prompt_name: str = "main_agent") -> None:
|
||||||
|
"""Парсит и полностью перезаписывает все узлы промпта из сырого текста."""
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("DELETE FROM system_prompt_nodes WHERE prompt_name = ?", (prompt_name,))
|
||||||
|
|
||||||
|
current_sec = 1
|
||||||
|
current_itm = 0
|
||||||
|
|
||||||
|
for raw_line in prompt_text.splitlines():
|
||||||
|
clean_line = re.sub(r'<[^>]+>', '', raw_line).strip()
|
||||||
|
if not clean_line:
|
||||||
|
continue
|
||||||
|
|
||||||
|
sub_match = re.match(r'^(\d+)[\.\s]+(\d+)[\.\s\:\-]+(.*)$', clean_line)
|
||||||
|
sec_match = re.match(r'^(\d+)[\.\s\:\-]+(.*)$', clean_line)
|
||||||
|
|
||||||
|
if sub_match:
|
||||||
|
current_sec = int(sub_match.group(1))
|
||||||
|
current_itm = int(sub_match.group(2))
|
||||||
|
content = sub_match.group(3).strip()
|
||||||
|
elif sec_match and not any(c.islower() for c in sec_match.group(2)[:15]):
|
||||||
|
current_sec = int(sec_match.group(1))
|
||||||
|
current_itm = 0
|
||||||
|
content = sec_match.group(2).strip()
|
||||||
|
else:
|
||||||
|
current_itm += 1
|
||||||
|
content = clean_line
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
INSERT OR REPLACE INTO system_prompt_nodes (prompt_name, section_id, item_id, content, is_active)
|
||||||
|
VALUES (?, ?, ?, ?, 1)
|
||||||
|
""", (prompt_name, current_sec, current_itm, content))
|
||||||
|
|
||||||
|
conn.commit()
|
||||||
@@ -0,0 +1,49 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/prompts/service.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: services / prompts
|
||||||
|
ROLE: Единый доменный сервис управления системным промптом.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Tuple, Dict, Any, List, Optional
|
||||||
|
from .repository import repo_get_active_prompt, repo_apply_prompt_action, repo_save_full_prompt
|
||||||
|
from .diff_engine import build_prompt_diff
|
||||||
|
|
||||||
|
|
||||||
|
def get_active_system_prompt() -> str:
|
||||||
|
"""Получить текущий активный системный промпт."""
|
||||||
|
return repo_get_active_prompt()
|
||||||
|
|
||||||
|
|
||||||
|
def apply_prompt_action(action: str, section_id: int, item_id: int, content: str = "") -> None:
|
||||||
|
"""Применить точечное изменение к узлу промпта."""
|
||||||
|
repo_apply_prompt_action(action, section_id, item_id, content)
|
||||||
|
|
||||||
|
|
||||||
|
def save_full_prompt_draft(draft_text: str, prompt_name: str = "main_agent") -> None:
|
||||||
|
"""Сохранить полный черновик промпта."""
|
||||||
|
repo_save_full_prompt(draft_text, prompt_name=prompt_name)
|
||||||
|
|
||||||
|
|
||||||
|
def create_prompt_preview(
|
||||||
|
action: str,
|
||||||
|
section_id: Optional[int] = None,
|
||||||
|
item_id: Optional[int] = None,
|
||||||
|
content: str = "",
|
||||||
|
delete_nodes: Optional[List[Tuple[int, int]]] = None
|
||||||
|
) -> Tuple[str, str, str]:
|
||||||
|
"""
|
||||||
|
Формирует черновик и diff.
|
||||||
|
Возвращает (merged_draft, diff_html, baseline_prompt).
|
||||||
|
"""
|
||||||
|
baseline = repo_get_active_prompt()
|
||||||
|
draft, diff_html = build_prompt_diff(
|
||||||
|
action=action,
|
||||||
|
section_id=section_id,
|
||||||
|
item_id=item_id,
|
||||||
|
content=content,
|
||||||
|
delete_nodes=delete_nodes
|
||||||
|
)
|
||||||
|
return draft, diff_html, baseline
|
||||||
@@ -0,0 +1,71 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/scud_etl/anomaly_detector.py
|
||||||
|
ROLE: Детектирование аномалий СКУД, дубликатов пропусков и несоответствий с 1С.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from typing import List, Dict, Any
|
||||||
|
import pandas as pd
|
||||||
|
from config import normalize_fio
|
||||||
|
|
||||||
|
logger = logging.getLogger("SCUD_ANOMALY")
|
||||||
|
|
||||||
|
|
||||||
|
def detect_registry_anomalies(df_merged: pd.DataFrame, df_raw_scud: pd.DataFrame = None) -> List[Dict[str, Any]]:
|
||||||
|
anomalies = []
|
||||||
|
if df_merged is None or df_merged.empty:
|
||||||
|
return anomalies
|
||||||
|
|
||||||
|
# 1. ⭐️ Поиск дубликатов учеток/пропусков в сыром СКУД
|
||||||
|
if df_raw_scud is not None and not df_raw_scud.empty:
|
||||||
|
df_raw = df_raw_scud.copy()
|
||||||
|
if 'fio_clean' not in df_raw.columns:
|
||||||
|
f_col = 'Сотрудник' if 'Сотрудник' in df_raw.columns else 'ФИО'
|
||||||
|
df_raw['fio_clean'] = df_raw[f_col].apply(normalize_fio)
|
||||||
|
|
||||||
|
counts = df_raw['fio_clean'].value_counts()
|
||||||
|
duplicate_fios = counts[counts > 1].index.tolist()
|
||||||
|
|
||||||
|
for dup_fio in duplicate_fios:
|
||||||
|
sub = df_raw[df_raw['fio_clean'] == dup_fio]
|
||||||
|
departments = ", ".join(sub['Подразделение'].astype(str).unique())
|
||||||
|
anomalies.append({
|
||||||
|
"fio": dup_fio,
|
||||||
|
"type": "DUPLICATE_SCUD_CARD",
|
||||||
|
"description": f"Сотрудник заведен в СКУД {len(sub)} раза (отделы: {departments}). События входа и выхода объединены автоматически."
|
||||||
|
})
|
||||||
|
|
||||||
|
# 2. Поиск кадровых аномалий и сбоев оборудования
|
||||||
|
for _, row in df_merged.iterrows():
|
||||||
|
if row.get("is_excluded", False):
|
||||||
|
continue
|
||||||
|
|
||||||
|
fio = row.get("fio_clean") or row.get("Сотрудник", "")
|
||||||
|
start_day = str(row.get("Начало_дня", "")).strip()
|
||||||
|
end_day = str(row.get("Конец_дня", "")).strip()
|
||||||
|
reason = str(row.get("причина отсутствия", row.get("Вид_отсутствия", ""))).strip()
|
||||||
|
reason_lower = reason.lower()
|
||||||
|
|
||||||
|
if "исключен" in reason_lower or "овк" in reason_lower:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Приход в офис во время отпуска/больничного (удаленка и командировки разрешены)
|
||||||
|
if start_day not in ["Нет входа", "—", "", "nan", "None"] and reason and reason != "nan":
|
||||||
|
if not ("удален" in reason_lower or "дистанцион" in reason_lower or "командировк" in reason_lower or "поездк" in reason_lower):
|
||||||
|
anomalies.append({
|
||||||
|
"fio": fio,
|
||||||
|
"type": "PHYSICAL_PRESENCE_DURING_ABSENCE",
|
||||||
|
"description": f"Присутствовал в здании ({start_day}), но в 1С числится документ: '{reason}'."
|
||||||
|
})
|
||||||
|
|
||||||
|
# Ошибка считывателя (есть выход без входа)
|
||||||
|
if start_day in ["Нет входа", "—", ""] and end_day not in ["Нет выхода", "—", "", "nan", "None"]:
|
||||||
|
anomalies.append({
|
||||||
|
"fio": fio,
|
||||||
|
"type": "SCUD_EQUIPMENT_ANOMALY",
|
||||||
|
"description": f"Зафиксирован выход ({end_day}) при отсутствии отметки утреннего входа."
|
||||||
|
})
|
||||||
|
|
||||||
|
return anomalies
|
||||||
@@ -0,0 +1,262 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/scud_etl/merger.py
|
||||||
|
ROLE: Агрегация реестров, выбор совместителей 1С по отделу СКУД и авто-связки.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from typing import Dict, Any, List
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
from services.knowledge.service import get_department_synonyms_dict
|
||||||
|
from config import normalize_fio, load_exceptions
|
||||||
|
from core.connection import get_connection
|
||||||
|
|
||||||
|
logger = logging.getLogger("SCUD_MERGER")
|
||||||
|
|
||||||
|
|
||||||
|
def load_identity_mappings() -> Dict[str, str]:
|
||||||
|
"""Загружает подтвержденные сопоставления ФИО (СКУД -> 1С:ЗУП) из SQLite."""
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
# Автоматическая инициализация таблицы при первом обращении
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS person_identity_mapping (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
scud_fio TEXT NOT NULL,
|
||||||
|
zup_fio TEXT NOT NULL,
|
||||||
|
scud_dept TEXT,
|
||||||
|
zup_dept TEXT,
|
||||||
|
match_source TEXT DEFAULT 'AI',
|
||||||
|
status TEXT DEFAULT 'ACTIVE',
|
||||||
|
confidence REAL DEFAULT 1.0,
|
||||||
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
UNIQUE(scud_fio, zup_fio)
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT scud_fio, zup_fio
|
||||||
|
FROM person_identity_mapping
|
||||||
|
WHERE status = 'ACTIVE'
|
||||||
|
""")
|
||||||
|
return {r[0]: r[1] for r in cursor.fetchall()}
|
||||||
|
|
||||||
|
|
||||||
|
def aggregate_scud_by_person(df_scud: pd.DataFrame) -> pd.DataFrame:
|
||||||
|
"""Схлопывает дубликаты пропусков одного человека в СКУД."""
|
||||||
|
if df_scud is None or df_scud.empty:
|
||||||
|
return df_scud
|
||||||
|
|
||||||
|
df = df_scud.copy()
|
||||||
|
if 'fio_clean' not in df.columns:
|
||||||
|
fio_col = 'Сотрудник' if 'Сотрудник' in df.columns else 'ФИО'
|
||||||
|
df['fio_clean'] = df[fio_col].apply(normalize_fio)
|
||||||
|
|
||||||
|
# Применяем сохраненный кэш связок ФИО
|
||||||
|
mapping_dict = load_identity_mappings()
|
||||||
|
if mapping_dict:
|
||||||
|
df['fio_clean'] = df['fio_clean'].apply(lambda f: mapping_dict.get(f, f))
|
||||||
|
|
||||||
|
aggregated_rows = []
|
||||||
|
for fio_clean, group in df.groupby('fio_clean'):
|
||||||
|
if len(group) == 1:
|
||||||
|
aggregated_rows.append(group.iloc[0].to_dict())
|
||||||
|
continue
|
||||||
|
|
||||||
|
base_row = group.sort_values(by='Пришел', ascending=False).iloc[0].to_dict()
|
||||||
|
|
||||||
|
valid_ins = [
|
||||||
|
str(t).strip() for t in group['Начало_дня']
|
||||||
|
if str(t).strip() not in ['Нет входа', '—', '', 'nan', 'None', '00:00:00', '00:00']
|
||||||
|
]
|
||||||
|
base_row['Начало_дня'] = min(valid_ins) if valid_ins else 'Нет входа'
|
||||||
|
|
||||||
|
valid_outs = [
|
||||||
|
str(t).strip() for t in group['Конец_дня']
|
||||||
|
if str(t).strip() not in ['Нет выхода', '—', '', 'nan', 'None', '00:00:00', '00:00']
|
||||||
|
]
|
||||||
|
base_row['Конец_дня'] = max(valid_outs) if valid_outs else 'Нет выхода'
|
||||||
|
|
||||||
|
valid_acts = [
|
||||||
|
str(t).strip() for t in group['Первая_активность']
|
||||||
|
if str(t).strip() not in ['—', '', 'nan', 'None', '00:00:00']
|
||||||
|
]
|
||||||
|
base_row['Первая_активность'] = min(valid_acts) if valid_acts else '—'
|
||||||
|
base_row['Пришел'] = any(group['Пришел'] == True) or (base_row['Начало_дня'] != 'Нет входа')
|
||||||
|
|
||||||
|
durations = [str(d) for d in group['Находился_в_здании'] if str(d) not in ['00:00', '', 'nan']]
|
||||||
|
if durations:
|
||||||
|
base_row['Находился_в_здании'] = max(durations)
|
||||||
|
|
||||||
|
aggregated_rows.append(base_row)
|
||||||
|
|
||||||
|
return pd.DataFrame(aggregated_rows)
|
||||||
|
|
||||||
|
|
||||||
|
def select_best_zup_position(df_staff_1c: pd.DataFrame, df_scud_agg: pd.DataFrame) -> pd.DataFrame:
|
||||||
|
"""
|
||||||
|
⭐️ Для совместителей с несколькими должностями в 1С:ЗУП
|
||||||
|
выбирает ту ставку, которая соответствует отделу физического нахождения по СКУД.
|
||||||
|
"""
|
||||||
|
if df_staff_1c is None or df_staff_1c.empty:
|
||||||
|
return pd.DataFrame()
|
||||||
|
|
||||||
|
df_staff = df_staff_1c.copy()
|
||||||
|
if 'fio_clean' not in df_staff.columns:
|
||||||
|
f_col = 'ФИО' if 'ФИО' in df_staff.columns else 'Сотрудник'
|
||||||
|
df_staff['fio_clean'] = df_staff[f_col].apply(normalize_fio)
|
||||||
|
|
||||||
|
scud_dept_map = {}
|
||||||
|
if df_scud_agg is not None and not df_scud_agg.empty:
|
||||||
|
for _, r in df_scud_agg.iterrows():
|
||||||
|
scud_dept_map[r.get('fio_clean', '')] = str(r.get('Подразделение', '')).strip().upper()
|
||||||
|
|
||||||
|
best_rows = []
|
||||||
|
for fio, group in df_staff.groupby('fio_clean'):
|
||||||
|
if len(group) == 1:
|
||||||
|
best_rows.append(group.iloc[0].to_dict())
|
||||||
|
continue
|
||||||
|
|
||||||
|
target_scud_dept = scud_dept_map.get(fio, "")
|
||||||
|
matched_row = None
|
||||||
|
|
||||||
|
if target_scud_dept:
|
||||||
|
for _, r in group.iterrows():
|
||||||
|
dept_1c = str(r.get('Подразделение', '')).strip().upper()
|
||||||
|
if dept_1c == target_scud_dept or target_scud_dept in dept_1c or dept_1c in target_scud_dept:
|
||||||
|
matched_row = r.to_dict()
|
||||||
|
break
|
||||||
|
|
||||||
|
if not matched_row:
|
||||||
|
matched_row = group.iloc[0].to_dict()
|
||||||
|
|
||||||
|
best_rows.append(matched_row)
|
||||||
|
|
||||||
|
return pd.DataFrame(best_rows)
|
||||||
|
|
||||||
|
|
||||||
|
def merge_scud_and_1c(
|
||||||
|
df_scud: pd.DataFrame,
|
||||||
|
df_staff_1c: pd.DataFrame,
|
||||||
|
df_absences_1c: pd.DataFrame
|
||||||
|
) -> pd.DataFrame:
|
||||||
|
if (df_scud is None or df_scud.empty) and (df_staff_1c is None or df_staff_1c.empty):
|
||||||
|
return pd.DataFrame(columns=[
|
||||||
|
'Сотрудник', 'fio_clean', 'Подразделение', 'Должность',
|
||||||
|
'Начало_дня', 'Первая_активность', 'Конец_дня', 'Находился_в_здании',
|
||||||
|
'Пришел', 'anomaly_flag', 'причина отсутствия', 'Вид_отсутствия', 'is_excluded'
|
||||||
|
])
|
||||||
|
|
||||||
|
synonyms = get_department_synonyms_dict()
|
||||||
|
exceptions_cfg = load_exceptions()
|
||||||
|
|
||||||
|
df_scud_agg = aggregate_scud_by_person(df_scud)
|
||||||
|
df_staff_agg = select_best_zup_position(df_staff_1c, df_scud_agg)
|
||||||
|
|
||||||
|
df_res = df_scud_agg.copy() if df_scud_agg is not None and not df_scud_agg.empty else df_staff_agg.copy()
|
||||||
|
|
||||||
|
if "Сотрудник" in df_res.columns:
|
||||||
|
df_res["fio_clean"] = df_res["Сотрудник"].apply(normalize_fio)
|
||||||
|
elif "ФИО" in df_res.columns:
|
||||||
|
df_res["Сотрудник"] = df_res["ФИО"]
|
||||||
|
df_res["fio_clean"] = df_res["ФИО"].apply(normalize_fio)
|
||||||
|
elif "fio_clean" not in df_res.columns:
|
||||||
|
df_res["fio_clean"] = ""
|
||||||
|
|
||||||
|
for col, default_val in [
|
||||||
|
('Начало_дня', 'Нет входа'),
|
||||||
|
('Первая_активность', '—'),
|
||||||
|
('Конец_дня', 'Нет выхода'),
|
||||||
|
('Находился_в_здании', '00:00'),
|
||||||
|
('Пришел', False),
|
||||||
|
('anomaly_flag', 'NONE')
|
||||||
|
]:
|
||||||
|
if col not in df_res.columns:
|
||||||
|
df_res[col] = default_val
|
||||||
|
|
||||||
|
reverse_synonyms = {v.lower(): k.upper() for k, v in synonyms.items()}
|
||||||
|
direct_synonyms = {k.lower(): k.upper() for k in synonyms.keys()}
|
||||||
|
all_dept_map = {**reverse_synonyms, **direct_synonyms, "отдел внутреннего контроля": "ОВК", "отдел вневедомственного контроля": "ОВК"}
|
||||||
|
|
||||||
|
if "Подразделение" in df_res.columns:
|
||||||
|
df_res["Подразделение"] = df_res["Подразделение"].apply(
|
||||||
|
lambda d: all_dept_map.get(str(d).strip().lower(), str(d).strip())
|
||||||
|
)
|
||||||
|
|
||||||
|
absences_map = {}
|
||||||
|
if df_absences_1c is not None and not df_absences_1c.empty:
|
||||||
|
fio_col = next((c for c in ["fio_clean", "ФИО", "Сотрудник"] if c in df_absences_1c.columns), None)
|
||||||
|
reason_col = next((c for c in ["Вид_отсутствия", "Причина", "причина отсутствия"] if c in df_absences_1c.columns), None)
|
||||||
|
|
||||||
|
if fio_col and reason_col:
|
||||||
|
for _, row in df_absences_1c.iterrows():
|
||||||
|
fio = normalize_fio(str(row[fio_col]))
|
||||||
|
reason = str(row[reason_col]).strip()
|
||||||
|
if reason and reason.lower() != "nan":
|
||||||
|
absences_map[fio] = reason
|
||||||
|
|
||||||
|
df_res["причина отсутствия"] = df_res["fio_clean"].map(absences_map)
|
||||||
|
df_res["Вид_отсутствия"] = df_res["причина отсутствия"]
|
||||||
|
|
||||||
|
exc_fios = [normalize_fio(f) for f in exceptions_cfg.get("fio", []) if f]
|
||||||
|
exc_depts = [d.strip().upper() for d in exceptions_cfg.get("departments", []) if d]
|
||||||
|
exc_pos = [p.strip().lower() for p in exceptions_cfg.get("positions", []) if p]
|
||||||
|
pos_kw = [k.strip().lower() for k in exceptions_cfg.get("position_keywords", []) if k]
|
||||||
|
whitelist_fios = [normalize_fio(f) for f in exceptions_cfg.get("include_fio", []) if f]
|
||||||
|
|
||||||
|
df_res["is_excluded"] = False
|
||||||
|
for idx, row in df_res.iterrows():
|
||||||
|
fio = row.get("fio_clean", "")
|
||||||
|
has_official_absence = pd.notna(row.get("Вид_отсутствия")) and str(row.get("Вид_отсутствия")).strip() not in ["", "nan", "None", "Исключение"]
|
||||||
|
|
||||||
|
if fio in whitelist_fios or has_official_absence:
|
||||||
|
df_res.at[idx, "is_excluded"] = False
|
||||||
|
continue
|
||||||
|
|
||||||
|
dep = str(row.get("Подразделение", "")).upper()
|
||||||
|
pos = str(row.get("Должность", "")).lower()
|
||||||
|
|
||||||
|
if fio in exc_fios or dep in exc_depts or any(d in dep for d in exc_depts) or pos in exc_pos or any(k in pos for k in pos_kw):
|
||||||
|
df_res.at[idx, "is_excluded"] = True
|
||||||
|
|
||||||
|
mask_exc = (df_res["is_excluded"] == True) & (df_res["Вид_отсутствия"].isna() | (df_res["Вид_отсутствия"] == ""))
|
||||||
|
df_res.loc[mask_exc, "Вид_отсутствия"] = "Исключение"
|
||||||
|
df_res.loc[mask_exc, "причина отсутствия"] = "Исключение"
|
||||||
|
|
||||||
|
return df_res
|
||||||
|
|
||||||
|
|
||||||
|
def calculate_summary_metrics(df_merged: pd.DataFrame) -> Dict[str, Any]:
|
||||||
|
active_df = df_merged[df_merged.get("is_excluded", False) == False].copy()
|
||||||
|
total_staff = len(active_df)
|
||||||
|
|
||||||
|
came_to_office_mask = active_df["Начало_дня"].astype(str).str.strip().ne("Нет входа")
|
||||||
|
working_in_office = active_df[came_to_office_mask]
|
||||||
|
working_in_office_count = len(working_in_office)
|
||||||
|
|
||||||
|
df_not_came = active_df[~came_to_office_mask]
|
||||||
|
|
||||||
|
reason_series = df_not_came["причина отсутствия"].astype(str).str.lower()
|
||||||
|
is_remote_mask = reason_series.str.contains("удален|дистанцион", regex=True, na=False)
|
||||||
|
remote_home = df_not_came[is_remote_mask]
|
||||||
|
remote_home_count = len(remote_home)
|
||||||
|
|
||||||
|
df_remaining_absent = df_not_came[~is_remote_mask]
|
||||||
|
has_doc_mask = df_remaining_absent["причина отсутствия"].notna() & \
|
||||||
|
df_remaining_absent["причина отсутствия"].ne("") & \
|
||||||
|
df_remaining_absent["причина отсутствия"].ne("nan") & \
|
||||||
|
(~df_remaining_absent["причина отсутствия"].astype(str).str.startswith("Исключение"))
|
||||||
|
official_absent_count = len(df_remaining_absent[has_doc_mask])
|
||||||
|
|
||||||
|
unknown = df_remaining_absent[~has_doc_mask]
|
||||||
|
unknown_count = len(unknown)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"total_staff": total_staff,
|
||||||
|
"working_in_office_count": working_in_office_count,
|
||||||
|
"remote_home_count": remote_home_count,
|
||||||
|
"official_absent_count": official_absent_count,
|
||||||
|
"unknown_count": unknown_count,
|
||||||
|
"unknown_list": unknown[["fio_clean", "Подразделение", "Должность"]].to_dict(orient="records") if not unknown.empty else []
|
||||||
|
}
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/scud_etl/otchet_generator.py
|
||||||
|
ROLE: Генератор Детального Отчета за прошлые смены (строго по итоговому Y-снапшоту).
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import logging
|
||||||
|
from typing import Dict, Any, Optional
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
from config import DATE_YESTERDAY
|
||||||
|
from core.database import load_scud_from_db_by_snapshot
|
||||||
|
from services.scud_etl.pipeline import load_1c_files_for_date
|
||||||
|
from services.scud_etl.merger import merge_scud_and_1c
|
||||||
|
from services.excel_exporter import generate_detailed_excel, get_dated_reports_dir, format_date_ru
|
||||||
|
|
||||||
|
logger = logging.getLogger("OTCHET_GENERATOR")
|
||||||
|
|
||||||
|
|
||||||
|
def generate_otchet_service(
|
||||||
|
target_date: Optional[str] = None,
|
||||||
|
snapshot_id: Optional[str] = None
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
Формирует детальный отчет за прошедшую смену:
|
||||||
|
- target_date: дата отчета (по умолчанию вчерашний рабочий день).
|
||||||
|
- snapshot_id: опциональный ID (по умолчанию выбирается итоговый вечерний срез Y).
|
||||||
|
"""
|
||||||
|
date_clean = str(target_date or DATE_YESTERDAY).replace('_', '.')
|
||||||
|
|
||||||
|
df_scud = load_scud_from_db_by_snapshot(date_clean, snapshot_param=snapshot_id)
|
||||||
|
if df_scud is None or df_scud.empty:
|
||||||
|
return {
|
||||||
|
"status": "error",
|
||||||
|
"message": f"Итоговый срез СКУД (Y) за {date_clean} не найден в базе данных."
|
||||||
|
}
|
||||||
|
|
||||||
|
df_staff, df_abs = load_1c_files_for_date(date_clean)
|
||||||
|
df_merged = merge_scud_and_1c(df_scud, df_staff, df_abs)
|
||||||
|
|
||||||
|
filename = f"{format_date_ru(date_clean)} отчет.xlsx"
|
||||||
|
generate_detailed_excel(df_merged, date_str=date_clean, filename=filename)
|
||||||
|
|
||||||
|
target_dir = get_dated_reports_dir(date_clean)
|
||||||
|
full_filepath = os.path.join(target_dir, filename)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"status": "success",
|
||||||
|
"report_type": "OTCHET",
|
||||||
|
"date": date_clean,
|
||||||
|
"snapshot_id": snapshot_id or "AUTO_Y_FINAL",
|
||||||
|
"filename": filename,
|
||||||
|
"filepath": full_filepath,
|
||||||
|
"download_url": f"/api/v1/files/download/reports/{os.path.basename(full_filepath)}",
|
||||||
|
"total_rows": len(df_merged),
|
||||||
|
"message": f"Детальный отчет за {date_clean} успешно сформирован."
|
||||||
|
}
|
||||||
@@ -0,0 +1,64 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/scud_etl/pipeline.py
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import logging
|
||||||
|
from typing import Optional, Dict, Any, Tuple
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
from core.connection import get_connection
|
||||||
|
from core.database import load_scud_from_db_by_snapshot
|
||||||
|
from config import DATA_DIR
|
||||||
|
from services.data_loader import load_staff_data, load_absent_data
|
||||||
|
|
||||||
|
logger = logging.getLogger("SCUD_PIPELINE")
|
||||||
|
|
||||||
|
|
||||||
|
def load_best_snapshot_for_date(date_str: str, prefer_final_y: bool = False) -> Optional[pd.DataFrame]:
|
||||||
|
"""
|
||||||
|
Загружает наилучший срез СКУД за дату.
|
||||||
|
Если prefer_final_y=True — отдает предпочтение финишному Y (23:59:59).
|
||||||
|
"""
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
target_snap_id = None
|
||||||
|
|
||||||
|
if prefer_final_y:
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT snapshot_id
|
||||||
|
FROM scud_logs
|
||||||
|
WHERE log_date = ?
|
||||||
|
AND (snapshot_id LIKE 'Y%' OR snapshot_time LIKE '%23:59:59' OR snapshot_time LIKE '%22:00:00')
|
||||||
|
ORDER BY id DESC LIMIT 1
|
||||||
|
""", (date_str,))
|
||||||
|
row = cursor.fetchone()
|
||||||
|
if row:
|
||||||
|
target_snap_id = row[0]
|
||||||
|
|
||||||
|
if not target_snap_id:
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT snapshot_id
|
||||||
|
FROM scud_logs
|
||||||
|
WHERE log_date = ?
|
||||||
|
ORDER BY id DESC LIMIT 1
|
||||||
|
""", (date_str,))
|
||||||
|
row = cursor.fetchone()
|
||||||
|
if row:
|
||||||
|
target_snap_id = row[0]
|
||||||
|
|
||||||
|
if not target_snap_id:
|
||||||
|
return None
|
||||||
|
|
||||||
|
return load_scud_from_db_by_snapshot(date_str, snapshot_param=target_snap_id)
|
||||||
|
|
||||||
|
|
||||||
|
def load_1c_files_for_date(date_str: str) -> Tuple[Optional[pd.DataFrame], Optional[pd.DataFrame]]:
|
||||||
|
"""
|
||||||
|
Загружает реестры штата и отсутствий 1С на указанную дату через data_loader.
|
||||||
|
"""
|
||||||
|
df_staff = load_staff_data(date_str)
|
||||||
|
df_abs = load_absent_data(date_str)
|
||||||
|
return df_staff, df_abs
|
||||||
@@ -0,0 +1,116 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/scud_etl/sql_queries.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: services / scud_etl
|
||||||
|
ROLE: Хранилище сырых SQL-шаблонов для выгрузки из MS SQL Server (СКУД Орион Pro).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[SQL_SCUD_EXPORT_TEMPLATE]: T-SQL запрос с расчетом первой активности и длительности.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
# ANCHOR[SQL_SCUD_EXPORT_TEMPLATE]
|
||||||
|
SCUD_EXPORT_QUERY_TEMPLATE = r"""
|
||||||
|
DECLARE @InputDate DATE = '{target_date}';
|
||||||
|
DECLARE @TargetDate DATE = @InputDate;
|
||||||
|
|
||||||
|
DECLARE @StartDate DATETIME = CAST(@TargetDate AS DATETIME);
|
||||||
|
DECLARE @EndDate DATETIME = DATEADD(SECOND, -1, DATEADD(DAY, 1, @StartDate));
|
||||||
|
|
||||||
|
WITH DailyLogs AS (
|
||||||
|
SELECT
|
||||||
|
log.HozOrgan AS EmployeeID,
|
||||||
|
log.TimeVal,
|
||||||
|
log.Event,
|
||||||
|
log.Mode,
|
||||||
|
CASE
|
||||||
|
WHEN log.Mode = 2 OR log.Event IN (29, 27, 33) THEN 'OUT'
|
||||||
|
WHEN log.Mode = 1 OR log.Event IN (28, 26, 32) THEN 'IN'
|
||||||
|
ELSE 'OTHER'
|
||||||
|
END AS Direction,
|
||||||
|
ROW_NUMBER() OVER (PARTITION BY log.HozOrgan ORDER BY log.TimeVal DESC) AS RowNumDesc
|
||||||
|
FROM pLogData log WITH (NOLOCK)
|
||||||
|
WHERE log.TimeVal BETWEEN @StartDate AND @EndDate
|
||||||
|
AND log.HozOrgan IS NOT NULL
|
||||||
|
AND log.HozOrgan > 0
|
||||||
|
AND log.Event IN (26, 27, 28, 29, 32, 33, 54, 55, 64, 65)
|
||||||
|
),
|
||||||
|
Passages AS (
|
||||||
|
SELECT
|
||||||
|
EmployeeID,
|
||||||
|
MIN(TimeVal) AS FirstRawEvent,
|
||||||
|
MAX(TimeVal) AS LastRawEvent,
|
||||||
|
MIN(CASE WHEN Direction = 'IN' THEN TimeVal END) AS FirstIn,
|
||||||
|
MAX(CASE WHEN Direction = 'OUT' THEN TimeVal END) AS LastOut,
|
||||||
|
MAX(CASE WHEN RowNumDesc = 1 THEN Direction END) AS LastEventType
|
||||||
|
FROM DailyLogs
|
||||||
|
GROUP BY EmployeeID
|
||||||
|
)
|
||||||
|
SELECT
|
||||||
|
N'ЛЕНМОРНИИПРОЕКТ' AS [Фирма],
|
||||||
|
ISNULL(CAST(div.Name AS NVARCHAR(255)), N'Без подразделения') AS [Подразделение],
|
||||||
|
LTRIM(RTRIM(
|
||||||
|
ISNULL(CAST(p.Name AS NVARCHAR(255)), N'') +
|
||||||
|
CASE WHEN p.FirstName IS NOT NULL AND CAST(p.FirstName AS NVARCHAR(255)) <> ''
|
||||||
|
THEN N' ' + CAST(p.FirstName AS NVARCHAR(255)) ELSE N'' END +
|
||||||
|
CASE WHEN p.MidName IS NOT NULL AND CAST(p.MidName AS NVARCHAR(255)) <> ''
|
||||||
|
THEN N' ' + CAST(p.MidName AS NVARCHAR(255)) ELSE N'' END
|
||||||
|
)) AS [Сотрудник],
|
||||||
|
ISNULL(CAST(post.Name AS NVARCHAR(255)), N'—') AS [Должность],
|
||||||
|
ISNULL(CAST(p.TabNumber AS NVARCHAR(50)), N'—') AS [Таб_№],
|
||||||
|
CONVERT(VARCHAR(10), @TargetDate, 104) AS [Дата],
|
||||||
|
ISNULL(CAST(CONVERT(VARCHAR(8), pass.FirstIn, 108) AS NVARCHAR(20)), N'Нет входа') AS [Начало_дня],
|
||||||
|
CASE
|
||||||
|
WHEN pass.FirstIn IS NULL AND pass.FirstRawEvent IS NOT NULL
|
||||||
|
THEN CAST(CONVERT(VARCHAR(8), pass.FirstRawEvent, 108) AS NVARCHAR(20))
|
||||||
|
ELSE N'—'
|
||||||
|
END AS [Первая_активность],
|
||||||
|
CASE
|
||||||
|
WHEN @TargetDate = CAST(GETDATE() AS DATE) AND (pass.LastEventType = 'IN' OR pass.LastOut IS NULL OR pass.LastOut <= pass.FirstIn)
|
||||||
|
THEN N'Нет выхода'
|
||||||
|
WHEN pass.LastOut IS NOT NULL AND pass.LastOut > pass.FirstIn
|
||||||
|
THEN CAST(CONVERT(VARCHAR(8), pass.LastOut, 108) AS NVARCHAR(20))
|
||||||
|
WHEN @TargetDate < CAST(GETDATE() AS DATE) AND pass.LastRawEvent IS NOT NULL AND pass.LastRawEvent > ISNULL(pass.FirstIn, pass.FirstRawEvent)
|
||||||
|
THEN CAST(CONVERT(VARCHAR(8), pass.LastRawEvent, 108) AS NVARCHAR(20))
|
||||||
|
ELSE N'Нет выхода'
|
||||||
|
END AS [Конец_дня],
|
||||||
|
CASE
|
||||||
|
WHEN pass.EmployeeID IS NOT NULL AND (pass.FirstIn IS NOT NULL OR pass.FirstRawEvent IS NOT NULL) THEN
|
||||||
|
RIGHT('0' + CAST(DATEDIFF(MINUTE,
|
||||||
|
ISNULL(pass.FirstIn, pass.FirstRawEvent),
|
||||||
|
CASE
|
||||||
|
WHEN @TargetDate = CAST(GETDATE() AS DATE) AND (pass.LastEventType = 'IN' OR pass.LastOut IS NULL OR pass.LastOut <= pass.FirstIn) THEN GETDATE()
|
||||||
|
ELSE ISNULL(pass.LastOut, pass.LastRawEvent)
|
||||||
|
END) / 60 AS VARCHAR), 2) + ':' +
|
||||||
|
RIGHT('0' + CAST(DATEDIFF(MINUTE,
|
||||||
|
ISNULL(pass.FirstIn, pass.FirstRawEvent),
|
||||||
|
CASE
|
||||||
|
WHEN @TargetDate = CAST(GETDATE() AS DATE) AND (pass.LastEventType = 'IN' OR pass.LastOut IS NULL OR pass.LastOut <= pass.FirstIn) THEN GETDATE()
|
||||||
|
ELSE ISNULL(pass.LastOut, pass.LastRawEvent)
|
||||||
|
END) % 60 AS VARCHAR), 2)
|
||||||
|
ELSE N'00:00'
|
||||||
|
END AS [Находился_в_здании],
|
||||||
|
CASE
|
||||||
|
WHEN pass.EmployeeID IS NOT NULL THEN N'Присутствовал'
|
||||||
|
ELSE N'Отсутствовал (Нет событий)'
|
||||||
|
END AS [Статус]
|
||||||
|
FROM pList p WITH (NOLOCK)
|
||||||
|
LEFT JOIN PDivision div WITH (NOLOCK) ON p.Section = div.ID
|
||||||
|
LEFT JOIN PPost post WITH (NOLOCK) ON p.Post = post.ID
|
||||||
|
LEFT JOIN Passages pass ON p.ID = pass.EmployeeID
|
||||||
|
WHERE
|
||||||
|
ISNULL(p.StatusRecord, 0) = 0
|
||||||
|
AND p.DateTimeInArchive IS NULL
|
||||||
|
AND ISNULL(CAST(div.Name AS NVARCHAR(255)), N'') NOT LIKE N'Аренд%'
|
||||||
|
AND ISNULL(CAST(div.Name AS NVARCHAR(255)), N'') NOT IN (N'Без подразделения', N'')
|
||||||
|
AND p.Name NOT LIKE N'бр.%'
|
||||||
|
AND p.Name NOT LIKE N'Гость%'
|
||||||
|
AND ISNULL(CAST(div.Name AS NVARCHAR(255)), N'') NOT IN (N'БГИ', N'КНР')
|
||||||
|
AND ISNULL(CAST(div.Name AS NVARCHAR(255)), N'') NOT LIKE N'Рабоч%'
|
||||||
|
AND ISNULL(CAST(div.Name AS NVARCHAR(255)), N'') NOT LIKE N'Врем%'
|
||||||
|
AND ISNULL(CAST(div.Name AS NVARCHAR(255)), N'') NOT LIKE N'Практика%'
|
||||||
|
AND ISNULL(CAST(div.Name AS NVARCHAR(255)), N'') NOT LIKE N'тест%'
|
||||||
|
AND ISNULL(CAST(post.Name AS NVARCHAR(255)), N'') NOT LIKE N'Практикант%'
|
||||||
|
ORDER BY p.Name ASC;
|
||||||
|
"""
|
||||||
@@ -0,0 +1,80 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/scud_etl/svodka_generator.py
|
||||||
|
ROLE: Генератор Ежедневной Сводки (оперативный контроль, текущий срез).
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import logging
|
||||||
|
from typing import Dict, Any, Optional
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
from config import DATE_TODAY
|
||||||
|
from core.database import load_scud_from_db_by_snapshot
|
||||||
|
from services.scud_etl.pipeline import load_1c_files_for_date
|
||||||
|
from services.scud_etl.merger import merge_scud_and_1c, calculate_summary_metrics
|
||||||
|
from services.scud_etl.anomaly_detector import detect_registry_anomalies
|
||||||
|
from services.snapshots.finder import find_or_create_snapshot_for_time
|
||||||
|
from services.excel_exporter import generate_summary_excel, get_dated_reports_dir, format_date_ru
|
||||||
|
|
||||||
|
logger = logging.getLogger("SVODKA_GENERATOR")
|
||||||
|
|
||||||
|
|
||||||
|
def generate_svodka_service(
|
||||||
|
target_date: Optional[str] = None,
|
||||||
|
target_time: Optional[str] = None,
|
||||||
|
snapshot_id: Optional[str] = None
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
Формирует оперативную сводку на указанную дату / время:
|
||||||
|
- target_date: дата сводки (по умолчанию сегодня).
|
||||||
|
- target_time: время среза (например '14:30').
|
||||||
|
- snapshot_id: точный ID среза.
|
||||||
|
"""
|
||||||
|
date_clean = str(target_date or DATE_TODAY).replace('_', '.')
|
||||||
|
applied_note = ""
|
||||||
|
|
||||||
|
# Если передано время, но не указан конкретный snapshot_id — ищем ближайший или запрашиваем экспорт
|
||||||
|
if target_time and not snapshot_id:
|
||||||
|
found_id, note = find_or_create_snapshot_for_time(date_clean, target_time, allow_ondemand_export=True)
|
||||||
|
snapshot_id = found_id
|
||||||
|
applied_note = note
|
||||||
|
if note:
|
||||||
|
logger.info(note)
|
||||||
|
|
||||||
|
df_scud = load_scud_from_db_by_snapshot(date_clean, snapshot_param=snapshot_id)
|
||||||
|
if df_scud is None or df_scud.empty:
|
||||||
|
return {
|
||||||
|
"status": "error",
|
||||||
|
"message": f"Срез СКУД за {date_clean} ({applied_note or snapshot_id or 'последний доступный'}) не найден в базе."
|
||||||
|
}
|
||||||
|
|
||||||
|
df_staff, df_abs = load_1c_files_for_date(date_clean)
|
||||||
|
|
||||||
|
df_merged = merge_scud_and_1c(df_scud, df_staff, df_abs)
|
||||||
|
metrics = calculate_summary_metrics(df_merged)
|
||||||
|
anomalies = detect_registry_anomalies(df_merged, df_raw_scud=df_scud)
|
||||||
|
|
||||||
|
# Добавляем суффикс времени в имя файла, если сводка строилась на точный срез
|
||||||
|
time_suffix = f" на {target_time.replace(':', '-')}" if target_time else ""
|
||||||
|
filename = f"{format_date_ru(date_clean)} сводка{time_suffix}.xlsx"
|
||||||
|
generate_summary_excel(df_merged, date_str=date_clean, filename=filename)
|
||||||
|
|
||||||
|
target_dir = get_dated_reports_dir(date_clean)
|
||||||
|
full_filepath = os.path.join(target_dir, filename)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"status": "success",
|
||||||
|
"report_type": "SVODKA",
|
||||||
|
"date": date_clean,
|
||||||
|
"target_time": target_time,
|
||||||
|
"snapshot_id": snapshot_id or "AUTO_LATEST",
|
||||||
|
"filename": filename,
|
||||||
|
"filepath": full_filepath,
|
||||||
|
"download_url": f"/api/v1/files/download/reports/{os.path.basename(full_filepath)}",
|
||||||
|
"metrics": metrics,
|
||||||
|
"anomalies_count": len(anomalies),
|
||||||
|
"note": applied_note,
|
||||||
|
"message": f"Ежедневная сводка на {date_clean} {target_time or ''} успешно сформирована."
|
||||||
|
}
|
||||||
@@ -28,6 +28,7 @@ LOG_FILE = os.path.join(LOG_DIR, f"export_{TODAY_DATE_STR}.log")
|
|||||||
|
|
||||||
logger = logging.getLogger("scud_export")
|
logger = logging.getLogger("scud_export")
|
||||||
logger.setLevel(logging.INFO)
|
logger.setLevel(logging.INFO)
|
||||||
|
logger.propagate = False
|
||||||
|
|
||||||
if not logger.handlers:
|
if not logger.handlers:
|
||||||
_file_handler = logging.FileHandler(LOG_FILE, encoding="utf-8")
|
_file_handler = logging.FileHandler(LOG_FILE, encoding="utf-8")
|
||||||
@@ -52,7 +53,7 @@ def log(message: str, level: str = "INFO"):
|
|||||||
logger.log(level_map.get(level, logging.INFO), message)
|
logger.log(level_map.get(level, logging.INFO), message)
|
||||||
|
|
||||||
|
|
||||||
SERVER_NAME = r"172.16.31.221\SQL"
|
SERVER_NAME = r"172.16.200.147\SQL"
|
||||||
DATABASE_NAME = "Orion-14.01.21-1"
|
DATABASE_NAME = "Orion-14.01.21-1"
|
||||||
SQL_USER = "sa"
|
SQL_USER = "sa"
|
||||||
SQL_PASSWORD = "123456"
|
SQL_PASSWORD = "123456"
|
||||||
@@ -236,7 +237,7 @@ def run_export(input_date: str | None = None, save_xlsx: bool = True, debug: boo
|
|||||||
continue
|
continue
|
||||||
|
|
||||||
if is_yesterday:
|
if is_yesterday:
|
||||||
snapshot_time = f"{processing_date.strftime('%Y-%m-%d')} 22:00:00"
|
snapshot_time = f"{processing_date.strftime('%Y-%m-%d')} 23:59:59"
|
||||||
else:
|
else:
|
||||||
snapshot_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
snapshot_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,89 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/snapshots/finder.py
|
||||||
|
ROLE: Поиск ближайшего снапшота в SQLite (Smart Snap-to-Grid) и On-Demand экспорт.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from datetime import datetime
|
||||||
|
from typing import Optional, Tuple
|
||||||
|
from core.connection import get_connection
|
||||||
|
from services.scud_export import run_export
|
||||||
|
|
||||||
|
logger = logging.getLogger("SNAPSHOT_FINDER")
|
||||||
|
|
||||||
|
|
||||||
|
def find_or_create_snapshot_for_time(
|
||||||
|
target_date_str: str,
|
||||||
|
target_time_str: str,
|
||||||
|
tolerance_minutes: int = 20,
|
||||||
|
allow_ondemand_export: bool = True
|
||||||
|
) -> Tuple[Optional[str], str]:
|
||||||
|
"""
|
||||||
|
Ищет ближайший срез за указанную дату и время (±tolerance_minutes).
|
||||||
|
Если не найден и allow_ondemand_export=True — запрашивает выгрузку из MS SQL на это время.
|
||||||
|
|
||||||
|
Возвращает: (snapshot_id, human_message)
|
||||||
|
"""
|
||||||
|
date_clean = target_date_str.replace('_', '.')
|
||||||
|
dt_target = datetime.strptime(f"{date_clean} {target_time_str}", "%d.%m.%Y %H:%M")
|
||||||
|
|
||||||
|
# 1. Поиск существующих снапшотов за эту дату в SQLite
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT DISTINCT snapshot_id, snapshot_time
|
||||||
|
FROM scud_logs
|
||||||
|
WHERE log_date = ? AND snapshot_time IS NOT NULL
|
||||||
|
""", (date_clean,))
|
||||||
|
rows = cursor.fetchall()
|
||||||
|
|
||||||
|
best_snapshot = None
|
||||||
|
min_diff_seconds = float('inf')
|
||||||
|
|
||||||
|
for snap_id, snap_time_str in rows:
|
||||||
|
try:
|
||||||
|
# Формат в базе: YYYY-MM-DD HH:MM:SS или DD.MM.YYYY HH:MM:SS
|
||||||
|
raw_time = str(snap_time_str).strip()
|
||||||
|
if '.' in raw_time.split()[0]:
|
||||||
|
dt_snap = datetime.strptime(raw_time, "%d.%m.%Y %H:%M:%S")
|
||||||
|
else:
|
||||||
|
dt_snap = datetime.strptime(raw_time, "%Y-%m-%d %H:%M:%S")
|
||||||
|
|
||||||
|
diff = abs((dt_snap - dt_target).total_seconds())
|
||||||
|
if diff < min_diff_seconds:
|
||||||
|
min_diff_seconds = diff
|
||||||
|
best_snapshot = (snap_id, snap_time_str, dt_snap)
|
||||||
|
except Exception:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Если найден срез в пределах допуска (по умолчанию 20 минут)
|
||||||
|
if best_snapshot and min_diff_seconds <= (tolerance_minutes * 60):
|
||||||
|
snap_id, snap_time, dt_s = best_snapshot
|
||||||
|
diff_mins = round(min_diff_seconds / 60)
|
||||||
|
return snap_id, f"Использован готовый срез {snap_id} за {dt_s.strftime('%H:%M')} (разница {diff_mins} мин)."
|
||||||
|
|
||||||
|
# 2. Если срез не найден и разрешен On-Demand экспорт из MS SQL Орион
|
||||||
|
if allow_ondemand_export:
|
||||||
|
logger.info(f"Снапшот на {date_clean} {target_time_str} не найден в SQLite. Запуск прямого среза из MS SQL...")
|
||||||
|
|
||||||
|
run_export(
|
||||||
|
input_date=date_clean,
|
||||||
|
debug=False,
|
||||||
|
save_xlsx=True
|
||||||
|
)
|
||||||
|
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT snapshot_id
|
||||||
|
FROM scud_logs
|
||||||
|
WHERE log_date = ?
|
||||||
|
ORDER BY id DESC LIMIT 1
|
||||||
|
""", (date_clean,))
|
||||||
|
row = cursor.fetchone()
|
||||||
|
if row:
|
||||||
|
return row[0], f"Создан новый срез {row[0]} из MS SQL на {target_time_str}."
|
||||||
|
|
||||||
|
return None, f"Срез на {date_clean} {target_time_str} не найден."
|
||||||
@@ -0,0 +1,69 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/snapshots/service.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: services / snapshots
|
||||||
|
ROLE: Бизнес-логика срезов СКУД (выборка, валидация Y-срезов, удаление).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[SNAPSHOT_GET_REGISTRY]: Выборка срезов с разметкой защищенных Y-снапшотов.
|
||||||
|
- ANCHOR[SNAPSHOT_DELETE_SAFE]: Безопасное удаление дневных срезов с защитой итоговых.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Dict, Any, Optional, List
|
||||||
|
from core.connection import get_connection
|
||||||
|
from core.repositories.scud_repo import get_available_snapshots, delete_snapshot_by_id
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[SNAPSHOT_GET_REGISTRY]
|
||||||
|
def get_snapshots_registry(date_str: Optional[str] = None) -> Dict[str, Any]:
|
||||||
|
"""Возвращает реестр снапшотов за дату или за все доступные дни."""
|
||||||
|
clean_date = date_str.strip() if date_str else ""
|
||||||
|
rows = get_available_snapshots(date_str=clean_date if clean_date else None)
|
||||||
|
|
||||||
|
snapshots = [
|
||||||
|
{
|
||||||
|
"snapshot_id": r[0],
|
||||||
|
"log_date": r[1],
|
||||||
|
"snapshot_time": r[2],
|
||||||
|
"record_count": r[3],
|
||||||
|
"is_final": str(r[0]).startswith("Y")
|
||||||
|
}
|
||||||
|
for r in rows
|
||||||
|
]
|
||||||
|
|
||||||
|
return {
|
||||||
|
"query_date": clean_date or "все",
|
||||||
|
"snapshots_count": len(snapshots),
|
||||||
|
"snapshots": snapshots
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[SNAPSHOT_DELETE_SAFE]
|
||||||
|
def delete_snapshots_safely(snapshot_ids: List[str]) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
Удаляет выбранные дневные снапшоты.
|
||||||
|
Итоговые вечерние срезы с префиксом 'Y' гарантированно защищены от удаления.
|
||||||
|
"""
|
||||||
|
if not snapshot_ids:
|
||||||
|
return {"status": "error", "message": "Не указаны ID снапшотов для удаления."}
|
||||||
|
|
||||||
|
safe_ids = [str(s).strip() for s in snapshot_ids if s and not str(s).strip().startswith("Y")]
|
||||||
|
|
||||||
|
if not safe_ids:
|
||||||
|
return {"status": "error", "message": "⚠️ Итоговый срез Y защищен от удаления. Выберите дневные снапшоты."}
|
||||||
|
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
placeholders = ",".join(["?"] * len(safe_ids))
|
||||||
|
cursor.execute(f"DELETE FROM scud_logs WHERE snapshot_id IN ({placeholders})", safe_ids)
|
||||||
|
deleted_count = cursor.rowcount
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
return {
|
||||||
|
"status": "success",
|
||||||
|
"deleted_count": deleted_count,
|
||||||
|
"deleted_ids": safe_ids,
|
||||||
|
"message": f"Успешно удалено снапшотов: {len(safe_ids)} шт."
|
||||||
|
}
|
||||||
@@ -0,0 +1,101 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/tasks/exporter.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: services / tasks
|
||||||
|
ROLE: Экспорт бэклога задач в форматированный Markdown файл (ROADMAP).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[TASK_EXPORT_MARKDOWN]: Построение структуры Markdown с чекбоксами.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import uuid
|
||||||
|
import logging
|
||||||
|
from datetime import datetime
|
||||||
|
from typing import Dict, Any, Optional, List
|
||||||
|
from config import OUTPUT_DIR
|
||||||
|
from .repository import repo_get_tasks
|
||||||
|
|
||||||
|
logger = logging.getLogger("TASK_EXPORTER")
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[TASK_EXPORT_MARKDOWN]
|
||||||
|
def export_tasks_to_markdown(user_id: int, filename: Optional[str] = None, status_filter: Optional[str] = None) -> Dict[str, Any]:
|
||||||
|
"""Генерирует Markdown-отчет задач с сохранением в output/web/tasks_export/{uuid}/."""
|
||||||
|
tasks = repo_get_tasks(user_id)
|
||||||
|
if not tasks:
|
||||||
|
return {"status": "error", "message": "Список задач пуст, экспорт отменен"}
|
||||||
|
|
||||||
|
# 1. Фильтрация задач по статусу
|
||||||
|
if status_filter and status_filter.upper() != "ALL":
|
||||||
|
tgt = status_filter.upper()
|
||||||
|
if tgt in ["COMPLETED", "DONE", "ВЫПОЛНЕННЫЕ"]:
|
||||||
|
tasks = [t for t in tasks if str(t.get("status", "")).upper() in ["COMPLETED", "DONE"]]
|
||||||
|
elif tgt in ["IN_PROGRESS", "PROGRESS", "В РАБОТЕ"]:
|
||||||
|
tasks = [t for t in tasks if str(t.get("status", "")).upper() in ["IN_PROGRESS", "PROGRESS"]]
|
||||||
|
elif tgt in ["BACKLOG", "PLANNED", "В ПЛАНАХ"]:
|
||||||
|
tasks = [t for t in tasks if str(t.get("status", "")).upper() in ["BACKLOG", "PLANNED"]]
|
||||||
|
|
||||||
|
if not tasks:
|
||||||
|
return {"status": "error", "message": f"Нет задач с фильтром '{status_filter}' для экспорта"}
|
||||||
|
|
||||||
|
target_filename = filename.strip() if (filename and filename.strip()) else "ROADMAP.md"
|
||||||
|
if not target_filename.endswith(".md"):
|
||||||
|
target_filename = f"{target_filename}.md"
|
||||||
|
|
||||||
|
now_str = datetime.now().strftime("%Y-%m-%d %H:%M")
|
||||||
|
|
||||||
|
# 2. Группировка по модулям
|
||||||
|
modules: Dict[str, List[Dict[str, Any]]] = {}
|
||||||
|
for t in tasks:
|
||||||
|
mod = t.get("module") or "general"
|
||||||
|
modules.setdefault(mod, []).append(t)
|
||||||
|
|
||||||
|
md_lines = [
|
||||||
|
"# 🗺️ Дорожная карта задач проекта (ROADMAP)\n",
|
||||||
|
f"> **Сформировано:** {now_str} | **Всего задач:** {len(tasks)}\n",
|
||||||
|
"---\n"
|
||||||
|
]
|
||||||
|
|
||||||
|
for mod_name, mod_tasks in sorted(modules.items()):
|
||||||
|
md_lines.append(f"## Модуль `{mod_name}`\n")
|
||||||
|
for t in sorted(mod_tasks, key=lambda x: x.get("id", 0)):
|
||||||
|
status = str(t.get("status", "BACKLOG")).upper()
|
||||||
|
is_done = status in ["COMPLETED", "DONE"]
|
||||||
|
is_progress = status in ["IN_PROGRESS", "PROGRESS"]
|
||||||
|
|
||||||
|
check_box = "[x]" if is_done else "[ ]"
|
||||||
|
t_id = t.get("id")
|
||||||
|
title = t.get("title", "Без названия")
|
||||||
|
prio = t.get("priority", "MEDIUM")
|
||||||
|
due = f" *(срок: {t['due_date']})*" if t.get("due_date") else ""
|
||||||
|
status_tag = " `[В РАБОТЕ]`" if is_progress else (" `[ЗАВЕРШЕНО]`" if is_done else "")
|
||||||
|
|
||||||
|
md_lines.append(f"- {check_box} **#{t_id}** [{prio}]{status_tag} {title}{due}")
|
||||||
|
|
||||||
|
md_lines.append("\n---\n")
|
||||||
|
|
||||||
|
content = "\n".join(md_lines)
|
||||||
|
|
||||||
|
# 3. Сохранение файла в изолированную сессионную папку
|
||||||
|
tool_dir = os.path.join(OUTPUT_DIR, "web", "tasks_export")
|
||||||
|
os.makedirs(tool_dir, exist_ok=True)
|
||||||
|
|
||||||
|
session_token = uuid.uuid4().hex[:8]
|
||||||
|
session_dir = os.path.join(tool_dir, session_token)
|
||||||
|
os.makedirs(session_dir, exist_ok=True)
|
||||||
|
|
||||||
|
filepath = os.path.join(session_dir, target_filename)
|
||||||
|
with open(filepath, "w", encoding="utf-8") as f:
|
||||||
|
f.write(content)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"status": "success",
|
||||||
|
"filename": target_filename,
|
||||||
|
"filepath": filepath,
|
||||||
|
"download_url": f"/api/v1/files/download/tasks_export/{session_token}/{target_filename}",
|
||||||
|
"tasks_count": len(tasks),
|
||||||
|
"message": f"Отчет успешно сформирован в файл `{target_filename}` (всего задач: {len(tasks)})."
|
||||||
|
}
|
||||||
@@ -0,0 +1,155 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/tasks/repository.py
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
import re
|
||||||
|
from typing import List, Dict, Any, Optional
|
||||||
|
from core.connection import get_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", "").replace("#", "")
|
||||||
|
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 repo_get_tasks(user_id: int, status: Optional[str] = None) -> List[Dict[str, Any]]:
|
||||||
|
with get_connection(row_factory=True) as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
if status and status.upper() != "ALL":
|
||||||
|
target_status = status.upper()
|
||||||
|
if target_status in ["PROGRESS", "В РАБОТЕ"]:
|
||||||
|
target_status = "IN_PROGRESS"
|
||||||
|
elif target_status in ["DONE", "ГОТОВО"]:
|
||||||
|
target_status = "COMPLETED"
|
||||||
|
elif target_status in ["PLANNED", "ПЛАНЫ"]:
|
||||||
|
target_status = "BACKLOG"
|
||||||
|
|
||||||
|
cursor.execute("""
|
||||||
|
SELECT id, task_id, module, title, priority, status, due_date, created_at
|
||||||
|
FROM tasks
|
||||||
|
WHERE user_id = ? AND (status = ? OR (status = 'BACKLOG' AND ? = 'PLANNED'))
|
||||||
|
ORDER BY id DESC
|
||||||
|
""", (user_id, target_status, target_status))
|
||||||
|
else:
|
||||||
|
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()
|
||||||
|
return [dict(r) for r in rows]
|
||||||
|
|
||||||
|
|
||||||
|
def repo_add_task(
|
||||||
|
user_id: int,
|
||||||
|
module: str,
|
||||||
|
title: str,
|
||||||
|
priority: str = "MEDIUM",
|
||||||
|
due_date: Optional[str] = None,
|
||||||
|
status: str = "BACKLOG"
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
target_status = status.upper() if status else "BACKLOG"
|
||||||
|
if target_status in ["PROGRESS", "В РАБОТЕ"]:
|
||||||
|
target_status = "IN_PROGRESS"
|
||||||
|
elif target_status in ["DONE", "ГОТОВО"]:
|
||||||
|
target_status = "COMPLETED"
|
||||||
|
elif target_status in ["PLANNED", "ПЛАНЫ", "BACKLOG"]:
|
||||||
|
target_status = "BACKLOG"
|
||||||
|
|
||||||
|
with get_connection() as conn:
|
||||||
|
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 (?, ?, ?, ?, ?, ?, ?)
|
||||||
|
""", (new_task_id, module or "general", title.strip(), priority.upper(), target_status, due_date, user_id))
|
||||||
|
conn.commit()
|
||||||
|
return {"status": "success", "task_id": new_task_id, "id": max_id + 1}
|
||||||
|
|
||||||
|
|
||||||
|
def repo_update_task(
|
||||||
|
user_id: int,
|
||||||
|
task_id: str,
|
||||||
|
title: Optional[str] = None,
|
||||||
|
priority: Optional[str] = None,
|
||||||
|
status: Optional[str] = None,
|
||||||
|
due_date: Optional[str] = None
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
clean_num = re.sub(r'\D', '', str(task_id))
|
||||||
|
formatted_id = normalize_task_id(task_id)
|
||||||
|
|
||||||
|
updates = []
|
||||||
|
params = []
|
||||||
|
|
||||||
|
if title is not None and title.strip():
|
||||||
|
updates.append("title = ?")
|
||||||
|
params.append(title.strip())
|
||||||
|
|
||||||
|
if priority is not None and priority.strip():
|
||||||
|
updates.append("priority = ?")
|
||||||
|
params.append(priority.strip().upper())
|
||||||
|
|
||||||
|
if status is not None and status.strip():
|
||||||
|
target_status = status.strip().upper()
|
||||||
|
if target_status in ["PROGRESS", "В РАБОТЕ"]:
|
||||||
|
target_status = "IN_PROGRESS"
|
||||||
|
elif target_status in ["DONE", "ГОТОВО"]:
|
||||||
|
target_status = "COMPLETED"
|
||||||
|
elif target_status in ["PLANNED", "ПЛАНЫ"]:
|
||||||
|
target_status = "BACKLOG"
|
||||||
|
updates.append("status = ?")
|
||||||
|
params.append(target_status)
|
||||||
|
|
||||||
|
if due_date is not None:
|
||||||
|
updates.append("due_date = ?")
|
||||||
|
params.append(due_date.strip() if due_date.strip() else None)
|
||||||
|
|
||||||
|
if not updates:
|
||||||
|
return {"status": "success", "message": "Нет данных для обновления"}
|
||||||
|
|
||||||
|
params.extend([clean_num, formatted_id, f"%{task_id.strip()}", user_id])
|
||||||
|
sql = f"""
|
||||||
|
UPDATE tasks
|
||||||
|
SET {', '.join(updates)}
|
||||||
|
WHERE (id = ? OR UPPER(task_id) = ? OR task_id LIKE ?) AND user_id = ?
|
||||||
|
"""
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute(sql, params)
|
||||||
|
rows_affected = cursor.rowcount
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
if rows_affected == 0:
|
||||||
|
return {"error": f"Задача {task_id} не найдена или принадлежит другому пользователю"}
|
||||||
|
|
||||||
|
return {"status": "success", "message": f"Задача #{task_id} успешно обновлена"}
|
||||||
|
|
||||||
|
|
||||||
|
def repo_delete_task(user_id: int, task_id: str) -> Dict[str, Any]:
|
||||||
|
clean_num = re.sub(r'\D', '', str(task_id))
|
||||||
|
formatted_id = normalize_task_id(task_id)
|
||||||
|
|
||||||
|
with get_connection() as conn:
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
DELETE FROM tasks
|
||||||
|
WHERE (id = ? OR UPPER(task_id) = ? OR task_id LIKE ?) AND user_id = ?
|
||||||
|
""", (clean_num, formatted_id, f"%{task_id.strip()}", user_id))
|
||||||
|
deleted = cursor.rowcount
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
if deleted == 0:
|
||||||
|
return {"error": f"Задача {task_id} не найдена"}
|
||||||
|
return {"status": "success", "message": f"Задача #{task_id} удалена"}
|
||||||
@@ -0,0 +1,72 @@
|
|||||||
|
"""
|
||||||
|
===============================================================================
|
||||||
|
FILE: services/tasks/service.py
|
||||||
|
PROJECT: SCUD Orion AI (Unified Architecture)
|
||||||
|
MODULE: services / tasks
|
||||||
|
ROLE: Единый доменный сервис задач (бизнес-логика и диспетчер операций).
|
||||||
|
|
||||||
|
AI-CONTEXT-ANCHORS:
|
||||||
|
- ANCHOR[TASK_SERVICE_DISPATCHER]: Маршрутизация действий ADD/UPDATE/DELETE/EXPORT.
|
||||||
|
===============================================================================
|
||||||
|
"""
|
||||||
|
|
||||||
|
from typing import Dict, Any, Optional, List
|
||||||
|
from .repository import repo_get_tasks, repo_add_task, repo_update_task, repo_delete_task
|
||||||
|
from .exporter import export_tasks_to_markdown
|
||||||
|
|
||||||
|
|
||||||
|
def get_tasks(user_id: int, status: Optional[str] = None) -> List[Dict[str, Any]]:
|
||||||
|
"""Получить список задач."""
|
||||||
|
return repo_get_tasks(user_id, status)
|
||||||
|
|
||||||
|
|
||||||
|
def add_task(user_id: int, module: str, title: str, priority: str = "MEDIUM", due_date: Optional[str] = None, status: str = "BACKLOG") -> Dict[str, Any]:
|
||||||
|
"""Создать задачу."""
|
||||||
|
res = repo_add_task(user_id, module, title, priority, due_date, status)
|
||||||
|
return {"status": "success", "task_id": res["task_id"], "message": f"Задача #{res['id']} создана и добавлена в планы"}
|
||||||
|
|
||||||
|
|
||||||
|
def update_task_details(user_id: int, task_id: str, title: Optional[str] = None, priority: Optional[str] = None, status: Optional[str] = None, due_date: Optional[str] = None) -> Dict[str, Any]:
|
||||||
|
"""Обновить задачу."""
|
||||||
|
return repo_update_task(user_id, task_id, title, priority, status, due_date)
|
||||||
|
|
||||||
|
|
||||||
|
def delete_task(user_id: int, task_id: str) -> Dict[str, Any]:
|
||||||
|
"""Удалить задачу."""
|
||||||
|
return repo_delete_task(user_id, task_id)
|
||||||
|
|
||||||
|
|
||||||
|
# ANCHOR[TASK_SERVICE_DISPATCHER]
|
||||||
|
def execute_task_action(
|
||||||
|
user_id: int,
|
||||||
|
action: str,
|
||||||
|
task_id: Optional[str] = None,
|
||||||
|
title: Optional[str] = None,
|
||||||
|
priority: Optional[str] = "MEDIUM",
|
||||||
|
status: Optional[str] = None,
|
||||||
|
module: Optional[str] = "general",
|
||||||
|
due_date: Optional[str] = None,
|
||||||
|
filename: Optional[str] = "ROADMAP.md"
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
"""Консолидированный диспетчер операций над задачами."""
|
||||||
|
act = (action or "").strip().upper()
|
||||||
|
|
||||||
|
if act == "ADD":
|
||||||
|
if not title:
|
||||||
|
return {"status": "error", "message": "Для создания задачи требуется указать title"}
|
||||||
|
return add_task(user_id, module or "general", title, priority or "MEDIUM", due_date, status or "BACKLOG")
|
||||||
|
|
||||||
|
elif act == "UPDATE":
|
||||||
|
if not task_id:
|
||||||
|
return {"status": "error", "message": "Для обновления требуется указать task_id"}
|
||||||
|
return update_task_details(user_id, str(task_id), title, priority, status, due_date)
|
||||||
|
|
||||||
|
elif act == "DELETE":
|
||||||
|
if not task_id:
|
||||||
|
return {"status": "error", "message": "Для удаления требуется указать task_id"}
|
||||||
|
return delete_task(user_id, str(task_id))
|
||||||
|
|
||||||
|
elif act == "EXPORT":
|
||||||
|
return export_tasks_to_markdown(user_id, filename=filename or "ROADMAP.md", status_filter=status)
|
||||||
|
|
||||||
|
return {"status": "error", "message": f"Неизвестное действие action='{action}'"}
|
||||||
+53
-60
@@ -5,85 +5,75 @@ from services.ai_verifier import ask_ollama
|
|||||||
from services.knowledge_base import load_knowledge_base
|
from services.knowledge_base import load_knowledge_base
|
||||||
|
|
||||||
|
|
||||||
def find_python_fuzzy_matches(unexplained_df, raw_absent_df):
|
def find_ai_identity_suggestions(unexplained_df, raw_staff_df):
|
||||||
"""
|
"""
|
||||||
Точный поиск совпадений ФИО силами Python (без галлюцинаций ИИ).
|
⭐️ Теневой ИИ-арбитраж: ищет потенциальные опечатки операторов СКУД
|
||||||
Сравнивает фамилию и полное имя с порогом сходства >= 0.75.
|
среди нераспознанных сотрудников и формирует рекомендации для администратора.
|
||||||
"""
|
"""
|
||||||
if unexplained_df.empty or raw_absent_df is None or raw_absent_df.empty:
|
if unexplained_df.empty or raw_staff_df is None or raw_staff_df.empty:
|
||||||
return []
|
return []
|
||||||
|
|
||||||
absent_dict = {}
|
staff_fios = raw_staff_df['fio_clean'].dropna().unique().tolist() if 'fio_clean' in raw_staff_df.columns else []
|
||||||
for idx, r in raw_absent_df.iterrows():
|
unexp_fios = unexplained_df['fio_clean'].dropna().unique().tolist() if 'fio_clean' in unexplained_df.columns else []
|
||||||
raw_fio = str(r.get('ФИО', ''))
|
|
||||||
clean_fio = str(r.get('fio_clean', ''))
|
|
||||||
reason = str(r.get('Вид_отсутствия', ''))
|
|
||||||
if clean_fio and clean_fio not in ['Ао "Ленморниипроект"', 'Сотрудник', 'Nan', 'None']:
|
|
||||||
absent_dict[clean_fio] = {'raw_fio': raw_fio, 'reason': reason}
|
|
||||||
|
|
||||||
matches = []
|
suggestions = []
|
||||||
unexplained_fios = unexplained_df['fio_clean'].unique() if 'fio_clean' in unexplained_df.columns else []
|
for scud_fio in unexp_fios:
|
||||||
|
f_parts = scud_fio.split()
|
||||||
for fio in unexplained_fios:
|
if not f_parts:
|
||||||
fio_parts = fio.split()
|
|
||||||
if not fio_parts:
|
|
||||||
continue
|
continue
|
||||||
surname = fio_parts[0].lower()
|
surname = f_parts[0].lower()
|
||||||
|
|
||||||
for abs_clean, abs_info in absent_dict.items():
|
for staff_fio in staff_fios:
|
||||||
abs_parts = abs_clean.split()
|
if staff_fio.lower().startswith(surname[:4]):
|
||||||
if not abs_parts:
|
ratio = difflib.SequenceMatcher(None, scud_fio.lower(), staff_fio.lower()).ratio()
|
||||||
continue
|
if 0.75 <= ratio < 1.0:
|
||||||
abs_surname = abs_parts[0].lower()
|
suggestions.append({
|
||||||
|
"scud_fio": scud_fio,
|
||||||
if surname == abs_surname:
|
"suggested_zup_fio": staff_fio,
|
||||||
ratio = difflib.SequenceMatcher(None, fio.lower(), abs_clean.lower()).ratio()
|
"confidence": round(ratio, 2)
|
||||||
if ratio >= 0.75:
|
|
||||||
matches.append({
|
|
||||||
"target_fio": fio,
|
|
||||||
"found_in_1c_raw": abs_info['raw_fio'],
|
|
||||||
"reason_1c": abs_info['reason']
|
|
||||||
})
|
})
|
||||||
break
|
break
|
||||||
|
|
||||||
return matches
|
return suggestions
|
||||||
|
|
||||||
|
|
||||||
def generate_markdown_report(merged_df, absent_explained, absent_unexplained, scud_present_but_absent_in_1c, anomalies_list=None, raw_scud_df=None, raw_staff_df=None, raw_absent_df=None, date_str="05.08.2026"):
|
def generate_markdown_report(merged_df, absent_explained, absent_unexplained, scud_present_but_absent_in_1c, anomalies_list=None, raw_scud_df=None, raw_staff_df=None, raw_absent_df=None, date_str="05.08.2026"):
|
||||||
# 1. Загрузка Базы Знаний
|
|
||||||
kb = load_knowledge_base()
|
kb = load_knowledge_base()
|
||||||
custom_rules = kb.get("rules", [])
|
custom_rules = kb.get("rules", [])
|
||||||
custom_rules_str = "\n".join([f"- {r}" for r in custom_rules]) if custom_rules else "Специфических правил компании пока нет."
|
custom_rules_str = "\n".join([f"- {r}" for r in custom_rules]) if custom_rules else "Специфических правил компании пока нет."
|
||||||
|
|
||||||
# 2. Предварительный расчет ключевых метрик (РЕШАЕТ NameError: total_staff)
|
|
||||||
total_staff = len(merged_df)
|
total_staff = len(merged_df)
|
||||||
is_working_mask = (merged_df['Пришел'] == True) | (
|
present_office_cnt = len(merged_df[merged_df['Пришел'] == True])
|
||||||
merged_df['Вид_отсутствия'].astype(str).str.lower().str.contains('командировк|удален|дистанцион|разъездн', regex=True, na=False)
|
|
||||||
)
|
reason_series = merged_df['Вид_отсутствия'].astype(str).str.lower()
|
||||||
present_cnt = len(merged_df[is_working_mask])
|
is_remote_reason = reason_series.str.contains('удален|дистанцион', regex=True, na=False)
|
||||||
explained_cnt = len(absent_explained)
|
remote_home_cnt = len(merged_df[(merged_df['Пришел'] == False) & is_remote_reason])
|
||||||
|
|
||||||
|
explained_cnt = len(merged_df[(merged_df['Пришел'] == False) & (merged_df['Вид_отсутствия'].notna()) & (~is_remote_reason) & (~merged_df['Вид_отсутствия'].astype(str).str.startswith('Исключение'))])
|
||||||
unexplained_cnt = len(absent_unexplained)
|
unexplained_cnt = len(absent_unexplained)
|
||||||
anomalies_cnt = len(anomalies_list) if anomalies_list else 0
|
anomalies_cnt = len(anomalies_list) if anomalies_list else 0
|
||||||
|
|
||||||
# 3. Точный расчет неточных совпадений через Python
|
# ⭐️ Теневые подсказки ИИ
|
||||||
fio_mismatches_python = find_python_fuzzy_matches(absent_unexplained, raw_absent_df)
|
ai_suggestions = find_ai_identity_suggestions(absent_unexplained, raw_staff_df)
|
||||||
|
|
||||||
|
anomalies_formatted = []
|
||||||
|
if anomalies_list:
|
||||||
|
for idx, a in enumerate(anomalies_list, 1):
|
||||||
|
anomalies_formatted.append(f"{idx}. {a.get('fio', 'Сотрудник')}: {a.get('description', '')}")
|
||||||
|
anomalies_text_block = "\n".join(anomalies_formatted) if anomalies_formatted else "Аномалий не обнаружено."
|
||||||
|
|
||||||
# 4. Формирование списка неизвестных с приоритетом аббревиатуры СКУД
|
|
||||||
unexplained_list = []
|
unexplained_list = []
|
||||||
if not absent_unexplained.empty:
|
if not absent_unexplained.empty:
|
||||||
for idx, r in absent_unexplained.iterrows():
|
for idx, r in absent_unexplained.iterrows():
|
||||||
fio = r.get('Сотрудник', r.get('fio_clean', ''))
|
fio = r.get('Сотрудник', r.get('fio_clean', ''))
|
||||||
|
|
||||||
dept_scud = str(r.get('department_scud', '')).strip()
|
dept_scud = str(r.get('department_scud', '')).strip()
|
||||||
dept_1c = str(r.get('Подразделение', '')).strip()
|
dept_1c = str(r.get('Подразделение', '')).strip()
|
||||||
dept = dept_scud if (dept_scud and dept_scud.lower() != 'nan' and dept_scud != '—') else dept_1c
|
dept = dept_scud if (dept_scud and dept_scud.lower() != 'nan' and dept_scud != '—') else dept_1c
|
||||||
if not dept:
|
if not dept:
|
||||||
dept = '—'
|
dept = '—'
|
||||||
|
|
||||||
pos = r.get('Должность', '—')
|
pos = r.get('Должность', '—')
|
||||||
unexplained_list.append(f"{fio} — {dept}, {pos}")
|
unexplained_list.append(f"{fio} — {dept}, {pos}")
|
||||||
|
|
||||||
# 5. Сборка промпта для Ollama
|
|
||||||
prompt = f"""
|
prompt = f"""
|
||||||
Ты — старший аудитор кадровой безопасности и контроллинга СКУД.
|
Ты — старший аудитор кадровой безопасности и контроллинга СКУД.
|
||||||
Сформируй итоговую сводку кадрового контроля на {date_str}.
|
Сформируй итоговую сводку кадрового контроля на {date_str}.
|
||||||
@@ -93,50 +83,53 @@ def generate_markdown_report(merged_df, absent_explained, absent_unexplained, sc
|
|||||||
|
|
||||||
🚨 ВХОДНЫЕ МЕТРИКИ:
|
🚨 ВХОДНЫЕ МЕТРИКИ:
|
||||||
- Всего сотрудников: {total_staff}
|
- Всего сотрудников: {total_staff}
|
||||||
- Работают (офис / удаленка / командировки): {present_cnt}
|
- Итого на работе (в офисе): {present_office_cnt}
|
||||||
|
- В том числе на удаленной работе: {remote_home_cnt}
|
||||||
- Официально отсутствуют: {explained_cnt}
|
- Официально отсутствуют: {explained_cnt}
|
||||||
- Неизвестно (истинно неотмеченные): {unexplained_cnt}
|
- Неизвестно (истинно неотмеченные): {unexplained_cnt}
|
||||||
- Выявлено аномалий/конфликтов реестров: {anomalies_cnt}
|
- Выявлено аномалий/конфликтов реестров: {anomalies_cnt}
|
||||||
|
|
||||||
---
|
---
|
||||||
🚨 ПОДТВЕРЖДЁННЫЕ АНОМАЛИИ ({anomalies_cnt} шт):
|
🚨 ПОДТВЕРЖДЁННЫЕ АНОМАЛИИ ({anomalies_cnt} шт):
|
||||||
{json.dumps(anomalies_list, ensure_ascii=False, indent=2)}
|
{anomalies_text_block}
|
||||||
|
|
||||||
---
|
---
|
||||||
⚠️ ПОДТВЕРЖДЁННЫЕ PYTHON ОШИБКИ СОПОСТАВЛЕНИЯ ФИО В 1С ({len(fio_mismatches_python)} шт):
|
💡 ПРЕДЛОЖЕНИЯ ИИ ПО СОПОСТАВЛЕНИЮ ФИО ({len(ai_suggestions)} шт):
|
||||||
{json.dumps(fio_mismatches_python, ensure_ascii=False, indent=2)}
|
{json.dumps(ai_suggestions, ensure_ascii=False, indent=2)}
|
||||||
|
|
||||||
---
|
---
|
||||||
📊 СПИСОК НЕИЗВЕСТНЫХ СЛУЧАЕВ ({unexplained_cnt} чел):
|
📊 СПИСОК НЕИЗВЕСТНЫХ СЛУЧАЕВ ({unexplained_cnt} чел):
|
||||||
{json.dumps(unexplained_list, ensure_ascii=False, indent=2)}
|
{json.dumps(unexplained_list, ensure_ascii=False, indent=2)}
|
||||||
|
|
||||||
СТРОГИЕ ИНСТРУКЦИИ ДЛЯ ИИ:
|
СТРОГИЕ ИНСТРУКЦИИ:
|
||||||
1. КАТЕГОРИЧЕСКИ ЗАПРЕЩЕНО придумывать несуществующие совпадения ФИО или приписывать суффиксы "(осн.)". Используй ТОЛЬКО массив `ПОДТВЕРЖДЁННЫЕ PYTHON ОШИБКИ СОПОСТАВЛЕНИЯ ФИО`. Если этот массив пуст, напиши в этом разделе: "Ошибок сопоставления ФИО и неточностей в 1С не обнаружено."
|
1. КАТЕГОРИЧЕСКИ ЗАПРЕЩЕНО выводить JSON! В блоке "Выявленные ИИ аномалии" выведи читаемый нумерованный список строк ровно так, как передано в разделе "ПОДТВЕРЖДЁННЫЕ АНОМАЛИИ".
|
||||||
2. В разделе "Неизвестные случаи" выведи НУМЕРОВАННЫЙ СПИСОК всех {unexplained_cnt} человек ровно в том виде, в котором они переданы выше.
|
2. В разделе "Подозрения на ошибки сопоставления ФИО" выведи рекомендации по сопоставлению из массива "ПРЕДЛОЖЕНИЯ ИИ". Если массив пуст — напиши "Ошибок сопоставления ФИО не обнаружено."
|
||||||
3. В разделе "Рекомендации" дай 2-3 конкретные системные рекомендации для кадровой службы. НЕ ПЕРЕЧИСЛЯЙ конкретные ФИО в тексте рекомендаций.
|
3. В разделе "Неизвестные случаи" выведи нумерованный список всех {unexplained_cnt} человек.
|
||||||
|
4. В разделе "Рекомендации" дай 2-3 системные рекомендации.
|
||||||
|
|
||||||
СТРОГИЙ ШАБЛОН ОТВЕТА:
|
СТРОГИЙ ШАБЛОН ОТВЕТА:
|
||||||
|
|
||||||
**Сводка контроллинга СКУД и 1С:ЗУП на {date_str}**
|
**Сводка контроллинга СКУД и 1С:ЗУП на {date_str}**
|
||||||
|
|
||||||
- Всего офисных сотрудников: **{total_staff}**
|
- Всего офисных сотрудников: **{total_staff}**
|
||||||
- Работают (офис / удаленка / командировки): **{present_cnt}**
|
- Итого на работе (в офисе): **{present_office_cnt}**
|
||||||
|
- В том числе на удаленной работе: **{remote_home_cnt}**
|
||||||
- Официально отсутствуют: **{explained_cnt}**
|
- Официально отсутствуют: **{explained_cnt}**
|
||||||
- Неизвестно (истинно неотмеченные): **{unexplained_cnt}** чел.
|
- Неизвестно (истинно неотмеченные): **{unexplained_cnt}** чел.
|
||||||
- Выявлено аномалий/конфликтов реестров: **{anomalies_cnt}** шт.
|
- Выявлено аномалий/конфликтов реестров: **{anomalies_cnt}** шт.
|
||||||
|
|
||||||
#### 🚨 Выявленные ИИ аномалии и конфликты источников ({anomalies_cnt}):
|
#### 🚨 Выявленные ИИ аномалии и конфликты источников ({anomalies_cnt}):
|
||||||
(Описание аномалий из anomalies_list. Если их нет — "Аномалий не обнаружено.")
|
(Список аномалий или 'Аномалий не обнаружено.')
|
||||||
|
|
||||||
#### ⚠️ Подозрения на ошибки сопоставления ФИО и несоответствия 1С:
|
#### 💡 Подозрения на ошибки сопоставления ФИО и предложения ИИ:
|
||||||
(Выведи данные ИСКЛЮЧИТЕЛЬНО из массива fio_mismatches_python. Если он пуст — "Ошибок сопоставления ФИО и неточностей в 1С не обнаружено.")
|
(Список предложений сопоставления или 'Ошибок сопоставления ФИО не обнаружено.')
|
||||||
|
|
||||||
#### Неизвестные случаи: {unexplained_cnt}
|
#### Неизвестные случаи: {unexplained_cnt}
|
||||||
(Выведи нумерованный список всех {unexplained_cnt} человек)
|
(Нумерованный список всех {unexplained_cnt} человек)
|
||||||
|
|
||||||
### Точечные рекомендации:
|
### Точечные рекомендации:
|
||||||
(2-3 системные рекомендации без указания ФИО сотрудников)
|
(2-3 системные рекомендации)
|
||||||
"""
|
"""
|
||||||
|
|
||||||
sys_prompt = "Ты — русскоязычный кадровый аудитор. Пиши СТРОГО на русском языке по предоставленному Markdown-шаблону. Категорически запрещено выводить JSON или китайские символы."
|
sys_prompt = "Ты — русскоязычный кадровый аудитор. Пиши СТРОГО на русском языке форматированным текстом. Запрещено выводить фигурные скобки JSON."
|
||||||
return ask_ollama(prompt, system_prompt=sys_prompt)
|
return ask_ollama(prompt, system_prompt=sys_prompt)
|
||||||
@@ -87,7 +87,7 @@ def fetch_zup_absences_from_sql(target_date) -> pd.DataFrame:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
conn_str = get_zup_connection_string()
|
conn_str = get_zup_connection_string()
|
||||||
with pyodbc.connect(conn_str, timeout=5) as conn:
|
with pyodbc.connect(conn_str, timeout=30) as conn:
|
||||||
df = pd.read_sql(query, conn, params=[target_date])
|
df = pd.read_sql(query, conn, params=[target_date])
|
||||||
|
|
||||||
if not df.empty:
|
if not df.empty:
|
||||||
|
|||||||
Reference in New Issue
Block a user