Портал технической службы: наряды и чек-листы, задачи, отказы и неисправности, анализ надёжности, карты, склады, приложение для Android

app/db.py

@@ -0,0 +1,53 @@
+import os
+from sqlalchemy import create_engine, event
+from sqlalchemy.orm import sessionmaker, DeclarativeBase
+from . import config, mode
+
+
+def _sqlite_unicode_lower(conn, _):
+    # встроенный lower() в SQLite не знает кириллицу — подменяем, чтобы поиск не зависел от регистра
+    conn.create_function("lower", 1, lambda s: s.lower() if isinstance(s, str) else s)
+
+
+def make_engine(url: str):
+    if url.startswith("sqlite"):
+        os.makedirs(config.DATA_DIR, exist_ok=True)
+        e = create_engine(url, connect_args={"check_same_thread": False})
+        event.listen(e, "connect", _sqlite_unicode_lower)
+        return e
+    return create_engine(url, pool_pre_ping=True)
+
+
+real_engine = make_engine(config.DATABASE_URL)  # боевая БД — с ней работают все, кто не в демо
+engine = real_engine
+SessionLocal = sessionmaker(bind=real_engine, autoflush=False, expire_on_commit=False)  # всегда боевая БД
+
+# демо-БД (персональный демо-режим): открывается только для запросов с демо-токеном
+demo_engine = None
+DemoSession = sessionmaker(autoflush=False, expire_on_commit=False)
+
+
+def set_demo_engine(e):
+    global demo_engine
+    old, demo_engine = demo_engine, e
+    if e is not None:
+        DemoSession.configure(bind=e)
+    return old
+
+
+def session_for(demo: bool):
+    s = DemoSession() if demo and demo_engine is not None else SessionLocal()
+    s.info["demo"] = bool(demo and demo_engine is not None)
+    return s
+
+
+class Base(DeclarativeBase):
+    pass
+
+
+def get_db():
+    db = session_for(mode.is_demo_request())
+    try:
+        yield db
+    finally:
+        db.close()