'''
Created on 15/06/2018

@author: rjag008
'''

try:
    from PySide import QtCore, QtGui
except ImportError:
    #from PyQt4 import QtCore, QtGui
    pass
    
class PaintDatasetItem(QtGui.QListWidgetItem):
        
    def __init__(self, key, name,color):
        QtGui.QListWidgetItem.__init__(self, None)

        self._key = key
        self._name = name
        self.color = color
        self.setText(name)
        self.setFlags(self.flags() | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsUserCheckable)
        self.setCheckState(QtCore.Qt.Checked) # initialize check state
        self.setBackground(color)
        
    def changeColor(self):
        color = QtGui.QColorDialog.getColor(self.color)
        if color.isValid():
            self.setBackground(color)
            self.color = color
            return color
        return None