mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
0f34946127
This is the first step toward bundling all assets related to a save game into a single item. That makes it easier to avoid clobbering "temporary" assets from other games like the state.json, but also makes it easier for players to file bug reports, since there's only a single asset to upload. This is only the first step because so far it only includes the various save files: start of turn, end of last turn before results processing, and "latest" (the game saved explicitly by the player).
135 lines
3.9 KiB
Python
135 lines
3.9 KiB
Python
"""Command-line utility for displaying human readable loadout configurations."""
|
|
import argparse
|
|
import sys
|
|
from collections.abc import Iterator
|
|
from pathlib import Path
|
|
from typing import Type
|
|
|
|
from dcs.helicopters import helicopter_map
|
|
from dcs.planes import plane_map
|
|
from dcs.unittype import FlyingType
|
|
|
|
from game import persistence
|
|
from game.ato import FlightType
|
|
from game.ato.loadouts import Loadout
|
|
from game.dcs.aircrafttype import AircraftType
|
|
|
|
# TODO: Move this logic out of the UI.
|
|
from qt_ui import liberation_install
|
|
from qt_ui.main import inject_custom_payloads
|
|
|
|
|
|
def non_empty_loadouts_for(
|
|
aircraft: Type[FlyingType],
|
|
) -> Iterator[tuple[FlightType, Loadout]]:
|
|
for task in FlightType:
|
|
try:
|
|
loadout = Loadout.default_for_task_and_aircraft(task, aircraft)
|
|
except KeyError:
|
|
# Not all aircraft have a unitPayloads field. This should maybe be handled
|
|
# in pydcs, but I'm not sure about the cause. For now, just ignore the field
|
|
# since we can be less robust in optional tooling.
|
|
continue
|
|
|
|
if loadout.name != "Empty":
|
|
yield task, loadout
|
|
|
|
|
|
def print_pylons(loadout: Loadout, prefix: str = "\t") -> None:
|
|
pylons = dict(sorted(loadout.pylons.items()))
|
|
for pylon_id, weapon in pylons.items():
|
|
if weapon is not None:
|
|
print(f"{prefix}{pylon_id}: {weapon.name}")
|
|
|
|
|
|
def show_all_loadouts(aircraft: Type[FlyingType]) -> None:
|
|
loadouts = list(non_empty_loadouts_for(aircraft))
|
|
if not loadouts:
|
|
return
|
|
|
|
print(f"Loadouts for {aircraft.id}:")
|
|
for task, loadout in loadouts:
|
|
print(f"\t{task.value}: {loadout.name}")
|
|
print_pylons(loadout, prefix="\t\t")
|
|
|
|
|
|
def task_by_name(name: str) -> FlightType:
|
|
for task in FlightType:
|
|
if task.value == name:
|
|
return task
|
|
raise KeyError(f"No FlightType named {name}")
|
|
|
|
|
|
def show_single_loadout(aircraft: Type[FlyingType], task_name: str) -> None:
|
|
task = task_by_name(task_name)
|
|
try:
|
|
loadout = Loadout.default_for_task_and_aircraft(task, aircraft)
|
|
except KeyError:
|
|
# Not all aircraft have a unitPayloads field. This should maybe be handled
|
|
# in pydcs, but I'm not sure about the cause. For now, just ignore the field
|
|
# since we can be less robust in optional tooling.
|
|
return
|
|
if loadout.pylons:
|
|
print(f"{aircraft.id} {loadout.name}:")
|
|
print_pylons(loadout)
|
|
|
|
|
|
def show_loadouts_for(aircraft: Type[FlyingType], task_name: str | None) -> None:
|
|
if task_name is None:
|
|
show_all_loadouts(aircraft)
|
|
else:
|
|
show_single_loadout(aircraft, task_name)
|
|
|
|
|
|
def show_all_aircraft(task_name: str | None) -> None:
|
|
for aircraft in AircraftType.each_dcs_type():
|
|
show_loadouts_for(aircraft, task_name)
|
|
|
|
|
|
def show_single_aircraft(aircraft_id: str, task_name: str | None) -> None:
|
|
try:
|
|
aircraft: Type[FlyingType] = plane_map[aircraft_id]
|
|
except KeyError:
|
|
aircraft = helicopter_map[aircraft_id]
|
|
show_loadouts_for(aircraft, task_name)
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument(
|
|
"--aircraft-id",
|
|
help=(
|
|
"ID of the aircraft to display loadouts for. By default all aircraft will "
|
|
+ "be displayed."
|
|
),
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--task",
|
|
help=(
|
|
"Name of the mission type to display. By default loadouts for all mission "
|
|
+ "types will be displayed."
|
|
),
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
first_start = liberation_install.init()
|
|
if first_start:
|
|
sys.exit(
|
|
"Cannot view payloads without configuring DCS Liberation. Start the UI for "
|
|
"the first run configuration."
|
|
)
|
|
|
|
inject_custom_payloads(Path(persistence.base_path()))
|
|
|
|
if args.aircraft_id is None:
|
|
show_all_aircraft(args.task)
|
|
else:
|
|
show_single_aircraft(args.aircraft_id, args.task)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|