Fix nacitani letu k eventu

This commit is contained in:
Michal Kváček
2023-12-12 23:54:28 +01:00
parent f4d709d621
commit 3edc1c853f
2 changed files with 5 additions and 0 deletions
+2
View File
@@ -24,6 +24,7 @@ class FlightQueries:
limit: int, limit: int,
offset: int = 0, offset: int = 0,
username: Optional[str] = None, username: Optional[str] = None,
event_id: Optional[int] = None,
public: Optional[bool] = False, public: Optional[bool] = False,
copilot_id: Optional[int] = None, copilot_id: Optional[int] = None,
point_of_interest_id: Optional[int] = None, point_of_interest_id: Optional[int] = None,
@@ -32,6 +33,7 @@ class FlightQueries:
query = FlightQueryResolver().get_query( query = FlightQueryResolver().get_query(
user_id=info.context.user_id, user_id=info.context.user_id,
username=username, username=username,
event_id=event_id,
only_public=public, only_public=public,
copilot_id=copilot_id, copilot_id=copilot_id,
aircraft_id=aircraft_id, aircraft_id=aircraft_id,
@@ -35,6 +35,9 @@ class FlightQueryResolver(BaseQueryResolver):
only_public=only_public only_public=only_public
) )
if kwargs.get("event_id"):
query = query.filter(models.Flight.event_id == kwargs['event_id'])
if kwargs.get("aircraft_id"): if kwargs.get("aircraft_id"):
query = query.filter(models.Flight.aircraft_id == kwargs['aircraft_id']) query = query.filter(models.Flight.aircraft_id == kwargs['aircraft_id'])