From 3b360cfcf533717e20ee75318c216e25e66aca52 Mon Sep 17 00:00:00 2001 From: Raffson Date: Sat, 27 Jul 2024 22:58:22 +0200 Subject: [PATCH] Pull up AEWC & Refueling mission types to NavalControlPoint --- game/theater/controlpoint.py | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) 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")