diff --git a/modifier/modifier_pyqt4.py b/modifier/modifier_pyqt4.py index a4ae064..3121551 100644 --- 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 index 9a31766..720b9b6 100644 --- 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 index b798407..cd62c8e 100644 --- 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 index 3d04502..aedd9f1 100644 --- 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