compat.py: Add utility function to get raw bytes from QImages

spyder-ide · Apr 18, 2024 · b467e4d · b467e4d
1 parent 3238de7
commit b467e4d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/qtpy/compat.py b/qtpy/compat.py
@@ -12,6 +12,7 @@
     PYQT6,
     PYSIDE2,
     PYSIDE6,
+    QtBindingsNotFoundError,
 )
 from .QtWidgets import QFileDialog
 
@@ -200,3 +201,14 @@ def isalive(obj):
 
         return shiboken.isValid(obj)
     return None
+
+
+# =============================================================================
+def getimagebytes(qimage):
+    if PYQT5:
+        return qimage.bits().asstring(qimage.byteCount())
+    if PYQT6:
+        return qimage.bits().asstring(qimage.sizeInBytes())
+    if PYSIDE2 or PYSIDE6:
+        return qimage.bits().tobytes()
+    raise QtBindingsNotFoundError