Pull up AEWC & Refueling mission types to NavalControlPoint

This commit is contained in:
Raffson 2024-07-27 22:58:22 +02:00
parent e5b06e6ba4
commit 3b360cfcf5
No known key found for this signature in database
GPG Key ID: B0402B2C9B764D99

View File

@ -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")