mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
from PySide2.QtWidgets import QSpinBox
|
|
from dcs.unitpropertydescription import UnitPropertyDescription
|
|
|
|
from game.ato.flightmember import FlightMember
|
|
from .missingpropertydataerror import MissingPropertyDataError
|
|
|
|
|
|
class PropertySpinBox(QSpinBox):
|
|
def __init__(
|
|
self, flight_member: FlightMember, prop: UnitPropertyDescription
|
|
) -> None:
|
|
super().__init__()
|
|
self.flight_member = flight_member
|
|
self.prop = prop
|
|
|
|
if prop.minimum is None:
|
|
raise MissingPropertyDataError("minimum cannot be None")
|
|
if prop.maximum is None:
|
|
raise MissingPropertyDataError("maximum cannot be None")
|
|
if prop.default is None:
|
|
raise MissingPropertyDataError("default cannot be None")
|
|
|
|
self.setMinimum(prop.minimum)
|
|
self.setMaximum(prop.maximum)
|
|
self.setValue(
|
|
self.flight_member.properties.get(self.prop.identifier, self.prop.default)
|
|
)
|
|
|
|
self.valueChanged.connect(self.on_value_changed)
|
|
|
|
def on_value_changed(self, value: int) -> None:
|
|
self.flight_member.properties[self.prop.identifier] = value
|
|
|
|
def set_flight_member(self, flight_member: FlightMember) -> None:
|
|
self.flight_member = flight_member
|
|
self.setValue(
|
|
self.flight_member.properties.get(self.prop.identifier, self.prop.default)
|
|
)
|