Skip to content

Commit

Permalink
compat.py: Add utility functions that convert between ints and Qt Enums
Browse files Browse the repository at this point in the history
  • Loading branch information
zjp committed Jul 21, 2024
1 parent 0f7b181 commit fef4bda
Show file tree
Hide file tree
Showing 2 changed files with 37 additions and 0 deletions.
24 changes: 24 additions & 0 deletions qtpy/compat.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
"""
Compatibility functions
"""
import enum
import sys

from . import (
Expand Down Expand Up @@ -200,3 +201,26 @@ def isalive(obj):

return shiboken.isValid(obj)
return None


# =============================================================================
def getenumasint(enum_value):
"""Get the integer value of a Qt enum
For example:
Qt.AlignmentFlag.AlignBaseline -> 256
Qt.WidgetAttribute.WA_AcceptDrops -> 78
If an integer is passed in, simply return it.
PySide2's enums are themselves classes, not enum values per se, so if
we get an integer or a class, return the class.
"""
if isinstance(enum_value, enum.Enum):
if PYSIDE2 or PYSIDE6:
return int(enum_value)
return enum_value.value
return enum_value


# =============================================================================
def getenumfromint(enum_class, i):
"""Get the Qt enum value from an integer"""
return enum_class(i)
13 changes: 13 additions & 0 deletions qtpy/tests/test_compat.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Test the compat module."""

import sys

import pytest
Expand All @@ -22,3 +23,15 @@ def test_isalive(qtbot):
with qtbot.waitSignal(test_widget.destroyed):
test_widget.deleteLater()
assert compat.isalive(test_widget) is False


def test_getenumasint():
"""Test compat.getenumasint"""
assert compat.getenumasint(QtWidgets.QSizePolicy.Policy.Maximum) == 4
assert compat.getenumasint(5) == 5


def test_getenumfromint():
"""Test compat.getenumfromint"""
enum_value = compat.getenumfromint(QtWidgets.QSizePolicy.Policy, 7)
assert enum_value == QtWidgets.QSizePolicy.Policy.Expanding

0 comments on commit fef4bda

Please sign in to comment.