From b65805aed5094f2108e675a2f7dce39f4dfd09ef Mon Sep 17 00:00:00 2001 From: april Date: Mon, 15 Jan 2024 17:10:13 -0600 Subject: [PATCH] Delete associated images when deleting flight log --- api/database/flights.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/api/database/flights.py b/api/database/flights.py index d3c319b..592649b 100644 --- a/api/database/flights.py +++ b/api/database/flights.py @@ -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))