【发布时间】：2013-12-16 19:06:56
【问题描述】：
检查最终代码的编辑！
所以...我承认由于缺乏知识，我在这里画了一个绝对空白，我只是展示我的代码并祈祷一下。
使用ekhumoro 编码的这个奇妙的 xml 到 QTreeWidget 生成器，我已在复选框（父节点的三态）中添加，现在我正在尝试遍历这些复选框并返回字典或列表列表或.. .something 与父：子（服务器：此树中的服务）关系，以便我可以根据树的检查结果构建文件夹的路径。这将在按下“开始”按钮时发生，作为系列中的第一个功能。
最终目标是拥有一个工具，该工具可以根据用户提供的开始和停止时间从平台上的多个服务器和每个服务器的服务中提取日志，并将这些日志传送到具有相同文件夹结构的管理框。
我已经成功地浏览了一个测试字典，根据 os.path.getctime 和 getmtime 搜索和定位文件，将它们压缩，然后将它们复制到一个单独的驱动器中，该驱动器具有作为函数的一部分创建的相同文件夹结构。 ..
但是，我发现几乎没有关于 TreeWidgetItemIterator 的文档（只有 http://pyqt.sourceforge.net/Docs/PyQt4/qtreewidgetitemiterator.html#details），这对我没有多大帮助。所以这个谜题的组成部分（也是最后一部分）让我迷失了！
XMLHandler：
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 制作小部件的类：
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)
小部件有父母和孩子两层，仅此而已：
http://i.imgur.com/npPhG41.jpg
现在我们已经到了我被困的地方。我可以将信号与按钮按下联系起来，我可以做其他所有事情，但从 QTreeWidget 中获取已检查的项目。看来我需要构造一个迭代器并在它的 init 中包含 Checked 标志，但我尝试过的任何东西都想出了 bupkiss。我很乐意找到解决方案，而不是为我提供解决方案，但没有起点一直令人沮丧。
编辑：所以我实际发布的所有内容都没有任何用处，但 ekhumoro 了解我的问题的症结并提供了他的解决方案（接受的答案）。我添加了一个elif 来处理检查父母的任何事情（因为它似乎没有读到孩子被检查作为这样的效果）：
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)
我确信代码很糟糕，但是通过只选择孩子，它会部分检查父母，这允许导出功能正常工作。
【问题讨论】：
标签： python pyqt pyqt4 qtreewidget qtreewidgetitem