Fix mypy issue.

This commit is contained in:
Dan Albert 2020-10-27 18:12:13 -07:00
parent fa01303460
commit 1620c602cf

View File

@ -1161,32 +1161,36 @@ class IngressBuilder(PydcsWaypointBuilder):
self.set_waypoint_tot(waypoint, self.timing.ingress) self.set_waypoint_tot(waypoint, self.timing.ingress)
return waypoint return waypoint
class CasIngressBuilder(IngressBuilder): class CasIngressBuilder(IngressBuilder):
def build(self) -> MovingPoint: def build(self) -> MovingPoint:
waypoint = super().build() waypoint = super().build()
try: cas_waypoint = self.flight.waypoint_with_type((FlightWaypointType.CAS,))
cas_waypoint = self.flight.waypoint_with_type((FlightWaypointType.CAS,)) if cas_waypoint is None:
waypoint.add_task(EngageTargetsInZone( logging.error(
position=cas_waypoint.position, "No CAS waypoint found. Falling back to search and engage")
radius=FRONTLINE_LENGTH / 2, waypoint.add_task(EngageTargets(
targets=[ max_distance=nm_to_meter(10),
Targets.All.GroundUnits.GroundVehicles, targets=[
Targets.All.GroundUnits.AirDefence.AAA, Targets.All.GroundUnits.GroundVehicles,
Targets.All.GroundUnits.Infantry, Targets.All.GroundUnits.AirDefence.AAA,
]) Targets.All.GroundUnits.Infantry,
])
) )
except AttributeError: else:
logging.exception('Unable to create CAS target zone. Falling back to search and engage') waypoint.add_task(EngageTargetsInZone(
waypoint.add_task(EngageTargets(max_distance=nm_to_meter(10), position=cas_waypoint.position,
targets=[ radius=FRONTLINE_LENGTH / 2,
Targets.All.GroundUnits.GroundVehicles, targets=[
Targets.All.GroundUnits.AirDefence.AAA, Targets.All.GroundUnits.GroundVehicles,
Targets.All.GroundUnits.Infantry, Targets.All.GroundUnits.AirDefence.AAA,
]) Targets.All.GroundUnits.Infantry,
])
) )
waypoint.add_task(OptROE(OptROE.Values.OpenFireWeaponFree)) waypoint.add_task(OptROE(OptROE.Values.OpenFireWeaponFree))
return waypoint return waypoint
class SeadIngressBuilder(IngressBuilder): class SeadIngressBuilder(IngressBuilder):
def build(self) -> MovingPoint: def build(self) -> MovingPoint:
waypoint = super().build() waypoint = super().build()