python - 在 PyQT4 中使用 QTreeWidgetItemIterator 以字典形式从 QTreeWidget 返回 isChecked(或...某物)

Question

检查最终代码的编辑!

所以...我承认，由于缺乏知识，我在这里完全空白，我将展示我的代码并祈祷一下。

将这个奇妙的 xml 用于 QTreeWidget 生成器 ekhumoro编码我已经添加了复选框(父节点的三态)，现在我试图遍历这些复选框并返回字典或列表列表或......带有父节点的东西:子节点(服务器:这棵树中的服务) 关系，以便我可以从树的检查结果中构建文件夹路径。这将在按下“开始”按钮时发生，作为系列中的第一个功能。

最终目标是拥有一个工具，可以根据用户提供的开始和停止时间从平台上的多个服务器和每个服务器的服务中提取日志，并将这些日志传送到具有相同文件夹结构的管理框。

我已经成功地浏览了一个测试字典，搜索并找到了基于 os.path.getctime 和 getmtime 的文件，将它们压缩，然后将它们复制到一个单独的驱动器，该驱动器具有相同的文件夹结构，该驱动器是作为函数的一部分创建的。 ..

但是，我发现几乎没有关于 TreeWidgetItemIterator 的文档(只有 http://pyqt.sourceforge.net/Docs/PyQt4/qtreewidgetitemiterator.html#details )，这对我不是很有帮助。所以这个拼图不可或缺的(也是最后的)部分让我迷路了!

XML 处理程序:

from PyQt4 import QtCore, QtGui, QtXml
from PyQt4.QtXml import *


class XmlHandler(QXmlDefaultHandler):
    def __init__(self, root):
        QtXml.QXmlDefaultHandler.__init__(self)
        self._root = root
        self._item = None
        self._text = ''
        self._error = ''

    def startElement(self, namespace, name, qname, attributes):
        if qname == 'Machine' or qname == 'Feature':
            if self._item is not None:
                self._item = QtGui.QTreeWidgetItem(self._item)
            else:
                self._item = QtGui.QTreeWidgetItem(self._root)
            self._item.setData(0, QtCore.Qt.UserRole, qname)
            self._item.setText(0, 'Unknown Machine')
            if qname == 'Machine':
                self._item.setExpanded(False)
                self._item.setCheckState(0, QtCore.Qt.Unchecked)
                self._item.setFlags(QtCore.Qt.ItemIsSelectable|QtCore.Qt.ItemIsDragEnabled|QtCore.Qt.ItemIsUserCheckable|QtCore.Qt.ItemIsEnabled|QtCore.Qt.ItemIsTristate)
            elif qname == 'FeatureName':
                self._item.setText(1, self._text)
        self._text = ''
        return True

    def endElement(self, namespace, name, qname):
        if qname == 'FeatureName' or qname == 'MachineName':
            if self._item is not None:
                self._item.setText(0, self._text)
                self._item.setCheckState(0, QtCore.Qt.Unchecked)
        elif qname == 'Feature' or qname == 'Machine':
            self._item = self._item.parent()
        return True

    def characters(self, text):
        self._text += text
        return True

    def fatalError(self, exception):
        print('Parse Error: line %d, column %d:\n  %s' % (
              exception.lineNumber(),
              exception.columnNumber(),
              exception.message(),
              ))
        return False

使用XmlHandler 制作widget 的Class:

class MakeWidget(QtGui.QTreeWidget):
    def __init__(self):
        QtGui.QTreeWidget.__init__(self)
        self.header().setResizeMode(QtGui.QHeaderView.Stretch)
        self.setHeaderLabels(['Servers and Services'])
        source = QtXml.QXmlInputSource()
        source.setData(xml)
        handler = XmlHandler(self)
        reader = QtXml.QXmlSimpleReader()
        reader.setContentHandler(handler)
        reader.setErrorHandler(handler)
        reader.parse(source) 

在我的 GUI 中创建小部件:

        self.treeServiceSelection = MakeWidget(xml, parent=self.ui.groupBoxServiceSelection)

widget 是父子两级，仅此而已:

http://i.imgur.com/npPhG41.jpg

现在我们来到了我被困的地方。我可以将信号与按钮按下联系起来，我可以做任何其他事情，但不能从 QTreeWidget 中获取该死的检查项目。看来我需要构建一个迭代器并在其 init 中包含 Checked 标志，但我尝试过的任何东西都出现了 bupkiss。我很乐意为解决方案而努力，而不是有人提供给我，但没有一个起点一直令人沮丧。

编辑:所以我实际发布的内容都没有任何用处，但 ekhumoro 理解我问题的关键并提供了他的解决方案(公认的答案)。我添加了一个 elif 来处理 parent 被检查的任何事情(因为它似乎没有读到 child 被检查是因为这个影响):

def exportTree(self):
    mapping = defaultdict(list)
    root = self.tree.invisibleRootItem()
    for index in range(root.childCount()):
        parent = root.child(index)
        if parent.checkState(0) == QtCore.Qt.PartiallyChecked:
            features = mapping[parent.text(0)]
            for row in range(parent.childCount()):
                child = parent.child(row)
                if child.checkState(0) == QtCore.Qt.Checked:
                    features.append(child.text(0))
        elif parent.checkState(0) == QtCore.Qt.Checked:
            features = mapping[parent.text(0)]
            for row in range(parent.childCount()):
                features.append((parent.child(row)).text(0))
    return mapping

然后能够使用此示例根据 gui 中其他地方的另一个组合框选择特定的子项:

def checkbox2mods(self):
    d = {'DMD2K8COR2VM': ['CC2DCP', 'Centercord'],
         'DMD2K8COR1VM': ['CC2DCP', 'Centercord']}
    root = self.tree.invisibleRootItem()
    if self.checkBox2.checkState() == QtCore.Qt.Checked:
        for index in range(root.childCount()):
            parent = root.child(index)
            for row in range(parent.childCount()):
                child = parent.child(row)
                x = d.values()
                y = d.keys()
                for _ in x:
                    if child.text(0) in _:
                        if parent.text(0) in y:
                            child.setCheckState(0, QtCore.Qt.Checked)
                            parent.setExpanded(True)

我确定代码很糟糕，但通过仅选择子项，它会部分检查父项，这允许导出功能正常工作。

Answer 1

我假设你的问题的核心是你想遍历一个树形小部件并构建一个包含选中项目的字典。

下面的示例假设树有两层深，并且不尝试对检查状态进行任何管理:

# not needed for python 3
import sip
sip.setapi('QString', 2)

from collections import defaultdict
from PyQt4 import QtGui, QtCore

class Window(QtGui.QWidget):
    def __init__(self):
        QtGui.QWidget.__init__(self)
        self.tree = QtGui.QTreeWidget(self)
        self.tree.header().hide()
        for index in range(5):
            parent = QtGui.QTreeWidgetItem(self.tree, ['NUS2K%s' % index])
            if index % 3:
                parent.setCheckState(0, QtCore.Qt.PartiallyChecked)
            else:
                parent.setCheckState(0, QtCore.Qt.Unchecked)
            features = 'Loader Reports Logging'.split()
            for count, item in enumerate(features):
                child = QtGui.QTreeWidgetItem(parent, [item])
                if index % 3 and count % 3:
                    child.setCheckState(0, QtCore.Qt.Checked)
                else:
                    child.setCheckState(0, QtCore.Qt.Unchecked)
            parent.setExpanded(True)
        self.button = QtGui.QPushButton('Export', self)
        self.button.clicked.connect(self.handleExport)
        layout = QtGui.QVBoxLayout(self)
        layout.addWidget(self.tree)
        layout.addWidget(self.button)

    def handleExport(self):
        mapping = self.exportTree()
        for machine, features in mapping.items():
            print('%s:' % machine)
            for feature in features:
                print('  %s' % feature)

    def exportTree(self):
        mapping = defaultdict(list)
        root = self.tree.invisibleRootItem()
        for index in range(root.childCount()):
            parent = root.child(index)
            if parent.checkState(0) == QtCore.Qt.PartiallyChecked:
                features = mapping[parent.text(0)]
                for row in range(parent.childCount()):
                    child = parent.child(row)
                    if child.checkState(0) == QtCore.Qt.Checked:
                        features.append(child.text(0))
        return mapping

if __name__ == '__main__':

    import sys
    app = QtGui.QApplication(sys.argv)
    window = Window()
    window.setGeometry(800, 300, 300, 300)
    window.show()
    sys.exit(app.exec_())