mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
d94c57afd6
If no airfield was selected (as sometimes happens, usually when there are no aircraft available anywhere), report that no aircraft are available.
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
"""Combo box for selecting a departure airfield."""
|
|
from typing import Iterable
|
|
|
|
from PySide2.QtWidgets import QComboBox
|
|
|
|
from dcs.planes import PlaneType
|
|
from game.inventory import GlobalAircraftInventory
|
|
from theater.controlpoint import ControlPoint
|
|
|
|
|
|
class QOriginAirfieldSelector(QComboBox):
|
|
"""A combo box for selecting a flight's departure airfield.
|
|
|
|
The combo box will automatically be populated with all departure airfields
|
|
that have unassigned inventory of the given aircraft type.
|
|
"""
|
|
|
|
def __init__(self, global_inventory: GlobalAircraftInventory,
|
|
origins: Iterable[ControlPoint],
|
|
aircraft: PlaneType) -> None:
|
|
super().__init__()
|
|
self.global_inventory = global_inventory
|
|
self.origins = list(origins)
|
|
self.aircraft = aircraft
|
|
self.rebuild_selector()
|
|
|
|
def change_aircraft(self, aircraft: PlaneType) -> None:
|
|
if self.aircraft == aircraft:
|
|
return
|
|
self.aircraft = aircraft
|
|
self.rebuild_selector()
|
|
|
|
def rebuild_selector(self) -> None:
|
|
self.clear()
|
|
for origin in self.origins:
|
|
inventory = self.global_inventory.for_control_point(origin)
|
|
available = inventory.available(self.aircraft)
|
|
if available:
|
|
self.addItem(f"{origin.name} ({available} available)", origin)
|
|
self.model().sort(0)
|
|
self.update()
|
|
|
|
@property
|
|
def available(self) -> int:
|
|
origin = self.currentData()
|
|
if origin is None:
|
|
return 0
|
|
inventory = self.global_inventory.for_control_point(origin)
|
|
return inventory.available(self.aircraft)
|