From c75cfb60ad5a4aaca5b76cd0897fbc3bfa7f2994 Mon Sep 17 00:00:00 2001
From: Emanuel Gardaya Calso <eman.calso@evren.co>
Date: Tue, 7 May 2024 14:37:53 +0800
Subject: [PATCH] Add pyside6 compatibility

---
 qtwidgets/colorbutton/colorbutton.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/qtwidgets/colorbutton/colorbutton.py b/qtwidgets/colorbutton/colorbutton.py
index 9d3ac1c..b03bc38 100644
--- a/qtwidgets/colorbutton/colorbutton.py
+++ b/qtwidgets/colorbutton/colorbutton.py
@@ -1,14 +1,17 @@
 import sys
-if 'PyQt5' in sys.modules:
-    from PyQt5 import QtCore, QtGui, QtWidgets
-    from PyQt5.QtCore import Qt, pyqtSignal as Signal
 
+if "PySide6" in sys.modules:
+    from PySide6 import QtGui, QtWidgets
+    from PySide6.QtCore import Qt, Signal
+elif "PyQt5" in sys.modules:
+    from PyQt5 import QtGui, QtWidgets
+    from PyQt5.QtCore import Qt
+    from PyQt5.QtCore import pyqtSignal as Signal
 else:
-    from PySide2 import QtCore, QtGui, QtWidgets
+    from PySide2 import QtGui, QtWidgets
     from PySide2.QtCore import Qt, Signal
 
 
-
 class ColorButton(QtWidgets.QPushButton):
     '''
     Custom Qt Widget to show a chosen color.