diff --git a/game/theater/controlpoint.py b/game/theater/controlpoint.py index 6eb8d6d3..04fde9eb 100644 --- a/game/theater/controlpoint.py +++ b/game/theater/controlpoint.py @@ -1388,6 +1388,11 @@ class NavalControlPoint( FlightType.SEAD_ESCORT, ] yield from super().mission_types(for_player) + if self.is_friendly(for_player): + yield from [ + FlightType.AEWC, + FlightType.REFUELING, + ] @property def heading(self) -> Heading: @@ -1486,16 +1491,6 @@ class Carrier(NavalControlPoint): def symbol_set_and_entity(self) -> tuple[SymbolSet, Entity]: return SymbolSet.SEA_SURFACE, SeaSurfaceEntity.CARRIER - def mission_types(self, for_player: bool) -> Iterator[FlightType]: - from game.ato.flighttype import FlightType - - yield from super().mission_types(for_player) - if self.is_friendly(for_player): - yield from [ - FlightType.AEWC, - FlightType.REFUELING, - ] - def capture(self, game: Game, events: GameUpdateEvents, for_player: bool) -> None: raise RuntimeError("Carriers cannot be captured")