From 365a1fc1b06cf73762db4b6d512136dd077f6065 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Michal=20Kv=C3=A1=C4=8Dek?= Date: Thu, 29 Aug 2024 06:25:26 +0200 Subject: [PATCH] Fix nastaveni titulni fotky --- .../entities/resolvers/flight.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/src/graphql_schema/entities/resolvers/flight.py b/src/graphql_schema/entities/resolvers/flight.py index cd18fbd..9a4a4e1 100644 --- a/src/graphql_schema/entities/resolvers/flight.py +++ b/src/graphql_schema/entities/resolvers/flight.py @@ -179,14 +179,17 @@ class FlightMutationResolver(BaseMutationResolver): def schedule_background_tasks(flight_id: int, flight_data: dict, context) -> None: context.background_tasks.add_task(add_terrain_elevation_to_flight, flight_id=flight_id) - context.background_tasks.add_task( - download_weather_for_flight, flight_id=id, airport_id=flight_data['takeoff_airport_id'], - date_time=flight_data['takeoff_datetime'], type_="takeoff" - ) - context.background_tasks.add_task( - download_weather_for_flight, flight_id=id, airport_id=flight_data['landing_airport_id'], - date_time=flight_data['landing_datetime'], type_="landing" - ) + if flight_data.get("takeoff_airport_id"): + context.background_tasks.add_task( + download_weather_for_flight, flight_id=id, airport_id=flight_data['takeoff_airport_id'], + date_time=flight_data['takeoff_datetime'], type_="takeoff" + ) + + if flight_data.get("landing_airport_id"): + context.background_tasks.add_task( + download_weather_for_flight, flight_id=id, airport_id=flight_data['landing_airport_id'], + date_time=flight_data['landing_datetime'], type_="landing" + ) async def handle_track_edit(db: AsyncSession, flight_id: int, track: list[TrackItemInput], user_id: int):