写了个把 xml 的 tag 解析目录树的小脚本， 各位大佬提点意见

求问， 各位大佬有没有更好的实现方式

"""解析 xml 的结构为目录树，能够快速的对 xml 的结构有基本的了解。这也是 nonlocal 新关键字的一个 demo
"""
from io import BytesIO

from lxml import etree


def init_xpath(page_source: str):
    """page_source 的 xml 文本转为可以解析的 etree 对象
    """
    xml_root = etree.parse(BytesIO(page_source.encode()))
    return xml_root


def fmt(fg):
    """格式化输出"""
    print("-" * fg, end="")


def tree_xml(root, result, flags=0, step=2):
    result.append((flags, root))

    def tree(root):
        nonlocal flags
        ch_root = root.getchildren()
        if ch_root:
            flags += step
            for ch in ch_root:
                tree_xml(ch, result, flags)
        else:
            pass
    tree(root)
    return result


def main():
    data = """
    <xml>
        <aa>
            <bb></bb>
            <cc>
                <a11></a11>
                <a22>
                    <mue></mue>
                </a22>
            </cc>
            <dd></dd>
        </aa>
        <ee>
            <ff></ff>
        </ee>
    </xml>
    """
    xml_root = init_xpath(data)
    res_xml = tree_xml(xml_root.getroot(), [])
    for fg, _root in res_xml:
        fmt(fg)
        print(_root.tag)


if __name__ == '__main__':
    main()

执行结果是这样的

xml
--aa
----bb
----cc
------a11
------a22
--------mue
----dd
--ee
----ff