80 lines
3.0 KiB
Python
80 lines
3.0 KiB
Python
"""FastAPI application factory and lifecycle.
|
|
|
|
Startup runs migrations and opens the database engine; shutdown disposes it so no
|
|
connection or engine resource leaks. The engine, session factory, and config live
|
|
on ``app.state`` for dependencies to use. The app binds to 127.0.0.1 by default
|
|
and exposes the versioned ``/api/v1`` surface; US01-02 ships only health.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from contextlib import asynccontextmanager
|
|
from pathlib import Path
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.staticfiles import StaticFiles
|
|
|
|
from photo_pipeline.api.routes import (
|
|
albums,
|
|
analysis,
|
|
duplicates,
|
|
health,
|
|
inventory,
|
|
jobs,
|
|
library,
|
|
renames,
|
|
safety,
|
|
thumbnails,
|
|
uploads,
|
|
workflow,
|
|
)
|
|
|
|
# Registers the safety_score / analysis job handlers on import.
|
|
import photo_pipeline.jobs.domain_handlers # noqa: F401
|
|
from photo_pipeline.config import Config
|
|
from photo_pipeline.db import create_db_engine, create_session_factory, run_migrations
|
|
from photo_pipeline.logging import configure_logging
|
|
from photo_pipeline.services.upload_batches import UploadBatchService
|
|
|
|
FRONTEND_DIR = Path(__file__).resolve().parents[2] / "frontend"
|
|
|
|
|
|
def create_app(config: Config | None = None) -> FastAPI:
|
|
config = config or Config.from_env()
|
|
configure_logging(config.log_level, config.log_format)
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
config.database_path.parent.mkdir(parents=True, exist_ok=True)
|
|
run_migrations(config.database_url)
|
|
engine = create_db_engine(config.database_url)
|
|
app.state.config = config
|
|
app.state.engine = engine
|
|
app.state.session_factory = create_session_factory(engine)
|
|
# An upload whose process died left no outcome behind; resolve it now so the
|
|
# uploader lane is free and the uncertain batch is visible (US05-02).
|
|
UploadBatchService(app.state.session_factory, config=config).recover()
|
|
try:
|
|
yield
|
|
finally:
|
|
engine.dispose()
|
|
app.state.engine = None
|
|
|
|
app = FastAPI(title="Photo Pipeline", version="0.1.0", lifespan=lifespan)
|
|
app.include_router(health.router, prefix="/api/v1")
|
|
app.include_router(inventory.router, prefix="/api/v1")
|
|
app.include_router(duplicates.router, prefix="/api/v1")
|
|
app.include_router(jobs.router, prefix="/api/v1")
|
|
app.include_router(thumbnails.router, prefix="/api/v1")
|
|
app.include_router(workflow.router, prefix="/api/v1")
|
|
app.include_router(safety.router, prefix="/api/v1")
|
|
app.include_router(analysis.router, prefix="/api/v1")
|
|
app.include_router(library.router, prefix="/api/v1")
|
|
app.include_router(albums.router, prefix="/api/v1")
|
|
app.include_router(renames.router, prefix="/api/v1")
|
|
app.include_router(uploads.router, prefix="/api/v1")
|
|
# Static single-page app (hash-routed). Mounted last so /api/v1 wins.
|
|
if FRONTEND_DIR.is_dir():
|
|
app.mount("/app", StaticFiles(directory=FRONTEND_DIR, html=True), name="app")
|
|
return app
|