PR: Add QEnum macro for PyQt bindings (#424)

Author: ccordoba12

qtpy/QtCore.py

@@ -18,6 +18,12 @@
     from PyQt5.QtCore import pyqtBoundSignal as SignalInstance
     from PyQt5.QtCore import pyqtSlot as Slot
     from PyQt5.QtCore import pyqtProperty as Property
+    try:
+        from PyQt5.QtCore import Q_ENUM as QEnum
+        del Q_ENUM
+    except ImportError:  # fallback for Qt5.9
+        from PyQt5.QtCore import Q_ENUMS as QEnum
+        del Q_ENUMS
     from PyQt5.QtCore import QT_VERSION_STR as __version__
 
     # Those are imported from `import *`
@@ -30,6 +36,7 @@
     from PyQt6.QtCore import pyqtBoundSignal as SignalInstance
     from PyQt6.QtCore import pyqtSlot as Slot
     from PyQt6.QtCore import pyqtProperty as Property
+    from PyQt6.QtCore import pyqtEnum as QEnum
     from PyQt6.QtCore import QT_VERSION_STR as __version__
 
     # For issue #311
@@ -48,7 +55,7 @@
     QThread.exec_ = lambda self, *args, **kwargs: self.exec(*args, **kwargs)
 
     # Those are imported from `import *`
-    del pyqtSignal, pyqtBoundSignal, pyqtSlot, pyqtProperty, QT_VERSION_STR
+    del pyqtSignal, pyqtBoundSignal, pyqtSlot, pyqtProperty, pyqtEnum, QT_VERSION_STR
 
     # Allow unscoped access for enums inside the QtCore module
     from .enums_compat import promote_enums

qtpy/tests/test_qtcore.py

@@ -1,20 +1,20 @@
 """Test QtCore."""
 
+import enum
 import sys
 from datetime import date, datetime, time
 
 import pytest
+from packaging.version import parse
 
 from qtpy import (
     PYQT5,
     PYQT6,
     PYSIDE2,
-    PYSIDE6,
     PYQT_VERSION,
     PYSIDE_VERSION,
     QtCore,
 )
-from qtpy.tests.utils import not_using_conda
 
 _now = datetime.now()
 # Make integer milliseconds; `floor` here, don't `round`!
@@ -70,6 +70,23 @@ def test_qthread_exec():
     assert QtCore.QThread.exec is not None
 
 
+@pytest.mark.skipif(PYSIDE2 and parse(PYSIDE_VERSION) < parse("5.15"),
+                    reason="QEnum macro doesn't seem to be present on PySide2 <5.15")
+def test_qenum():
+    """Test QEnum macro"""
+    class EnumTest(QtCore.QObject):
+        class Position(enum.IntEnum):
+            West = 0
+            North = 1
+            South = 2
+            East = 3
+
+        QtCore.QEnum(Position)
+
+    obj = EnumTest()
+    assert obj.metaObject().enumerator(0).name() == "Position"
+
+
 def test_QLibraryInfo_location_and_path():
     """Test `QLibraryInfo.location` and `QLibraryInfo.path`"""
     assert QtCore.QLibraryInfo.location is not None