#Clicking into a color editor segfaults with PySide, but not with PyQt4.

USE_PYSIDE = False
USE_PYSIDE = True

if USE_PYSIDE:
    from PySide import QtCore
    from PySide import QtGui
    from PySide.QtCore import QProperty as pyqtProperty
    qcolor = QtGui.QColor
else:
    import sip
    sip.setapi('QString', 2)
    sip.setapi('QVariant', 2)
    from PyQt4 import QtCore
    from PyQt4 import QtGui
    from PyQt4.QtCore import pyqtProperty
    qcolor = QtCore.QVariant.Color


class ColorListEditor(QtGui.QComboBox):
    def __init__(self, parent):
        QtGui.QComboBox.__init__(self, parent)
        self.populateList()

    def getColor(self):
        return self.itemData(self.currentIndex(),
                                QtCore.Qt.DecorationRole) #.toPyObject()

    def setColor(self, color):
        self.setCurrentIndex(self.findData(color, QtCore.Qt.DecorationRole))

    color = pyqtProperty('QColor', getColor, setColor, user=True)

    def populateList(self):
        for i, name in enumerate(QtGui.QColor.colorNames()):
            self.insertItem(i, name)
            self.setItemData(i, QtGui.QColor(name),
                                QtCore.Qt.DecorationRole)


class ColorListCreator(QtGui.QItemEditorCreatorBase):
    def __init__(self):
        QtGui.QItemEditorCreatorBase.__init__(self)

    def createWidget(self, parent):
        return ColorListEditor(parent)


class Window(QtGui.QWidget):
    def __init__(self):
        QtGui.QWidget.__init__(self)
        factory = QtGui.QItemEditorFactory()
        colorListCreator = ColorListCreator()
        factory.registerEditor(qcolor, colorListCreator)
        QtGui.QItemEditorFactory.setDefaultFactory(factory)
        self.createGUI()

    def createGUI(self):
        data = ((self.tr('Alice'), QtGui.QColor('aliceblue')),
                (self.tr('Neptun'), QtGui.QColor('aquamarine')),
                (self.tr('Ferdinand'), QtGui.QColor('springgreen')))
        table = QtGui.QTableWidget(3, 2)
        table.setHorizontalHeaderLabels(['Name', 'Hair Color'])
        table.verticalHeader().setVisible(False)
        table.resize(150, 50)
        for i, (name, color) in enumerate(data):
            nameItem = QtGui.QTableWidgetItem(name)
            colorItem = QtGui.QTableWidgetItem()
            colorItem.setData(QtCore.Qt.DisplayRole, color)
            table.setItem(i, 0, nameItem)
            table.setItem(i, 1, colorItem)
        table.resizeColumnToContents(0)
        table.horizontalHeader().setStretchLastSection(True)
        layout = QtGui.QGridLayout()
        layout.addWidget(table, 0, 0)
        self.setLayout(layout)
        self.setWindowTitle('Color Editor Factory')


if __name__ == '__main__':

    import sys

    app = QtGui.QApplication(sys.argv)
    window = Window()
    window.show()
    sys.exit(app.exec_())