32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
|
from PyQt4.QtGui import *
|
||
|
|
from PyQt4.QtCore import *
|
||
|
|
|
||
|
|
BB = QDialogButtonBox
|
||
|
|
|
||
|
|
class ColorDialog(QColorDialog):
|
||
|
|
def __init__(self, parent=None):
|
||
|
|
super(ColorDialog, self).__init__(parent)
|
||
|
|
self.setOption(QColorDialog.ShowAlphaChannel)
|
||
|
|
# The Mac native dialog does not support our restore button.
|
||
|
|
self.setOption(QColorDialog.DontUseNativeDialog)
|
||
|
|
## Add a restore defaults button.
|
||
|
|
# The default is set at invocation time, so that it
|
||
|
|
# works across dialogs for different elements.
|
||
|
|
self.default = None
|
||
|
|
self.bb = self.layout().itemAt(1).widget()
|
||
|
|
self.bb.addButton(BB.RestoreDefaults)
|
||
|
|
self.bb.clicked.connect(self.checkRestore)
|
||
|
|
|
||
|
|
def getColor(self, value=None, title=None, default=None):
|
||
|
|
self.default = default
|
||
|
|
if title:
|
||
|
|
self.setWindowTitle(title)
|
||
|
|
if value:
|
||
|
|
self.setCurrentColor(value)
|
||
|
|
return self.currentColor() if self.exec_() else None
|
||
|
|
|
||
|
|
def checkRestore(self, button):
|
||
|
|
if self.bb.buttonRole(button) & BB.ResetRole and self.default:
|
||
|
|
self.setCurrentColor(self.default)
|
||
|
|
|