Dan Albert fa8c0d9660 Clean up flight plan code.
Split the oversized file into one per plan type. This also moves the
layout responsibility out of the oversized FlightPlanBuilder and into
each flight plan type file.
2022-03-09 02:15:07 -08:00

32 lines
1003 B
Python

from __future__ import annotations
import logging
from typing import Type
from game.theater import Airfield
from .formationattack import FormationAttackBuilder, FormationAttackFlightPlan
from .invalidobjectivelocation import InvalidObjectiveLocation
from ..flightwaypointtype import FlightWaypointType
class OcaAircraftFlightPlan(FormationAttackFlightPlan):
@staticmethod
def builder_type() -> Type[Builder]:
return Builder
class Builder(FormationAttackBuilder[OcaAircraftFlightPlan]):
def build(self) -> FormationAttackFlightPlan:
location = self.package.target
if not isinstance(location, Airfield):
logging.exception(
f"Invalid Objective Location for OCA/Aircraft flight "
f"{self.flight=} at {location=}."
)
raise InvalidObjectiveLocation(self.flight.flight_type, location)
return self._build(
OcaAircraftFlightPlan, FlightWaypointType.INGRESS_OCA_AIRCRAFT
)