ayurishchevandClaude Sonnet 5
26dd1f8be4
Глобально уникальные ID (docs/changes/016): - у устройств, групп, задач, резервных копий и записей журнала ID вида <префикс>_<uuid7> (dev_, grp_, job_, bkp_, evt_): типы не пересекаются, внутри типа ID не повторяются и сортируются по времени; - миграция БД v1 заменяет числовые ID с пересчётом ссылок (копия файла БД перед миграцией, сверка числа строк, одна транзакция); - резервная копия = пара файлов с одним bkp_ ID, файлы в S3 получают метаданные backup-id/device-id; синхронизация метаданных с бакетом; - журнал событий в БД (events): создание/изменение/удаление, задачи, бэкапы, смена online/offline, вход в UI; API чтения GET /api/v1/events. Журнал в UI, ротация и очистка (docs/changes/017): - страница «Журнал»: фильтры, подгрузка «Показать ещё», окно записи; - настройки ротации (срок и максимум записей) хранятся в БД (схема v2), ротация при старте, раз в час и после сохранения настроек; - очистка журнала только через окно с паролем пользователя, блокировка после 5 неверных попыток, остаётся запись о факте очистки; через API очистки нет. Тесты: 20 из 20. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
from contextlib import contextmanager
|
|
from pathlib import Path
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import DeclarativeBase, sessionmaker
|
|
|
|
from app.config import get_settings
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
_engine = None
|
|
_SessionLocal = None
|
|
|
|
|
|
def init_db(url: str | None = None) -> None:
|
|
"""Создаёт engine и таблицы. Вызывается при старте приложения (и в тестах)."""
|
|
global _engine, _SessionLocal
|
|
url = url or get_settings().database_url
|
|
if url.startswith("sqlite:///") and ":memory:" not in url:
|
|
Path(url.removeprefix("sqlite:///")).parent.mkdir(parents=True, exist_ok=True)
|
|
_engine = create_engine(url, connect_args={"check_same_thread": False})
|
|
_SessionLocal = sessionmaker(_engine, expire_on_commit=False)
|
|
from app import models # noqa: F401 (регистрация моделей)
|
|
|
|
Base.metadata.create_all(_engine) # у новой БД — все таблицы, у старой — только недостающие (events)
|
|
_migrate()
|
|
|
|
from app import migrations
|
|
|
|
db_file = Path(url.removeprefix("sqlite:///")) if url.startswith("sqlite:///") and ":memory:" not in url else None
|
|
migrations.run(_engine, db_file)
|
|
|
|
|
|
def _migrate() -> None:
|
|
"""create_all не добавляет колонки в существующие таблицы — докидываем вручную."""
|
|
with _engine.begin() as conn:
|
|
cols = {r[1] for r in conn.exec_driver_sql("PRAGMA table_info(devices)")}
|
|
if "use_tls" not in cols:
|
|
conn.exec_driver_sql("ALTER TABLE devices ADD COLUMN use_tls BOOLEAN NOT NULL DEFAULT 1")
|
|
if "group_id" not in cols:
|
|
conn.exec_driver_sql("ALTER TABLE devices ADD COLUMN group_id VARCHAR(40)")
|
|
if "note" not in cols:
|
|
conn.exec_driver_sql("ALTER TABLE devices ADD COLUMN note TEXT")
|
|
|
|
|
|
@contextmanager
|
|
def session_scope():
|
|
"""Короткая транзакция: commit при успехе, rollback при ошибке."""
|
|
session = _SessionLocal()
|
|
try:
|
|
yield session
|
|
session.commit()
|
|
except Exception:
|
|
session.rollback()
|
|
raise
|
|
finally:
|
|
session.close()
|