Fix nacitani trasy pro plan jen s jednim bodem

This commit is contained in:
Michal Kváček
2024-04-17 13:17:59 +02:00
parent cee5de003f
commit c2b066d371
+7 -3
View File
@@ -190,16 +190,20 @@ class FlightPlanMarker:
@strawberry.type @strawberry.type
class FlightPlanTrack: class FlightPlanTrack:
bearing: int bearing: Optional[int]
distance: float distance: Optional[float]
from_: FlightPlanMarker = strawberry.field(name="from") from_: FlightPlanMarker = strawberry.field(name="from")
to: FlightPlanMarker to: Optional[FlightPlanMarker]
@strawberry_sqlalchemy_type(model=models.FlightPlan) @strawberry_sqlalchemy_type(model=models.FlightPlan)
class FlightPlan: class FlightPlan:
async def resolve_track(self): async def resolve_track(self):
markers = await self.markers() markers = await self.markers()
if len(markers) == 1:
return [FlightPlanTrack(bearing=None, distance=None, from_=markers[0], to=None)]
navigation = [] navigation = []
for i in range(len(markers) - 1): for i in range(len(markers) - 1):
from_ = markers[i] from_ = markers[i]