fix(api): run migrations on register_routes so production db has tables
This commit is contained in:
@@ -1,17 +1,37 @@
|
||||
"""HTTP routes for the SPA Manager + extension API."""
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
|
||||
from aiohttp import web
|
||||
|
||||
from engine.db.connection import get_connection
|
||||
from engine.db.migrations import migrate
|
||||
from wp_api import categories as _categories
|
||||
from wp_api import import_export as _import_export
|
||||
from wp_api import modules as _modules
|
||||
from wp_api import spa as _spa
|
||||
from wp_api import test_runner as _test_runner
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def _ensure_db_migrated() -> None:
|
||||
"""Run pending migrations on the configured DB. Idempotent."""
|
||||
conn = get_connection()
|
||||
try:
|
||||
migrate(conn)
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
|
||||
def register_routes(app: web.Application) -> None:
|
||||
"""Mount all /wp + /wp/api/* routes on the given app."""
|
||||
try:
|
||||
_ensure_db_migrated()
|
||||
except Exception: # noqa: BLE001 - never crash ComfyUI on migration failure
|
||||
logger.exception("wildcard-pipeline: db migration failed")
|
||||
|
||||
_modules.register(app.router)
|
||||
_categories.register(app.router)
|
||||
_test_runner.register(app.router)
|
||||
|
||||
Reference in New Issue
Block a user