113 lines
5.3 KiB
Python
113 lines
5.3 KiB
Python
"""Сборка приложения. Роутеры подключаются по мере готовности — см. tasks/."""
|
||
|
||
import asyncio
|
||
import logging
|
||
from contextlib import asynccontextmanager
|
||
|
||
from fastapi import FastAPI
|
||
from sqlalchemy.exc import SQLAlchemyError
|
||
from starlette.middleware.sessions import SessionMiddleware
|
||
|
||
from pathlib import Path
|
||
|
||
from app.api import auth
|
||
from app.api.http import admin as admin_api
|
||
from app.api.http import ekp as ekp_api
|
||
from app.api.http import scenarios as scenarios_api
|
||
from app.api.http import sessions
|
||
from app.api.http import trainees
|
||
from app.api.ws import call as call_ws
|
||
from app.api.ws import control as control_ws
|
||
from app.api.ws import observe as observe_ws
|
||
from app.api.ws import station as station_ws
|
||
from app.config import get_settings
|
||
from app.db.base import get_sessionmaker
|
||
from app.dialog.runtime import get_embedder
|
||
from app.voice.models import get_voice_models
|
||
from app.scenarios import store
|
||
from app.session.hub import hub
|
||
from app.session.journal import DbJournal
|
||
from app.scenarios.loader import ScenarioError
|
||
|
||
|
||
LIBRARY = Path(__file__).resolve().parents[2] / "scenarios"
|
||
|
||
# uvicorn настраивает только собственные логгеры: без этого INFO из модулей
|
||
# приложения («получено N кадров», замеры задержки голоса) молча теряется,
|
||
# а до лога доходят одни предупреждения. Чужие библиотеки — от WARNING.
|
||
logging.basicConfig(level=logging.WARNING, format="%(levelname)-8s %(name)s: %(message)s")
|
||
logging.getLogger("app").setLevel(logging.INFO)
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
# Библиотека проверяется на старте целиком: сломанный сценарий, найденный
|
||
# посреди занятия, — сценарий, которого не должно случиться.
|
||
try:
|
||
loaded = store.load_from_disk(LIBRARY)
|
||
except ScenarioError as exc:
|
||
raise RuntimeError(f"библиотека сценариев не прошла проверку: {exc}") from exc
|
||
app.state.scenarios_loaded = len(loaded)
|
||
|
||
# Утверждённые преподавателем сценарии хранятся в БД и должны переживать
|
||
# перезапуск процесса. При недоступной БД остаётся базовая YAML-библиотека.
|
||
try:
|
||
# БД может ещё подниматься или отсутствовать в unit-тесте. Не задерживаем
|
||
# старт АРМ на минуту ради необязательной пользовательской библиотеки.
|
||
async with asyncio.timeout(1):
|
||
async with get_sessionmaker()() as db:
|
||
app.state.scenarios_loaded += await store.restore_published(db)
|
||
except (SQLAlchemyError, OSError, TimeoutError) as exc:
|
||
logging.getLogger(__name__).warning(
|
||
"не удалось восстановить утверждённые сценарии из БД: %s", exc
|
||
)
|
||
|
||
# Журнал: всё, что не записано, для оценки не существует.
|
||
hub.journal = DbJournal(get_sessionmaker())
|
||
|
||
# Эмбеддинги для слот-автомата — грузятся один раз, до первого занятия.
|
||
app.state.embeddings_ready = get_embedder() is not None
|
||
|
||
# Модели речи: ~5 секунд на старте стенда вместо паузы на первом звонке.
|
||
app.state.models_ready = get_voice_models() is not None
|
||
yield
|
||
|
||
await hub.shutdown()
|
||
for state in list(hub._sessions.values()):
|
||
if state.voice is not None:
|
||
await state.voice.close()
|
||
|
||
|
||
app = FastAPI(title="Учебный симулятор занятия для системы 112", lifespan=lifespan)
|
||
# Сессия ставится до роутеров: роль должна быть известна и на HTTP, и в момент
|
||
# рукопожатия сокета, иначе проверять её в канале будет нечем (app/api/auth.py).
|
||
app.add_middleware(
|
||
SessionMiddleware,
|
||
secret_key=get_settings().session_secret,
|
||
session_cookie="lct_session",
|
||
https_only=False, # учебный класс живёт по http во внутренней сети
|
||
max_age=12 * 60 * 60, # смена занятий, не месяц
|
||
)
|
||
app.include_router(auth.router)
|
||
app.include_router(sessions.router)
|
||
app.include_router(scenarios_api.router)
|
||
app.include_router(ekp_api.router)
|
||
app.include_router(admin_api.router)
|
||
app.include_router(trainees.router)
|
||
app.include_router(call_ws.router)
|
||
app.include_router(observe_ws.router)
|
||
app.include_router(control_ws.router)
|
||
app.include_router(station_ws.router)
|
||
|
||
|
||
@app.get("/api/health")
|
||
async def health() -> dict:
|
||
"""Готовность стенда. Фронт показывает экран «модели прогреваются»."""
|
||
settings = get_settings()
|
||
return {
|
||
"status": "ok",
|
||
"models_ready": getattr(app.state, "models_ready", False),
|
||
"scenarios_loaded": getattr(app.state, "scenarios_loaded", 0),
|
||
"embeddings_ready": getattr(app.state, "embeddings_ready", False),
|
||
"offline": settings.offline,
|
||
}
|