主页 / coding / 问题 / 78845737
Accepted
Floren
Asked: 2024-08-08 05:27:05 +0800 CST

使用 ElementTree 将 XML 文件转换为字典

  • 772

我有一个旧版软件使用的 XML 配置文件，我无法更改或格式化它。目标是使用 Python 3.9 并仅使用xml.etree.ElementTree库将 XML 文件转换为字典。

我最初正在看这个回复，它几乎产生了预期的结果。

Scenario.xml文件内容：

<Scenario Name="{{ env_name }}">
    <Gateways>
        <Alpha Host="{{ host.alpha_name }}" Order="1">
           <Config>{{ CONFIG_DIR }}/alpha.xml</Config>
           <Arguments>-t1 -t2</Arguments>
        </Alpha>
        <Beta Host="{{ host.beta_name }}" Order="2">
           <Config>{{ CONFIG_DIR }}/beta.xml</Config>
           <Arguments>-t1</Arguments>
        </Beta>
        <Gamma Host="{{ host.gamma_name }}" Order="3">
           <Config>{{ CONFIG_DIR }}/gamma.xml</Config>
           <Arguments>-t2</Arguments>
           <!--<Data Count="58" />-->
        </Gamma>
    </Gateways>
</Scenario>

将 XML 文件转换为字典的 Python 代码：

from pprint import pprint
from xml.etree import ElementTree


def format_xml_to_dictionary(element: ElementTree.Element):
    '''
    Format xml to dictionary

    :param element: Tree element
    :return: Dictionary formatted result
    '''
    try:
        return {
            **element.attrib,
            '#text': element.text.strip(),
            **{i.tag: format_xml_to_dictionary(i) for i in element}
        }
    except ElementTree.ParseError as e:
        raise e

if __name__ == '__main__':
    tree = ElementTree.parse('Scenario.xml').getroot()
    scenario = format_xml_to_dictionary(tree)
    pprint(scenario)

带注释的功能代码输出<!--<Data Count="58" />-->：

$ python test.py
{'#text': '',
 'Gateways': {'#text': '',
              'Alpha': {'#text': '',
                        'Arguments': {'#text': '-t1 -t2'},
                        'Config': {'#text': '{{ CONFIG_DIR }}/alpha.xml'},
                        'Host': '{{ host.alpha_name }}',
                        'Order': '1'},
              'Beta': {'#text': '',
                       'Arguments': {'#text': '-t1'},
                       'Config': {'#text': '{{ CONFIG_DIR }}/beta.xml'},
                       'Host': '{{ host.beta_name }}',
                       'Order': '2'},
              'Gamma': {'#text': '',
                        'Arguments': {'#text': '-t2'},
                        'Config': {'#text': '{{ CONFIG_DIR }}/gamma.xml'},
                        'Host': '{{ host.gamma_name }}',
                        'Order': '3'}},
 'Name': '{{ env_name }}'}

我正在尝试解决两个问题：

  1. Scenario字典键中缺少，因为根节点已经是标签Scenario，我不确定我需要做什么，才能使其成为字典的一部分
  2. 如果我取消注释<Data Count="58" />，我会收到以下错误：
AttributeError: 'NoneType' object has no attribute 'strip'

我不确定我需要实现什么类型的 if/else 条件，我尝试了类似的方法，但它将所有#text值设置为''而不是剥离它们：

'#text': element.text.strip() if isinstance(
    element.text, ElementTree.Element
) else '',
python

1 个回答

  1. Best Answer

    要Scenario进入结果，请tree.tag在调用函数时使用最外层字典中的键。

    要处理没有文本的节点，请#text在单独的语句中将键添加到字典中，以便它可以是条件的。

    def format_xml_to_dictionary(element: ElementTree.Element):
    '''
    Format xml to dictionary

    :param element: Tree element
    :return: Dictionary formatted result
    '''
    try:
        result = {
            **element.attrib,
            **{i.tag: format_xml_to_dictionary(i) for i in element}
        }
        if element.text:
            result['#text'] = element.text.strip()
        return result
    except ElementTree.ParseError as e:
        raise e

if __name__ == '__main__':
    tree = ElementTree.parse('Scenario.xml').getroot()
    scenario = {
        tree.tag: format_xml_to_dictionary(tree)
    }
    pprint(scenario)
    • 3

相关问题