diff --git a/spoolman/api/v1/filament.py b/spoolman/api/v1/filament.py index c026b06..e83df22 100644 --- a/spoolman/api/v1/filament.py +++ b/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 diff --git a/spoolman/api/v1/router.py b/spoolman/api/v1/router.py index fc25f1e..da2aa2e 100644 --- a/spoolman/api/v1/router.py +++ b/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}}, ) diff --git a/spoolman/api/v1/spool.py b/spoolman/api/v1/spool.py index 28434a3..4a3cabb 100644 --- a/spoolman/api/v1/spool.py +++ b/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 diff --git a/spoolman/api/v1/vendor.py b/spoolman/api/v1/vendor.py index efaef3b..c09675f 100644 --- a/spoolman/api/v1/vendor.py +++ b/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)],