39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
|
from pathlib import Path
|
||
|
|
|
||
|
|
from sqlmodel import SQLModel, Session, create_engine
|
||
|
|
|
||
|
|
DB_PATH = Path.home() / ".debian-manager" / "data.db"
|
||
|
|
DB_PATH.parent.mkdir(parents=True, exist_ok=True)
|
||
|
|
|
||
|
|
engine = create_engine(
|
||
|
|
f"sqlite:///{DB_PATH}",
|
||
|
|
echo=False,
|
||
|
|
connect_args={"check_same_thread": False},
|
||
|
|
)
|
||
|
|
# expire_on_commit=False — объекты остаются читаемыми после закрытия сессии
|
||
|
|
from sqlalchemy.orm import sessionmaker
|
||
|
|
from sqlmodel import Session as _Session
|
||
|
|
|
||
|
|
_SessionFactory = sessionmaker(bind=engine, class_=_Session, expire_on_commit=False)
|
||
|
|
|
||
|
|
|
||
|
|
def init_db() -> None:
|
||
|
|
from app.db import models # noqa: F401 — регистрация таблиц
|
||
|
|
SQLModel.metadata.create_all(engine)
|
||
|
|
_migrate()
|
||
|
|
|
||
|
|
|
||
|
|
def _migrate() -> None:
|
||
|
|
"""ALTER TABLE для колонок, добавленных после первого запуска."""
|
||
|
|
import sqlite3
|
||
|
|
conn = sqlite3.connect(DB_PATH)
|
||
|
|
cols = [r[1] for r in conn.execute("PRAGMA table_info(server)").fetchall()]
|
||
|
|
if "note" not in cols:
|
||
|
|
conn.execute("ALTER TABLE server ADD COLUMN note TEXT DEFAULT ''")
|
||
|
|
conn.commit()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
|
||
|
|
def get_session() -> Session:
|
||
|
|
return _SessionFactory()
|