update response object

HHS · Dec 18, 2024 · c051f3a · c051f3a
1 parent 61eddce
commit c051f3a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/bloom_nofos/nofos/api/api.py b/bloom_nofos/nofos/api/api.py
@@ -41,11 +41,10 @@ def create_nofo(request, payload: NofoSchema):
         _build_nofo(nofo, sections)
         nofo.save()
 
-        return 201, {
-            "message": f"NOFO {nofo.id} imported successfully",
-            "Location": f"/api/nofos/{nofo.id}",
-            "nofo": nofo,
-        }
+        serialized_nofo = NofoSchema.from_orm(nofo)
+        return_response = api.create_response(request, serialized_nofo, status=201)
+        return_response.headers["Location"] = f"/api/nofos/{nofo.id}"
+        return return_response
 
     except ValidationError as e:
         return 400, {"message": "Model validation error", "details": e.message_dict}

diff --git a/bloom_nofos/nofos/api/schemas.py b/bloom_nofos/nofos/api/schemas.py
@@ -76,6 +76,4 @@ class ErrorSchema(Schema):
 
 
 class SuccessSchema(Schema):
-    message: str
-    Location: str
     nofo: NofoSchema