Support Qt 6

erikstantoncarlson · erikstantoncarlson · commit 4b4479c0b5e4 · 2025-11-10T15:28:35.000-08:00
diff --git a/hab_gui/actions/edit_custom_variables_action.py b/hab_gui/actions/edit_custom_variables_action.py
@@ -29,7 +29,7 @@ def __init__(self, settings, parent=None):
 
     def edit_custom_variables(self):
         dlg = CustomVariableEditor.create_dialog(self.settings, parent=self.parent())
-        dlg.exec_()
+        utils.exec_obj(dlg)
 
         # Ensure the hab_gui respects any changes the user may have made
         self.settings.root_widget.refresh_cache()
diff --git a/hab_gui/actions/verbosity_action.py b/hab_gui/actions/verbosity_action.py
@@ -1,4 +1,4 @@
-from Qt import QtCore, QtWidgets
+from Qt import QtWidgets
 
 
 class VerbosityAction(QtWidgets.QAction):
@@ -48,10 +48,7 @@ def __init__(self, settings, parent=None):
             action = menu.addAction(key)
             action.setData(value)
             action.setCheckable(True)
-            if verbosity == value:
-                action.setChecked(QtCore.Qt.Checked)
-            else:
-                action.setChecked(QtCore.Qt.Unchecked)
+            action.setChecked(verbosity == value)
         self.setMenu(menu)
 
     def load_config(self):
@@ -71,7 +68,4 @@ def refresh(self):
         verbosity = self.settings.verbosity
         for action in self.menu().actions():
             value = action.data()
-            if verbosity == value:
-                action.setChecked(QtCore.Qt.Checked)
-            else:
-                action.setChecked(QtCore.Qt.Unchecked)
+            action.setChecked(verbosity == value)
diff --git a/hab_gui/cli.py b/hab_gui/cli.py
@@ -170,7 +170,7 @@ def launch(settings, verbosity, uri, alias, args):
     if splash:
         splash.finish(window)
 
-    app.exec_()
+    utils.exec_obj(app)
 
 
 @gui.command()
diff --git a/hab_gui/dialogs/error_message_box.py b/hab_gui/dialogs/error_message_box.py
@@ -24,20 +24,20 @@ def __init__(self, etype, value, tb, parent=None):
         self.stack_limit = -5
 
         self.setWindowTitle("Exception")
-        self.setTextFormat(QtCore.Qt.RichText)
-        self.setStandardButtons(QtWidgets.QMessageBox.Ok)
+        self.setTextFormat(QtCore.Qt.TextFormat.RichText)
+        self.setStandardButtons(QtWidgets.QMessageBox.StandardButton.Ok)
         # Using detailedText seems to disable the close button, enable it
-        self.setEscapeButton(QtWidgets.QMessageBox.Ok)
+        self.setEscapeButton(QtWidgets.QMessageBox.StandardButton.Ok)
 
         # Create a button allowing the user to copy the non-highlighted text
-        copy_btn = self.addButton("Copy", QtWidgets.QMessageBox.ActionRole)
+        copy_btn = self.addButton("Copy", QtWidgets.QMessageBox.ButtonRole.ActionRole)
         copy_btn.setToolTip("Copy the full traceback for error reporting.")
         # Disconnect the QMessageBox signals that would cause the box to close
         # when this button is pressed and add our own signal connection
         copy_btn.disconnect()
         copy_btn.released.connect(self.copy_traceback)
 
-        self.setDefaultButton(QtWidgets.QMessageBox.Ok)
+        self.setDefaultButton(QtWidgets.QMessageBox.StandardButton.Ok)
         self.refresh()
 
     @QtCore.Slot()
diff --git a/hab_gui/dialogs/uri_picker_dialog.py b/hab_gui/dialogs/uri_picker_dialog.py
@@ -68,9 +68,11 @@ def init_gui(self):
         )
 
         self.uiButtonsBOX = QtWidgets.QDialogButtonBox(
-            QtWidgets.QDialogButtonBox.Cancel, self
+            QtWidgets.QDialogButtonBox.StandardButton.Cancel, self
+        )
+        self.uiButtonsBOX.addButton(
+            "Launch", QtWidgets.QDialogButtonBox.ButtonRole.AcceptRole
         )
-        self.uiButtonsBOX.addButton("Launch", QtWidgets.QDialogButtonBox.AcceptRole)
         self.uiButtonsBOX.accepted.connect(self.accept)
         self.uiButtonsBOX.rejected.connect(self.reject)
 
@@ -151,7 +153,7 @@ def should_show(cls, settings, alias=None):
         # Note: Not using `keyboardModifiers` because it is not updated when
         # calling this from the cli module.
         modifiers = QtWidgets.QApplication.queryKeyboardModifiers()
