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'])