Pouzor
05c0da53e6
A scan runs on a background thread inside the API process. If that process dies mid-scan — an OOM kill, docker stop, a crash — the ScanRun row stays "running" for ever, because nothing is left alive to finish it. That row is not just cosmetic clutter in Scan History: the trigger endpoints reject a new scan while one is "running" for the same target, so a single kill locks that range out permanently. Nothing can legitimately be "running" the moment we boot, so lifespan() now marks every such row "error" — the same word run_scan and run_device_scan write when they fail themselves — with finished_at and an explanatory message. Reported alongside the OOM itself in #374, which is what produced the orphans. Fixes #374 ha-relevant: maybe
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
import logging
|
|
import logging.config
|
|
from collections.abc import AsyncGenerator
|
|
from contextlib import asynccontextmanager
|
|
from typing import Any
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from starlette.middleware.sessions import SessionMiddleware
|
|
|
|
from app.api.routes import (
|
|
auth,
|
|
canvas,
|
|
designs,
|
|
edges,
|
|
liveview,
|
|
media,
|
|
nodes,
|
|
proxmox,
|
|
racks,
|
|
scan,
|
|
stats,
|
|
status,
|
|
zigbee,
|
|
zwave,
|
|
)
|
|
from app.api.routes import settings as settings_routes
|
|
from app.core.config import settings
|
|
from app.core.scheduler import start_scheduler, stop_scheduler
|
|
from app.core.security import OIDCCSRFMiddleware
|
|
from app.db.database import AsyncSessionLocal, init_db
|
|
from app.services.scanner import reconcile_orphan_runs
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
|
|
# Ensure app logs are visible: attach a handler to the root logger if none
|
|
# exists (uvicorn only installs handlers on its own loggers, not the root).
|
|
root_logger = logging.getLogger()
|
|
if not root_logger.handlers:
|
|
handler = logging.StreamHandler()
|
|
handler.setFormatter(logging.Formatter("%(levelname)s:%(name)s:%(message)s"))
|
|
root_logger.addHandler(handler)
|
|
root_logger.setLevel(logging.INFO)
|
|
logging.getLogger("app").setLevel(logging.INFO)
|
|
logging.getLogger("app.services.scanner").setLevel(logging.INFO)
|
|
await init_db()
|
|
settings.load_overrides()
|
|
# A scan runs on a background thread in this process, so anything still
|
|
# flagged "running" now was orphaned by a previous one that died mid-scan.
|
|
async with AsyncSessionLocal() as db:
|
|
await reconcile_orphan_runs(db)
|
|
start_scheduler()
|
|
yield
|
|
stop_scheduler()
|
|
|
|
|
|
app = FastAPI(
|
|
title="Homelable API",
|
|
version="1.9.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
app.add_middleware(OIDCCSRFMiddleware)
|
|
app.add_middleware(
|
|
SessionMiddleware,
|
|
secret_key=settings.secret_key,
|
|
session_cookie=("__Host-homelable-oidc-flow" if settings.oidc_cookie_secure else "homelable-oidc-flow"),
|
|
max_age=settings.oidc_transaction_expire_seconds,
|
|
same_site="lax",
|
|
https_only=settings.oidc_cookie_secure,
|
|
)
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=settings.cors_origins,
|
|
allow_credentials=True,
|
|
allow_methods=["GET", "POST", "PUT", "PATCH", "DELETE"],
|
|
allow_headers=["Authorization", "Content-Type", "X-Homelable-CSRF", "X-MCP-Service-Key"],
|
|
)
|
|
|
|
app.include_router(auth.router, prefix="/api/v1/auth", tags=["auth"])
|
|
app.include_router(nodes.router, prefix="/api/v1/nodes", tags=["nodes"])
|
|
app.include_router(edges.router, prefix="/api/v1/edges", tags=["edges"])
|
|
app.include_router(canvas.router, prefix="/api/v1/canvas", tags=["canvas"])
|
|
app.include_router(designs.router, prefix="/api/v1/designs", tags=["designs"])
|
|
app.include_router(racks.router, prefix="/api/v1/racks", tags=["racks"])
|
|
app.include_router(scan.router, prefix="/api/v1/scan", tags=["scan"])
|
|
app.include_router(status.router, prefix="/api/v1/status", tags=["status"])
|
|
app.include_router(settings_routes.router, prefix="/api/v1/settings", tags=["settings"])
|
|
app.include_router(liveview.router, prefix="/api/v1/liveview", tags=["liveview"])
|
|
app.include_router(zigbee.router, prefix="/api/v1/zigbee", tags=["zigbee"])
|
|
app.include_router(zwave.router, prefix="/api/v1/zwave", tags=["zwave"])
|
|
app.include_router(proxmox.router, prefix="/api/v1/proxmox", tags=["proxmox"])
|
|
app.include_router(stats.router, prefix="/api/v1/stats", tags=["stats"])
|
|
app.include_router(media.router, prefix="/api/v1/media", tags=["media"])
|
|
|
|
|
|
@app.get("/api/v1/health")
|
|
async def health() -> dict[str, Any]:
|
|
return {"status": "ok"}
|