fix status_code in FastAPIWrapper (#228)

piccolo-orm · Apr 26, 2023 · 8fb9b45 · 8fb9b45
1 parent 40d4316
commit 8fb9b45
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/piccolo_api/fastapi/endpoints.py b/piccolo_api/fastapi/endpoints.py
@@ -11,7 +11,7 @@
 from enum import Enum
 from inspect import Parameter, Signature, isclass
 
-from fastapi import APIRouter, FastAPI, Request
+from fastapi import APIRouter, FastAPI, Request, status
 from fastapi.params import Query
 from pydantic import BaseModel as PydanticBaseModel
 from pydantic.main import BaseModel
@@ -265,6 +265,7 @@ async def delete(request: Request, **kwargs):
                 path=root_url,
                 endpoint=delete,
                 response_model=None,
+                status_code=status.HTTP_204_NO_CONTENT,
                 methods=["DELETE"],
                 **fastapi_kwargs.get_kwargs("delete"),
             )
@@ -288,6 +289,7 @@ async def post(request: Request, model):
                 path=root_url,
                 endpoint=post,
                 response_model=self.ModelOut,
+                status_code=status.HTTP_201_CREATED,
                 methods=["POST"],
                 **fastapi_kwargs.get_kwargs("post"),
             )
@@ -324,6 +326,7 @@ async def delete_single(row_id: str, request: Request):
                 path=self.join_urls(root_url, "/{row_id:str}/"),
                 endpoint=delete_single,
                 response_model=None,
+                status_code=status.HTTP_204_NO_CONTENT,
                 methods=["DELETE"],
                 **fastapi_kwargs.get_kwargs("delete_single"),
             )
@@ -346,7 +349,8 @@ async def put(row_id: str, request: Request, model):
             fastapi_app.add_api_route(
                 path=self.join_urls(root_url, "/{row_id:str}/"),
                 endpoint=put,
-                response_model=self.ModelOut,
+                response_model=None,
+                status_code=status.HTTP_204_NO_CONTENT,
                 methods=["PUT"],
                 **fastapi_kwargs.get_kwargs("put"),
             )