mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
from dataclasses import dataclass, field
|
|
from typing import List
|
|
|
|
from game.ato import FlightType
|
|
from game.ato.package import Package
|
|
|
|
|
|
@dataclass
|
|
class AirTaskingOrder:
|
|
"""The entire ATO for one coalition."""
|
|
|
|
#: The set of all planned packages in the ATO.
|
|
packages: List[Package] = field(default_factory=list)
|
|
|
|
@property
|
|
def has_awacs_package(self) -> bool:
|
|
return any(
|
|
[
|
|
p
|
|
for p in self.packages
|
|
if any([f for f in p.flights if f.flight_type is FlightType.AEWC])
|
|
]
|
|
)
|
|
|
|
def add_package(self, package: Package) -> None:
|
|
"""Adds a package to the ATO."""
|
|
self.packages.append(package)
|
|
|
|
def remove_package(self, package: Package) -> None:
|
|
"""Removes a package from the ATO."""
|
|
# Remove all the flights individually so the database gets updated.
|
|
for flight in list(package.flights):
|
|
package.remove_flight(flight)
|
|
self.packages.remove(package)
|
|
|
|
def clear(self) -> None:
|
|
"""Removes all packages from the ATO."""
|
|
# Remove all packages individually so the database gets updated.
|
|
for package in list(self.packages):
|
|
self.remove_package(package)
|