-        if modifiers == QtCore.Qt.ShiftModifier:
+        if modifiers == QtCore.Qt.KeyboardModifier.ShiftModifier:
             return True
 
         # always_ask is checked for this alias
diff --git a/hab_gui/entry_points/message_box.py b/hab_gui/entry_points/message_box.py
@@ -1,5 +1,6 @@
 import logging
 
+from .. import utils
 from .logging_exception import LoggingExceptionInit
 
 logger = logging.getLogger(__name__)
@@ -17,4 +18,4 @@ def excepthook(self, cls, exception, tb):
         from ..dialogs.error_message_box import ErrorMessageBox
 
         box = ErrorMessageBox(cls, exception, tb, parent=None)
-        box.exec_()
+        utils.exec_obj(box)
diff --git a/hab_gui/utils.py b/hab_gui/utils.py
@@ -11,7 +11,7 @@
 
 
 @contextmanager
-def cursor_override(cursor=QtCore.Qt.BusyCursor):
+def cursor_override(cursor=QtCore.Qt.CursorShape.BusyCursor):
     """Change the application cursor to wait while running the context/decorator.
     Ensures that the cursor is restored even if an exception is raised.
     """
@@ -220,3 +220,14 @@ def block_signals(objs):
     finally:
         for o, b in blocked:
             o.blockSignals(b)
+
+
+def exec_obj(obj, *args, **kwargs):
+    """Work around the removal of `exec_` from Qt6(especially PyQt6).
+
+    This calls the `obj.exec` method if it exists, and falls back to `obj.exec_`
+    otherwise.
+    """
+    if hasattr(obj, "exec"):
+        return obj.exec(*args, **kwargs)
+    return obj.exec_(*args, **kwargs)
diff --git a/hab_gui/widgets/alias_button.py b/hab_gui/widgets/alias_button.py
@@ -23,8 +23,9 @@ def __init__(self, cfg, alias_name, parent=None):
         self.alias_name = alias_name
 
         self.alias_dict = self.cfg.aliases
-        qsize_policy = QtWidgets.QSizePolicy
-        size_policy = qsize_policy(qsize_policy.Minimum, qsize_policy.Preferred)
+        size_policy = QtWidgets.QSizePolicy(
+            QtWidgets.QSizePolicy.Policy.Minimum, QtWidgets.QSizePolicy.Policy.Preferred
+        )
         self.setSizePolicy(size_policy)
         self.clicked.connect(self._button_action)
         self.refresh()
diff --git a/hab_gui/widgets/alias_icon_button.py b/hab_gui/widgets/alias_icon_button.py
@@ -22,7 +22,7 @@ class AliasIconButton(AliasButton):
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        self.setToolButtonStyle(QtCore.Qt.ToolButtonTextBesideIcon)
+        self.setToolButtonStyle(QtCore.Qt.ToolButtonStyle.ToolButtonTextBesideIcon)
 
     def refresh(self):
         alias = self.alias_dict[self.alias_name]
diff --git a/hab_gui/widgets/custom_variable_editor/custom_variable_editor.py b/hab_gui/widgets/custom_variable_editor/custom_variable_editor.py
@@ -70,7 +70,7 @@ def editing_finished(self, top_left, bottom_right, roles):
         if self._is_refreshing:
             return
 
-        if QtCore.Qt.EditRole in roles:
+        if QtCore.Qt.ItemDataRole.EditRole in roles:
             item = self.uiVariableTREE.itemFromIndex(top_left)
             column = top_left.column()
             if column == 0:
diff --git a/hab_gui/widgets/custom_variable_editor/file_tree_widget_item.py b/hab_gui/widgets/custom_variable_editor/file_tree_widget_item.py
@@ -18,7 +18,7 @@ def __init__(self, parent, parser):
 
         # Add a child item that shows the filename. It should not be editable.
         self.filename_item = QtWidgets.QTreeWidgetItem(self)
-        self.filename_item.setFlags(QtCore.Qt.NoItemFlags)
+        self.filename_item.setFlags(QtCore.Qt.ItemFlag.NoItemFlags)
 
         self.refresh()
 
diff --git a/hab_gui/widgets/custom_variable_editor/variable_tree_widget_item.py b/hab_gui/widgets/custom_variable_editor/variable_tree_widget_item.py
@@ -8,7 +8,7 @@ def __init__(self, parent, variable_name):
         super().__init__(parent)
         self.parser = parent.parser
         self._variable_name = variable_name
-        self.setFlags(self.flags() | QtCore.Qt.ItemIsEditable)
+        self.setFlags(self.flags() | QtCore.Qt.ItemFlag.ItemIsEditable)
         self.refresh()
 
     def remove_variable(self):
diff --git a/hab_gui/widgets/menu_button.py b/hab_gui/widgets/menu_button.py
@@ -29,7 +29,7 @@ def __init__(self, settings, parent=None):
 
         self.setText("Menu")
         self.setIcon(utils.Paths.icon("menu.svg"))
