pyqt5 保存QTreeWidget的结构到xml文件

PyQt5中的`QTreeWidget`是一个用于显示树状数据的控件，如果想要将它的结构保存为XML文件，可以按照以下步骤操作：

1. 首先，你需要遍历`QTreeWidget`的内容，获取每个节点的信息。这包括节点的数据、子节点以及它们的位置等。

from xml.etree.ElementTree import Element, SubElement, tostring

def qtree_to_xml(tree_widget):
    root = Element('root')
    for index in tree_widget.selectedIndexes():
        node_data = tree_widget.itemFromIndex(index)
        item_element = SubElement(root, 'node', {'id': str(index.row())})
        item_element.text = node_data.text()
        # 如果有子节点，递归处理
        if node_data.childCount() > 0:
            sub_items = qtree_to_xml(tree_widget, node_data)
            item_element.extend(sub_items)

    return [root]

2. 然后创建一个XML字符串，通常会使用`tostring()`函数将其转换为字节串，方便写入文件。

def save_to_xml(file_name, xml_tree):
    xml_string = tostring(xml_tree[0], encoding='utf-8', method='xml')
    with open(file_name, 'w', newline='', encoding='utf-8') as f:
        f.write(xml_string.decode())

3. 最后，调用这两个函数保存结构到XML文件。

xml_tree = qtree_to_xml(your_qtwidget)
save_to_xml('tree_structure.xml', xml_tree)