Don't apply 404 model to all endpoints

spoolman/api/v1/filament.py

@@ -137,6 +137,7 @@ async def find(
     name="Get filament",
     description="Get a specific filament.",
     response_model_exclude_none=True,
+    responses={404: {"model": Message}},
 )
 async def get(
     db: Annotated[AsyncSession, Depends(get_db_session)],
@@ -181,6 +182,7 @@ async def create(
     name="Update filament",
     description="Update any attribute of a filament. Only fields specified in the request will be affected.",
     response_model_exclude_none=True,
+    responses={404: {"model": Message}},
 )
 async def update(
     db: Annotated[AsyncSession, Depends(get_db_session)],
@@ -210,6 +212,7 @@ async def update(
     response_model=Message,
     responses={
         403: {"model": Message},
+        404: {"model": Message},
     },
 )
 async def delete(  # noqa: ANN201

spoolman/api/v1/router.py

@@ -20,7 +20,6 @@ app = FastAPI(
     title="Spoolman REST API v1",
     version="1.0.0",
     root_path_in_servers=False,
-    responses={404: {"model": models.Message}},
 )
 
 

spoolman/api/v1/spool.py

@@ -123,6 +123,7 @@ async def find(
     name="Get spool",
     description="Get a specific spool.",
     response_model_exclude_none=True,
+    responses={404: {"model": Message}},
 )
 async def get(
     db: Annotated[AsyncSession, Depends(get_db_session)],
@@ -189,6 +190,7 @@ async def create(  # noqa: ANN201
     response_model=Spool,
     responses={
         400: {"model": Message},
+        404: {"model": Message},
     },
 )
 async def update(  # noqa: ANN201
@@ -229,6 +231,7 @@ async def update(  # noqa: ANN201
     "/{spool_id}",
     name="Delete spool",
     description="Delete a spool.",
+    responses={404: {"model": Message}},
 )
 async def delete(
     db: Annotated[AsyncSession, Depends(get_db_session)],
@@ -248,6 +251,7 @@ async def delete(
     response_model=Spool,
     responses={
         400: {"model": Message},
+        404: {"model": Message},
     },
 )
 async def use(  # noqa: ANN201

spoolman/api/v1/vendor.py

@@ -64,6 +64,7 @@ async def find(
     name="Get vendor",
     description="Get a specific vendor.",
     response_model_exclude_none=True,
+    responses={404: {"model": Message}},
 )
 async def get(
     db: Annotated[AsyncSession, Depends(get_db_session)],
@@ -97,6 +98,7 @@ async def create(
     name="Update vendor",
     description="Update any attribute of a vendor. Only fields specified in the request will be affected.",
     response_model_exclude_none=True,
+    responses={404: {"model": Message}},
 )
 async def update(
     db: Annotated[AsyncSession, Depends(get_db_session)],
@@ -123,6 +125,7 @@ async def update(
     description=(
         "Delete a vendor. The vendor attribute of any filaments who refer to the deleted vendor will be cleared."
     ),
+    responses={404: {"model": Message}},
 )
 async def delete(
     db: Annotated[AsyncSession, Depends(get_db_session)],