add QEnum macro for PyQt bindings

Author: phil65

qtpy/QtCore.py

@@ -17,6 +17,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 *`
@@ -29,6 +35,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
@@ -47,7 +54,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,5 +1,6 @@
 """Test QtCore."""
 
+import enum
 import sys
 from datetime import date, datetime, time
 
@@ -69,6 +70,21 @@ def test_qthread_exec_():
     assert QtCore.QThread.exec_ is not None
 
 
+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