From 3edc1c853f4f7a3313ad771bb3a11143407dc57b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Michal=20Kv=C3=A1=C4=8Dek?= Date: Tue, 12 Dec 2023 23:54:28 +0100 Subject: [PATCH] Fix nacitani letu k eventu --- src/graphql_schema/entities/flight.py | 2 ++ src/graphql_schema/entities/resolvers/flight.py | 3 +++ 2 files changed, 5 insertions(+) diff --git a/src/graphql_schema/entities/flight.py b/src/graphql_schema/entities/flight.py index af494fa..e5fef59 100644 --- a/src/graphql_schema/entities/flight.py +++ b/src/graphql_schema/entities/flight.py @@ -24,6 +24,7 @@ class FlightQueries: limit: int, offset: int = 0, username: Optional[str] = None, + event_id: Optional[int] = None, public: Optional[bool] = False, copilot_id: Optional[int] = None, point_of_interest_id: Optional[int] = None, @@ -32,6 +33,7 @@ class FlightQueries: query = FlightQueryResolver().get_query( user_id=info.context.user_id, username=username, + event_id=event_id, only_public=public, copilot_id=copilot_id, aircraft_id=aircraft_id, diff --git a/src/graphql_schema/entities/resolvers/flight.py b/src/graphql_schema/entities/resolvers/flight.py index f977a27..6469d1b 100644 --- a/src/graphql_schema/entities/resolvers/flight.py +++ b/src/graphql_schema/entities/resolvers/flight.py @@ -35,6 +35,9 @@ class FlightQueryResolver(BaseQueryResolver): only_public=only_public ) + if kwargs.get("event_id"): + query = query.filter(models.Flight.event_id == kwargs['event_id']) + if kwargs.get("aircraft_id"): query = query.filter(models.Flight.aircraft_id == kwargs['aircraft_id'])