-        self.setPopupMode(self.InstantPopup)
+        self.setPopupMode(QtWidgets.QToolButton.ToolButtonPopupMode.InstantPopup)
         self.refresh()
 
     @property
diff --git a/hab_gui/widgets/name_picker.py b/hab_gui/widgets/name_picker.py
@@ -46,7 +46,7 @@ def names(self):
         ret = {}
         for index in range(self.name_tree.topLevelItemCount()):
             item = self.name_tree.topLevelItem(index)
-            checked = item.checkState(0) == QtCore.Qt.Checked
+            checked = item.checkState(0) == QtCore.Qt.CheckState.Checked
             ret[item.text(0)] = [item.text(1), checked]
         return ret
 
@@ -61,7 +61,7 @@ def set_names(self, names, uri=None):
                 # Build the `default_selection` set for the current URI
                 if len(settings) > 1 and settings[1]:
                     self.default_selection.add(name)
-                item.setCheckState(0, QtCore.Qt.Unchecked)
+                item.setCheckState(0, QtCore.Qt.CheckState.Unchecked)
 
         self.name_tree.resizeColumnToContents(0)
         user_selection = self.user_selection(uri)
@@ -84,7 +84,7 @@ def selected(self):
         ret = set()
         for index in range(self.name_tree.topLevelItemCount()):
             item = self.name_tree.topLevelItem(index)
-            if item.checkState(0) == QtCore.Qt.Checked:
+            if item.checkState(0) == QtCore.Qt.CheckState.Checked:
                 ret.add(item.text(0))
         return ret
 
@@ -95,7 +95,10 @@ def set_selected(self, selected):
                 item = self.name_tree.topLevelItem(index)
                 name = item.text(0)
                 item.setCheckState(
-                    0, QtCore.Qt.Checked if name in selected else QtCore.Qt.Unchecked
+                    0,
+                    QtCore.Qt.CheckState.Checked
+                    if name in selected
+                    else QtCore.Qt.CheckState.Unchecked,
                 )
 
     def sizeHint(self):  # noqa: N802
diff --git a/hab_gui/widgets/pinned_uris_button.py b/hab_gui/widgets/pinned_uris_button.py
@@ -31,7 +31,7 @@ def __init__(self, settings, uri_widget, parent=None):
         self.setToolTip("Select and manage quick access to commonly used URI's.")
         self.setText(self._text_main)
         self.setIcon(utils.Paths.icon("pin-outline.svg"))
-        self.setPopupMode(self.InstantPopup)
+        self.setPopupMode(QtWidgets.QToolButton.ToolButtonPopupMode.InstantPopup)
         self.refresh()
 
     def add_uri(self, uri):
diff --git a/hab_gui/windows/alias_launch_window.py b/hab_gui/windows/alias_launch_window.py
@@ -1,4 +1,5 @@
 import logging
+import math
 from functools import partial
 
 import hab
@@ -62,7 +63,7 @@ def __init__(
         refresh_time = refresh_time[0]
         if refresh_time:
             self.refresh_timer.timeout.connect(partial(self.refresh_cache, False))
-            refresh_time = utils.interval(refresh_time)
+            refresh_time = math.ceil(utils.interval(refresh_time))
             logger.debug(f"Setting auto-refresh interval to {refresh_time} seconds")
             self.refresh_timer.start(refresh_time * 1000)
 
@@ -86,7 +87,10 @@ def apply_layout(self):
             self.layout.addWidget(self.footer_widget, 2, 0, 1, -1)
         else:
             self.spacer_item = QtWidgets.QSpacerItem(
-                0, 0, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding
+                0,
+                0,
+                QtWidgets.QSizePolicy.Policy.Minimum,
+                QtWidgets.QSizePolicy.Policy.Expanding,
             )
             self.layout.addItem(self.spacer_item, self.layout.rowCount(), 0, 1, -1)
 
@@ -227,4 +231,4 @@ def main():
     app = QtWidgets.QApplication([])
     window = AliasLaunchWindow(hab.Resolver(target="hab-gui"), verbosity=1)
     window.show()
-    app.exec_()
+    utils.exec_obj(app)
diff --git a/tests/test_echo_message.py b/tests/test_echo_message.py
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -0,0 +1,22 @@
+import hab_gui.utils
+
+
+def test_exec_obj():
+    class ExecBoth:
+        def exec_(self):
+            return "exec_"
+
+        def exec(self):
+            return "exec"
+
+    class Exec_:
+        def exec_(self):
+            return "exec_"
+
+    class Exec:
+        def exec(self):
+            return "exec"
+
+    assert hab_gui.utils.exec_obj(ExecBoth()) == "exec"
+    assert hab_gui.utils.exec_obj(Exec_()) == "exec_"
+    assert hab_gui.utils.exec_obj(Exec()) == "exec"
