From c2b066d371d738f6a0fd5e41423e9f6ccc876b87 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Michal=20Kv=C3=A1=C4=8Dek?= Date: Wed, 17 Apr 2024 13:17:59 +0200 Subject: [PATCH] Fix nacitani trasy pro plan jen s jednim bodem --- src/graphql_schema/entities/types/types.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/graphql_schema/entities/types/types.py b/src/graphql_schema/entities/types/types.py index e6db576..f60ef8a 100644 --- a/src/graphql_schema/entities/types/types.py +++ b/src/graphql_schema/entities/types/types.py @@ -190,16 +190,20 @@ class FlightPlanMarker: @strawberry.type class FlightPlanTrack: - bearing: int - distance: float + bearing: Optional[int] + distance: Optional[float] from_: FlightPlanMarker = strawberry.field(name="from") - to: FlightPlanMarker + to: Optional[FlightPlanMarker] @strawberry_sqlalchemy_type(model=models.FlightPlan) class FlightPlan: async def resolve_track(self): markers = await self.markers() + + if len(markers) == 1: + return [FlightPlanTrack(bearing=None, distance=None, from_=markers[0], to=None)] + navigation = [] for i in range(len(markers) - 1): from_ = markers[i]