From 86bc41c15c8dad68d76f740e6be5824dfae16c28 Mon Sep 17 00:00:00 2001 From: Walter Date: Mon, 2 Nov 2020 19:19:41 -0600 Subject: [PATCH] Add base menu to new package dialog for red cp --- qt_ui/widgets/map/QMapControlPoint.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/qt_ui/widgets/map/QMapControlPoint.py b/qt_ui/widgets/map/QMapControlPoint.py index d90a973d..e59cdbfb 100644 --- a/qt_ui/widgets/map/QMapControlPoint.py +++ b/qt_ui/widgets/map/QMapControlPoint.py @@ -90,3 +90,10 @@ class QMapControlPoint(QMapObject): # Reinitialized ground planners and the like. self.game_model.game.initialize_turn() GameUpdateSignal.get_instance().updateGame(self.game_model.game) + + def open_new_package_dialog(self) -> None: + """Extends the default packagedialog to redirect to base menu for red air base.""" + if not self.control_point.captured: + self.on_click() + else: + super(QMapControlPoint, self).open_new_package_dialog()