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