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

spyder-ide · Apr 18, 2024 · abb601c · abb601c
1 parent 3238de7
commit abb601c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/qtpy/compat.py b/qtpy/compat.py
@@ -200,3 +200,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()