解析 XML 元素的父/子关系

Question

给定以下 XML（ant build xml）：

    <project name="pj1">
        <target name="t1">
            ...
            <antcall target="t2"/>
            <a>
                <antcall target="t4"/>
            </a>
            ...
        </target>
        <target name="t2">
            ...
            <antcall target="t3"/>
            ...
        </target>
        <target name="t3">
            ...
            ...
        </target>
        <target name="t4">
            ...
            <antcall target="t2"/>
            ...
        </target>
        <target name="t5">
            ...
            ...
        </target>
    </project>

我想显示目标元素的父/子关系如下（如果目标嵌套在另一个目标中，则不将目标显示为第一级元素）

t1
    t2
        t3
    t4
        t2
            t3
t5

有人可以帮忙吗？

提前致谢。

Answer 1

当我需要将 XML 树操作为其他表示时，我发现先将其转换为抽象表示，然后再转换为最终的具体表示很有用。

在这种情况下，首先我们创建一个表示目标依赖结构的列表字典，然后我们漂亮地打印该字典。

#!/usr/bin/python

import xml.etree.ElementTree as ET
from itertools import chain

def parse(filename):
    tree = ET.parse(filename)
    root = tree.getroot()
    result = {}
    for target in root.findall('target'):
        target_name = target.get('name')
        result[target_name] = []
        for antcall in target.findall('.//antcall'):
            result[target_name].append(antcall.get('target'))
    return result

def display(tree):
    def recurse(node, indent):
        print "%*s%s"%(indent*4, "", node)
        for node in sorted(tree[node]):
            recurse(node, indent+1)
    for item in sorted(tree):
        if item in chain(*tree.values()): continue
        recurse(item,0)

if __name__=="__main__":
    import argparse
    parser = argparse.ArgumentParser(description='Dump ANT files')
    parser.add_argument('antfile',
                        nargs='+',
                        type=argparse.FileType('r'),
                        help='ANT build file')
    args = parser.parse_args()
    for antfile in args.antfile:
        display(parse(antfile))