Account for 'is_ownfor' in package dialog

This commit is contained in:
Raffson 2024-05-25 23:05:35 +02:00
parent 12d0632415
commit 8562c48753
No known key found for this signature in database
GPG Key ID: B0402B2C9B764D99
2 changed files with 5 additions and 9 deletions

View File

@ -44,9 +44,7 @@ class Dialog:
@classmethod @classmethod
def open_edit_package_dialog(cls, package_model: PackageModel): def open_edit_package_dialog(cls, package_model: PackageModel):
"""Opens the dialog to edit the given package.""" """Opens the dialog to edit the given package."""
cls.edit_package_dialog = QEditPackageDialog( cls.edit_package_dialog = QEditPackageDialog(cls.game_model, package_model)
cls.game_model, cls.game_model.ato_model, package_model
)
cls.edit_package_dialog.show() cls.edit_package_dialog.show()
@classmethod @classmethod

View File

@ -23,7 +23,7 @@ from game.radio.radios import RadioFrequency
from game.server import EventStream from game.server import EventStream
from game.sim import GameUpdateEvents from game.sim import GameUpdateEvents
from game.theater.missiontarget import MissionTarget from game.theater.missiontarget import MissionTarget
from qt_ui.models import AtoModel, GameModel, PackageModel from qt_ui.models import GameModel, PackageModel
from qt_ui.uiconstants import EVENT_ICONS from qt_ui.uiconstants import EVENT_ICONS
from qt_ui.widgets.QFrequencyWidget import QFrequencyWidget from qt_ui.widgets.QFrequencyWidget import QFrequencyWidget
from qt_ui.widgets.ato import QFlightList from qt_ui.widgets.ato import QFlightList
@ -333,11 +333,9 @@ class QEditPackageDialog(QPackageDialog):
Changes to existing packages occur immediately. Changes to existing packages occur immediately.
""" """
def __init__( def __init__(self, gm: GameModel, package: PackageModel) -> None:
self, game_model: GameModel, model: AtoModel, package: PackageModel super().__init__(gm, package)
) -> None: self.ato_model = gm.ato_model if gm.is_ownfor else gm.red_ato_model
super().__init__(game_model, package)
self.ato_model = model
self.delete_button = QPushButton("Delete package") self.delete_button = QPushButton("Delete package")
self.delete_button.setProperty("style", "btn-danger") self.delete_button.setProperty("style", "btn-danger")