Delete associated images when deleting flight log

This commit is contained in:
april 2024-01-15 17:10:13 -06:00
parent bb2388fac8
commit b65805aed5

View File

@ -13,6 +13,7 @@ from .aircraft import retrieve_aircraft_by_tail, update_aircraft_field
from .db import flight_collection, aircraft_collection
from schemas.flight import FlightConciseSchema, FlightDisplaySchema, FlightCreateSchema, flight_display_helper, \
flight_add_helper, FlightPatchSchema
from .img import delete_image
logger = logging.getLogger("api")
@ -265,5 +266,9 @@ async def delete_flight(id: str) -> FlightDisplaySchema:
if flight is None:
raise HTTPException(404, "Flight not found")
# Delete associated images
for image in flight.images:
await delete_image(image)
await flight_collection.delete_one({"_id": to_objectid(id)})
return FlightDisplaySchema(**flight_display_helper(flight))