Search code examples
pythonpyqt4qtreewidgetitem

Get contents from Qtable which is a child of QTreeWidget


I've created a treewidget with children which contain tables. I'd like to acces the contents of the QtableWidget but I cannot find how to do this?

The treewidget looks like: treewidget with tables as children

I've generated the treewidget like:

        software = QTreeWidgetItem(['Software'])
        hardware = QTreeWidgetItem(['Hardware'])
        beide = QTreeWidgetItem(['Beide'])
        andere = QTreeWidgetItem(['Andere'])          
        i = 0
        for key, value in sorted(data.items()):
                if value['Subtype'] == 'Software':
                    sub = software
                if value['Subtype'] == 'Hardware':
                    sub = hardware
                if value['Subtype'] == 'Beide':
                    sub = beide
                if value['Subtype'] == 'Andere':
                    sub = andere

                l1 = QTreeWidgetItem(sub)

                if value['Privacy'] == 'Voorzichtig':
                    l1.setBackgroundColor(0, QColor('orange'))
                if value['Privacy'] == 'Vertrouwelijk':
                    l1.setBackgroundColor(0, QColor('red'))
                    l1.setTextColor(0, QColor('white'))

                l1.setText(0, value['sDesc'])

                self.treeMainDisplay.addTopLevelItem(l1)

                l1_child = QTreeWidgetItem(l1)

                self.item_table = QTableWidget()
                self.item_table.verticalHeader().setVisible(False)
                self.item_table.horizontalHeader().setVisible(False)
                self.item_table.setColumnCount(5)
                self.item_table.setRowCount(5)
                c1_item = QTableWidgetItem("%s" % value['sDesc'].encode('utf-8'))
                self.item_table.setItem(0, 0, c1_item)
                c2_item = QTableWidgetItem("%s" % value['Type'].encode('utf-8'))
                self.item_table.setItem(1,0, c2_item)

                qt_child = self.treeMainDisplay.setItemWidget(l1_child, 0,  self.item_table)  

        self.treeMainDisplay.addTopLevelItem(software)
        self.treeMainDisplay.addTopLevelItem(hardware)
        self.treeMainDisplay.addTopLevelItem(beide)
        self.treeMainDisplay.addTopLevelItem(andere)

I'm iterating over the treewidgetitems but don't know how to access the table contents:

def testItems(self):
    iterator = QTreeWidgetItemIterator(self.treeMainDisplay)
    while iterator.value():
        item = iterator.value()
        if not item.text(0):
            #Get Table Object?
            # item.item(0,0).text()
        else:
            print item.text(0)
        iterator += 1

It seems I can't get acces to the QTableWidget object, I only get the QTreeWidgetItem object.

All feedback is highly appreciated!


Solution

  • The item widgets must be access via the tree-widget using the itemWidget method:

    def testItems(self):
        iterator = QTreeWidgetItemIterator(self.treeMainDisplay)
        while iterator.value():
            item = iterator.value()
            if not item.text(0):
                # Get Table Object
                table = self.treeMainDisplay.itemWidget(item, 0)
            else:
                print item.text(0)
            iterator += 1