mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
29 lines
954 B
Python
29 lines
954 B
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
|
|
from game.ato.flighttype import FlightType
|
|
from game.commander.tasks.packageplanningtask import PackagePlanningTask
|
|
from game.commander.theaterstate import TheaterState
|
|
from game.theater import MissionTarget
|
|
|
|
|
|
@dataclass
|
|
class PlanRefueling(PackagePlanningTask[MissionTarget]):
|
|
def preconditions_met(self, state: TheaterState) -> bool:
|
|
if (
|
|
state.context.coalition.player
|
|
and not state.context.settings.auto_ato_behavior_tankers
|
|
):
|
|
return False
|
|
if not super().preconditions_met(state):
|
|
return False
|
|
return self.target in state.refueling_targets
|
|
|
|
def apply_effects(self, state: TheaterState) -> None:
|
|
state.refueling_targets.remove(self.target)
|
|
|
|
def propose_flights(self) -> None:
|
|
self.propose_flight(FlightType.REFUELING, 1)
|
|
self.propose_flight(FlightType.ESCORT, 2)
|