azpsen/tailfin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change update_flight return value to be consistent with others

Browse Source
This commit is contained in:
april 2024-01-05 15:41:33 -06:00
parent d61b473869
commit 403ce0d9bc
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/database/flights.py
View File

@ -71,7 +71,10 @@ async def update_flight(body: FlightCreateSchema, id: str) -> FlightDisplaySchem
raise HTTPException(404, "Flight not found") raise HTTPException(404, "Flight not found")
updated_flight = await flight_collection.update_one({"_id": ObjectId(id)}, {"$set": body.model_dump()}) updated_flight = await flight_collection.update_one({"_id": ObjectId(id)}, {"$set": body.model_dump()})
return updated_flight.upserted_id if updated_flight is None:
raise HTTPException(500, "Failed to update flight")
return id
async def delete_flight(id: str) -> FlightDisplaySchema: async def delete_flight(id: str) -> FlightDisplaySchema:

6
api/routes/flights.py
View File

@ -100,9 +100,9 @@ async def add_flight(flight_body: FlightCreateSchema, user: UserDisplaySchema =
return {"id": str(flight)} return {"id": str(flight)}
@router.put('/{flight_id}', summary="Update the given flight with new information", status_code=201) @router.put('/{flight_id}', summary="Update the given flight with new information", status_code=200)
async def update_flight(flight_id: str, flight_body: FlightCreateSchema, async def update_flight(flight_id: str, flight_body: FlightCreateSchema,
user: UserDisplaySchema = Depends(get_current_user)) -> str: user: UserDisplaySchema = Depends(get_current_user)) -> dict:
""" """
Update the given flight with new information Update the given flight with new information
@ -121,7 +121,7 @@ async def update_flight(flight_id: str, flight_body: FlightCreateSchema,
updated_flight_id = await db.update_flight(flight_body, flight_id) updated_flight_id = await db.update_flight(flight_body, flight_id)
return str(updated_flight_id) return {"id": str(updated_flight_id)}
@router.delete('/{flight_id}', summary="Delete the given flight", status_code=200, response_model=FlightDisplaySchema) @router.delete('/{flight_id}', summary="Delete the given flight", status_code=200, response_model=FlightDisplaySchema)