Fix mypy errors

This commit is contained in:
Raffson 2022-11-13 11:40:47 +01:00
parent 2270300177
commit e71ff17a23
No known key found for this signature in database
GPG Key ID: B0402B2C9B764D99
2 changed files with 9 additions and 6 deletions

View File

@ -92,6 +92,7 @@ class SweepFlightPlan(LoiterFlightPlan):
GroundSpeed.for_flight(self.flight, self.layout.hold.alt), GroundSpeed.for_flight(self.flight, self.layout.hold.alt),
) )
@property
def mission_departure_time(self) -> timedelta: def mission_departure_time(self) -> timedelta:
return self.sweep_end_time return self.sweep_end_time

View File

@ -1,7 +1,5 @@
import random import random
from typing import List, Optional from typing import List
Optional
from dcs.point import MovingPoint from dcs.point import MovingPoint
from dcs.task import ( from dcs.task import (
@ -65,8 +63,8 @@ class JoinPointBuilder(PydcsWaypointBuilder):
self, self,
waypoint: MovingPoint, waypoint: MovingPoint,
target_types: List[str], target_types: List[str],
max_dist: Optional[float] = 30.0, max_dist: float = 30.0,
vertical_spacing: Optional[float] = 2000, vertical_spacing: float = 2000.0,
) -> None: ) -> None:
rx = (random.random() + 0.1) * 1000 rx = (random.random() + 0.1) * 1000
@ -76,9 +74,13 @@ class JoinPointBuilder(PydcsWaypointBuilder):
lastwpt = 6 if self.package.primary_task == FlightType.STRIKE else 5 lastwpt = 6 if self.package.primary_task == FlightType.STRIKE else 5
group_id = None
if self.package.primary_flight is not None:
group_id = self.package.primary_flight.group_id
waypoint.tasks.append( waypoint.tasks.append(
EscortTaskAction( EscortTaskAction(
group_id=self.package.primary_flight.group_id, group_id=group_id,
engagement_max_dist=int(nautical_miles(max_dist).meters), engagement_max_dist=int(nautical_miles(max_dist).meters),
lastwpt=lastwpt, lastwpt=lastwpt,
targets=target_types, targets=target_types,