Rewrite QTreeWidget >>> QTreeView with checkbox?

Question

I do:

  class Window(QtGui.QWidget):

    def __init__(self):
        QtGui.QWidget.__init__(self)
        self.modelTree = QtGui.QTreeView()
        self.model = QtGui.QStandardItemModel()
        self.addItems(self.model, data)
        self.modelTree.setModel(self.model)
        self.modelTree.connect(self.modelTree, QtCore.SIGNAL('clicked(QModelIndex)'), self.treefunction)
    def treefunction(self, index):
        print index.model().itemFromIndex(index).text()

        '''if item.checkState(column) == QtCore.Qt.Checked:
            print "checked", item.text(column)
        if item.checkState(column) == QtCore.Qt.Unchecked:
            print "NOT checked",item.text(column)'''

    def addItems(self, parent,elements):
        column = 0
        clients_item = QtGui.QTreeWidgetItem(parent, ['Serwis'])
        clients_item.setData(column, QtCore.Qt.UserRole, 'serwis 111')
        clients_item.setExpanded(True)


        item_1 = QtGui.QTreeWidgetItem(clients_item, ['Wartswa1'])
        item_1.setData(column, QtCore.Qt.UserRole, 'Wasrtwas 1')
        item_1.setCheckState(column, QtCore.Qt.Unchecked)

        item_2 = QtGui.QTreeWidgetItem(clients_item, ['Wartswa2'])
        item_2.setData(column, QtCore.Qt.UserRole, 'Wasrtwas 2')
        item_2.setCheckState(column, QtCore.Qt.Unchecked)

I want get info if row is checked or unchecked. I found a example with this but for the QTreeWidget. I look for w way to do this with QTreeView. How to rewrite treefunction?

Answer 1

What you're looking for is Qt::CheckStateRole.

In treefunction, do:

if self.model.data(index, Qt.CheckStateRole) == Qt.Checked:
    print 'Checked!'
elif self.model.data(index, Qt.CheckStateRole) == Qt.Unchecked:
    print 'Unchecked!'