Fixes for colorbutton.

Author: mfitzp

qtwidgets/colorbutton/colorbutton.py

@@ -8,6 +8,7 @@
     from PySide2.QtCore import Qt, Signal
 
 
+
 class ColorButton(QtWidgets.QPushButton):
     '''
     Custom Qt Widget to show a chosen color.
@@ -18,14 +19,15 @@ class ColorButton(QtWidgets.QPushButton):
 
     colorChanged = Signal(object)
 
-    def __init__(self, *args, **kwargs):
+    def __init__(self, *args, color=None, **kwargs):
         super(ColorButton, self).__init__(*args, **kwargs)
 
         self._color = None
+        self._default = color
         self.pressed.connect(self.onColorPicker)
 
-    def sizeHint(self):
-        return QtCore.QSize(32,32)
+        # Set the initial/default state.
+        self.setColor(self._default)
 
     def setColor(self, color):
         if color != self._color:
@@ -56,6 +58,6 @@ def onColorPicker(self):
 
     def mousePressEvent(self, e):
         if e.button() == Qt.RightButton:
-            self.setColor(None)
+            self.setColor(self._default)
 
-        return super(ColorButton, self).mousePressEvent(e)
+        return super(ColorButton, self).mousePressEvent(e)

qtwidgets/colorbutton/demo_pyqt5.py

@@ -7,7 +7,7 @@ class Window(QtWidgets.QMainWindow):
     def __init__(self):
         super().__init__()
 
-        palette = ColorButton()
+        palette = ColorButton(color='red')
         palette.colorChanged.connect(self.show_selected_color)
         self.setCentralWidget(palette)
 

qtwidgets/colorbutton/demo_pyside2.py

@@ -7,7 +7,7 @@ class Window(QtWidgets.QMainWindow):
     def __init__(self):
         super().__init__()
 
-        palette = ColorButton()
+        palette = ColorButton(color='red')
         palette.colorChanged.connect(self.show_selected_color)
         self.setCentralWidget(palette)
 

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-version = '0.16'
+version = '0.17'
 
 with open("README.md", "r") as fh:
     long_description = fh.read()