dcs-retribution/qt_ui/dialogs.py
Walter cdb16cc591 Fixes #268
Changes red base info to show defenses first, and adds an attack option
2020-11-02 18:06:10 -08:00

67 lines
2.3 KiB
Python

"""Application-wide dialog management."""
from typing import Optional
from gen.flights.flight import Flight
from theater.missiontarget import MissionTarget
from .models import GameModel, PackageModel
from .windows.mission.QEditFlightDialog import QEditFlightDialog
from .windows.mission.QPackageDialog import (
QEditPackageDialog,
QNewPackageDialog,
)
class Dialog:
"""Dialog management singleton.
Opens dialogs and keeps references to dialog windows so that their creators
do not need to worry about the lifetime of the dialog object, and can open
dialogs without needing to have their own reference to common data like the
game model.
"""
#: The game model. Is only None before initialization, as the game model
#: itself is responsible for handling the case where no game is loaded.
game_model: Optional[GameModel] = None
new_package_dialog: Optional[QNewPackageDialog] = None
edit_package_dialog: Optional[QEditPackageDialog] = None
edit_flight_dialog: Optional[QEditFlightDialog] = None
@classmethod
def set_game(cls, game_model: GameModel) -> None:
"""Sets the game model."""
cls.game_model = game_model
@classmethod
def open_new_package_dialog(cls, mission_target: MissionTarget, parent=None):
"""Opens the dialog to create a new package with the given target."""
cls.new_package_dialog = QNewPackageDialog(
cls.game_model,
cls.game_model.ato_model,
mission_target,
parent=parent
)
cls.new_package_dialog.show()
@classmethod
def open_edit_package_dialog(cls, package_model: PackageModel):
"""Opens the dialog to edit the given package."""
cls.edit_package_dialog = QEditPackageDialog(
cls.game_model,
cls.game_model.ato_model,
package_model
)
cls.edit_package_dialog.show()
@classmethod
def open_edit_flight_dialog(cls, package_model: PackageModel,
flight: Flight) -> None:
"""Opens the dialog to edit the given flight."""
cls.edit_flight_dialog = QEditFlightDialog(
cls.game_model,
package_model.package,
flight
)
cls.edit_flight_dialog.show()