commit message

1.add QColorDialog-setCustomColor modifier
astips · Mar 27, 2019 · 880aaf6 · 880aaf6
1 parent 99eaee3
commit 880aaf6
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 2 deletions.
diff --git a/modifier/modifier_pyqt4.py b/modifier/modifier_pyqt4.py
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 
 from . import QtModifier
+from PyQt4 import QtGui
 
 
 qt_modifier = QtModifier()
@@ -27,3 +28,8 @@ def unwrapinstance(*args, **kwargs):
 @qt_modifier.register('QtCore.QString.__str__')
 def to_string(self):
     return str(self.toUtf8)
+
+
+@qt_modifier.register("QtWidgets.QColorDialog.set_custom_color")
+def set_custom_color(self, index, color):
+    QtGui.QColorDialog.setCustomColor(index, color.rgb())
diff --git a/modifier/modifier_pyqt5.py b/modifier/modifier_pyqt5.py
@@ -27,7 +27,7 @@ def unwrapinstance(*args, **kwargs):
 
 @qt_modifier.register('QtWidgets.QTreeWidgetItem.setBackgroundColor')
 def setBackgroundColor(self, p_int, QColor):
-    QtWidgets.QTreeWidgetItem.setBackgroundColor(self, p_int, QColor)
+    QtWidgets.QTreeWidgetItem.setBackground(self, p_int, QColor)
 
 
 @qt_modifier.register('QtWidgets.QTreeWidgetItem.setTextColor')
@@ -37,7 +37,7 @@ def setTextColor(self, p_int, QColor):
 
 @qt_modifier.register('QtWidgets.QTableWidgetItem.setBackgroundColor')
 def setBackgroundColor(self, QColor):
-    QtWidgets.QTableWidgetItem.setBackgroundColor(self, QColor)
+    QtWidgets.QTableWidgetItem.setBackground(self, QColor)
 
 
 @qt_modifier.register("QtWidgets.QTableWidgetItem.setTextColor")
@@ -58,3 +58,8 @@ def delta(self):
 @qt_modifier.register("QtGui.QDrag.start")
 def start(self, supportedActions=None):
     QtGui.QDrag.exec_(self, supportedActions)
+
+
+@qt_modifier.register("QtWidgets.QColorDialog.set_custom_color")
+def set_custom_color(self, index, color):
+    QtWidgets.QColorDialog.setCustomColor(index, color)
diff --git a/modifier/modifier_pyside.py b/modifier/modifier_pyside.py
@@ -55,6 +55,11 @@ def setTextColor(self, p_int, QColor):
     QtGui.QTreeWidgetItem.setForeground(self, p_int, QColor)
 
 
+@qt_modifier.register("QtWidgets.QColorDialog.set_custom_color")
+def set_custom_color(self, index, color):
+    QtGui.QColorDialog.setCustomColor(index, color.rgb())
+
+
 original_set_focus_policy = QtGui.QWidget.setFocusPolicy
 
 

diff --git a/modifier/modifier_pyside2.py b/modifier/modifier_pyside2.py
@@ -67,6 +67,11 @@ def setResizeMode(self, mode):
     QtWidgets.QHeaderView.setSectionResizeMode(self, mode)
 
 
+@qt_modifier.register("QtWidgets.QColorDialog.set_custom_color")
+def set_custom_color(self, index, color):
+    QtWidgets.QColorDialog.setCustomColor(index, color)
+
+
 original_set_focus_policy = QtWidgets.QWidget.setFocusPolicy