Python - 如何确定解析的 XML 元素的层次结构级别？

Question

我正在尝试使用 Python 从 XML 文件中解析具有特定标记的元素并生成输出 excel 文档，该文档将包含元素并保留它们的层次结构。

我的问题是我无法弄清楚每个元素（解析器迭代的）嵌套的深度。

XML 样本提取（3 个元素，它们可以任意嵌套在自己内部）：

<A>
   <B>
      <C>
      </C>
   </B>
</A>
<B>
    <A>
    </A>
</B>

以下代码使用 ElementTree 可以很好地迭代元素。但我认为 ElementTree 无法确定每个元素的嵌套深度。见下文：

import xml.etree.ElementTree as ET

root = ET.parse('XML_file.xml')
tree = root.getroot()
for element in tree.iter():
    if element.tag in ("A","B","C"):
        print(element.tag)

这将为我提供按正确顺序排列的元素 A、B、C 的列表。但我需要将它们打印出来，并附上他们的等级信息，

不仅如此：

A
B
C
B
A

但是类似：

A
--B
----C
B
--A

为了能够做到这一点，我需要获取每个元素的级别。有没有合适的python解析器可以轻松做到这一点？我会想象像“element.hierarchyLevel”这样会返回一些整数索引的东西......

Answer 1

你可以使用xml.sax.saxhandler:

import xml.sax as sax
import xml.sax.handler as saxhandler

class TreeBuilder(saxhandler.ContentHandler):
    # http://docs.python.org/library/xml.sax.handler.html#contenthandler-objects
    def __init__(self):
        self.level = 0
    def startElement(self, name, attrs):
        print('--'*self.level + name)
        self.level += 1
    def endElement(self, name):
        self.level -= 1

builder = TreeBuilder()
src = '''\
<root>
<A>
   <B>
      <C>
      </C>
   </B>
</A>
<B>
    <A>
    </A>
</B>
</root>
'''
sax.parseString(src, builder)

产量

root
--A
----B
------C
--B
----A

Answer 2

尝试使用递归函数来跟踪您的“级别”。

import xml.etree.ElementTree as ET

def perf_func(elem, func, level=0):
    func(elem,level)
    for child in elem.getchildren():
        perf_func(child, func, level+1)

def print_level(elem,level):
    print '-'*level+elem.tag

root = ET.parse('XML_file.xml')
perf_func(root.getroot(), print_level)