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

Fix aircraft updating

Browse Source
This commit is contained in:
april 2024-01-11 14:54:14 -06:00
parent 03ea9d0235
commit 4122521aa3
2 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
api/database/aircraft.py
View File

@ -68,12 +68,13 @@ async def insert_aircraft(body: AircraftCreateSchema, id: str) -> ObjectId:
return aircraft.inserted_id return aircraft.inserted_id
async def update_aircraft(body: AircraftCreateSchema, id: str) -> AircraftDisplaySchema: async def update_aircraft(body: AircraftCreateSchema, id: str, user: str) -> AircraftDisplaySchema:
""" """
Update given aircraft in the database Update given aircraft in the database
:param body: Updated aircraft data :param body: Updated aircraft data
:param id: ID of aircraft to update :param id: ID of aircraft to update
:param user: ID of updating user
:return: Updated aircraft :return: Updated aircraft
""" """
aircraft = await aircraft_collection.find_one({"_id": ObjectId(id)}) aircraft = await aircraft_collection.find_one({"_id": ObjectId(id)})
@ -81,11 +82,17 @@ async def update_aircraft(body: AircraftCreateSchema, id: str) -> AircraftDispla
if aircraft is None: if aircraft is None:
raise HTTPException(404, "Aircraft not found") raise HTTPException(404, "Aircraft not found")
updated_aircraft = await aircraft_collection.update_one({"_id": ObjectId(id)}, {"$set": body.model_dump()}) updated_aircraft = await aircraft_collection.update_one({"_id": ObjectId(id)},
{"$set": aircraft_add_helper(body.model_dump(), user)})
if updated_aircraft is None: if updated_aircraft is None:
raise HTTPException(500, "Failed to update flight") raise HTTPException(500, "Failed to update aircraft")
return AircraftDisplaySchema(**body.model_dump()) aircraft = await aircraft_collection.find_one({"_id": ObjectId(id)})
if aircraft is None:
raise HTTPException(500, "Failed to fetch updated aircraft")
return AircraftDisplaySchema(**aircraft_display_helper(aircraft))
async def update_aircraft_field(field: str, value: Any, id: str) -> AircraftDisplaySchema: async def update_aircraft_field(field: str, value: Any, id: str) -> AircraftDisplaySchema:

2
api/routes/aircraft.py
View File

@ -139,7 +139,7 @@ async def update_aircraft(aircraft_id: str, aircraft_body: AircraftCreateSchema,
logger.info("Attempted access to unauthorized aircraft by %s", user.username) logger.info("Attempted access to unauthorized aircraft by %s", user.username)
raise HTTPException(403, "Unauthorized access") raise HTTPException(403, "Unauthorized access")
updated_aircraft_id = await db.update_aircraft(aircraft_body, aircraft_id) updated_aircraft_id = await db.update_aircraft(aircraft_body, aircraft_id, user.id)
return {"id": str(updated_aircraft_id)} return {"id": str(updated_aircraft_id)}