如何用递归构建带缩进的嵌套XML结构-Python教程

本文介绍如何将任务关系列表递归转换为结构清晰、缩进美观的嵌套xml，重点解决“末级子节点丢失”和“缩进不生效”两大常见问题，并提供高效字典索引与`xml.etree.elementtree.indent()`的完整实践方案。

在Python中，将扁平的任务依赖列表（如 (parent_id, parent_type, parent_desc, child_id, child_type, child_desc) 元组序列）转化为深度嵌套的XML，需兼顾逻辑正确性与输出可读性。原始代码存在两个关键缺陷：

1. 末级节点缺失：当某任务的 child_id 在原始列表中无对应父记录（即无后续子节点）时，递归终止且未生成该叶子节点；
2. 缩进未启用：ET.tostring() 默认不格式化，即使调用 indent() 也需显式传入 space 和 level 参数并确保使用 Python 3.9+。

以下为优化后的完整解决方案：

import xml.etree.ElementTree as ET

tasks = [
    ('task1', 'Type1', 'Description1', 'task11', 'Type11', 'Description11'),
    ('task2', 'Type2', 'Description2', 'task22', 'Type22', 'Description22'),
    ('task11', 'Type11', 'Description11', 'task33', 'Type33', 'Description33'),
    ('task33', 'Type33', 'Description33', 'task3', 'Type3', 'Description3'),
    ('task3', 'Type3', 'Description3', 'task5', 'Type5', 'Description5'),
    ('task4', 'Type4', 'Description4', 'task6', 'Type6', 'Description6'),
    ('task6', 'Type6', 'Description6', 'task7', 'Type7', 'Description7'),
    ('task7', 'Type7', 'Description7', 'task8', 'Type8', 'Description8'),
    ('taskX', 'TypeX', 'DescriptionX', 'task33', 'Type33', 'Description33'),
]

def create_task_element(task_id, task_type, task_desc, is_related=False):
    """创建基础 task 元素，支持主节点与关联节点属性名区分"""
    elem = ET.Element('task')
    if is_related:
        elem.set('taskIdRelated', task_id)
        elem.set('taskIdRelatedType', task_type)
        elem.set('taskIdRelatedDescription', task_desc)
    else:
        elem.set('taskId', task_id)
        elem.set('taskIdType', task_type)
        elem.set('taskIdDescription', task_desc)
    return elem

def build_xml_tree(task_dict, task_tuple):
    """
    递归构建XML树：
    - 若 task_tuple 存在，则创建主节点，并尝试查找其子节点；
    - 若子节点不存在于字典中（即叶子节点），仍创建空关联节点（含全部6字段中的后3个）；
    - 避免无限递归：通过字典O(1)查表替代列表遍历。
    """
    if not task_tuple:
        return None

    # 创建当前层级主节点（使用前3字段）
    current_elem = create_task_element(*task_tuple[:3], is_related=False)

    # 获取子任务ID（第4字段）
    child_id = task_tuple[3]
    child_task = task_dict.get(child_id)

    # 构建子节点：若存在则递归，否则用当前元组后3字段创建叶子节点
    if child_task:
        child_elem = build_xml_tree(task_dict, child_task)
    else:
        # 关键修复：即使 child_id 不在字典中，也要生成叶子节点
        child_elem = create_task_element(*task_tuple[3:], is_related=True)

    current_elem.append(child_elem)
    return current_elem

def find_root_tasks(task_dict):
    """找出所有根节点：即 taskId 不作为任何其他任务的 taskIdRelated 的任务"""
    all_ids = set(task_dict.keys())
    related_ids = {t[3] for t in task_dict.values()}
    root_ids = all_ids - related_ids
    return [task_dict[tid] for tid in root_ids]

# 主流程：预构建字典 → 查找根节点 → 逐个构建并格式化输出
task_dict = {t[0]: t for t in tasks}
root_tasks = find_root_tasks(task_dict)

for root_task in root_tasks:
    root_elem = build_xml_tree(task_dict, root_task)
    # 启用缩进（Python 3.9+）
    ET.indent(root_elem, space="    ", level=0)  # 使用4空格缩进
    print(ET.tostring(root_elem, encoding='unicode'))

✅ 核心改进点说明：