PyQT4 TreeModel example

[annndrey]

Hello! I've got a question about implementation of a standart tree model for TreeView.

I have a response from database that lools like

[
[A, [a,b,c]],
[B,[a,b,c]],
.
.
.
[N,[a,b,c]]
]

And I want to pass it through treemodel to treeview, for in GUI it looks like that:

A
|
+---a
+---b
+---c
|
B
+
|
+---a
+---b
+---c

I've took a look at example, that comes with PyQT4, but in that example they retrieve data from a text file, where tree structure is marked by tabs. and I cannot realize how I can modify this standart example for it works with my data...

I'll be thankful for any help...

[annndrey]

Finally I've fount a solution!

(Idea taken from here: http://www.mail-archive.com/[e-mail]/msg19400.html)

class TreeModel(QtCore.QAbstractItemModel):  
        def __init__(self, data, parent=None):  
            super(TreeModel, self).__init__(parent)  
            self.parents=[]  
            self.dbdata = data  
            self.rootItem = TreeItem([u"NameOfColumn"])  
            self.setupModelData(self.dbdata, self.rootItem)  
      
        def setData(self, index, value, role):  
           if index.isValid() and role == QtCore.Qt.EditRole:  
     
               prev_value = self.getValue(index)  
     
               item = index.internalPointer()  
     
               item.setData(unicode(value.toString()))  
     
               return True  
           else:  
               return False  
     
       def removeRows(self, position=0, count=1,  parent=QtCore.QModelIndex()):  
     
           node = self.nodeFromIndex(parent)  
           self.beginRemoveRows(parent, position, position + count - 1)  
           node.childItems.pop(position)  
           self.endRemoveRows()  
     
       def nodeFromIndex(self, index):  
           if index.isValid():  
               return index.internalPointer()  
           else:  
               return self.rootItem  
     
       def getValue(self, index):  
           item = index.internalPointer()  
           return item.data(index.column())  
     
       def columnCount(self, parent):  
           if parent.isValid():  
               return parent.internalPointer().columnCount()  
           else:  
               return self.rootItem.columnCount()  
     
       def data(self, index, role):  
           if not index.isValid():  
               return None  
           if role != QtCore.Qt.DisplayRole:  
               return None  
     
           item = index.internalPointer()  
           return QtCore.QVariant(item.data(index.column()))  
     
      def flags(self, index):  
           if not index.isValid():  
               return QtCore.Qt.NoItemFlags  
     
           return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable  
     
       def headerData(self, section, orientation, role):  
           if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:  
               return QtCore.QVariant(self.rootItem.data(section)[0])  
     
           return None  
     
       def index(self, row, column, parent):  
     
           if row < 0 or column < 0 or row >= self.rowCount(parent) or column >= self.columnCount(parent):  
               return QtCore.QModelIndex()  
     
           if not parent.isValid():  
               parentItem = self.rootItem  
           else:  
               parentItem = parent.internalPointer()  
     
           childItem = parentItem.child(row)  
           if childItem:  
               return self.createIndex(row, column, childItem)  
           else:  
               return QtCore.QModelIndex()  
     
       def parent(self, index):  
           if not index.isValid():  
               return QtCore.QModelIndex()  
     
           childItem = index.internalPointer()  
           parentItem = childItem.parent()  
     
           if parentItem == self.rootItem:  
               return QtCore.QModelIndex()  
     
           return self.createIndex(parentItem.row(), 0, parentItem)  
     
       def rowCount(self, parent):  
           if parent.column() > 0:  
               return 0  
     
           if not parent.isValid():  
               parentItem = self.rootItem  
           else:  
               parentItem = parent.internalPointer()  
    
           return parentItem.childCount()  
     
       def setupModelData(self, lines, parent):  
           ind = []  
           self.parents.append(parent)  
           ind.append(0)  
           col_numb=parent.columnCount()  
           numb = 0  
     
           for line in lines:  
               numb+=1  
               lineData=line[0]  
               self.parents[-1].appendChild(TreeItem(lineData, self.parents[-1]))  
     
               columnData = line[1]  
    
               self.parents.append(self.parents[-1].child(self.parents[-1].childCount() - 1))  
    
               for j in columnData:  
                    self.parents[-1].appendChild(TreeItem(j, self.parents[-1]))  
               if len(self.parents) > 0:  
                    self.parents.pop()  
    
  class TreeItem(object):  
      def __init__(self, data, parent=None):  
          self.parentItem = parent  
          self.itemData = data  
          self.childItems = []  
    
      def appendChild(self, item):  
          self.childItems.append(item)  
    
      def child(self, row):  
          return self.childItems[row]  
    
      def childCount(self):  
          return len(self.childItems)  
    
      def columnCount(self):  
          return len(self.itemData)  
    
      def data(self, column):  
          try:  
              return self.itemData  
   
          except IndexError:  
              return None  
    
      def parent(self):  
          return self.parentItem  
    
      def row(self):  
          if self.parentItem:  
              return self.parentItem.childItems.index(self)  
    
          return 0  
      def setData(self, data):  
          self.itemData = data

[annndrey]

Now it works but I still don't know how to add a support for several columns in this model...

[Odyssey2001]

Could someone explain to me how setupModelData works??

[vegaseat]

Another example ...
http://rowinggolfer.blogspot.com/2010/05/qtreeview-and-qabractitemmodel-example.html