"""Router setup for the v1 version of the API.""" # ruff: noqa: D103 import logging from fastapi import FastAPI from fastapi.responses import JSONResponse from starlette.requests import Request from starlette.responses import Response from spoolman.database.database import backup_global_db from spoolman.exceptions import ItemNotFoundError from . import filament, models, spool, vendor logger = logging.getLogger(__name__) app = FastAPI( title="Spoolman REST API v1", version="1.0.0", root_path_in_servers=False, ) @app.exception_handler(ItemNotFoundError) async def itemnotfounderror_exception_handler(_request: Request, exc: ItemNotFoundError) -> Response: logger.debug(exc, exc_info=True) return JSONResponse( status_code=404, content={"message": exc.args[0]}, ) # Add health check endpoint @app.get("/health") async def health() -> models.HealthCheck: """Return a health check.""" return models.HealthCheck(status="healthy") # Add endpoint for triggering a db backup @app.post( "/backup", description="Trigger a database backup. Only applicable for SQLite databases.", response_model=models.BackupResponse, responses={500: {"model": models.Message}}, ) async def backup(): # noqa: ANN201 """Trigger a database backup.""" path = await backup_global_db() if path is None: return JSONResponse( status_code=500, content={"message": "Backup failed. See server logs for more information."}, ) return models.BackupResponse(path=str(path)) # Add routers app.include_router(filament.router) app.include_router(spool.router) app.include_router(vendor.router)