PythonElementTree：条件提取XML属性技巧

编程并不是一个机械性的工作，而是需要有思考，有创新的工作，语法是固定的，但解决问题的思路则是依靠人的思维，这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《Python ElementTree：条件提取XML属性方法》，文章讲解的知识点主要包括，如果你对文章方面的知识点感兴趣，就不要错过golang学习网，在这可以对大家的知识积累有所帮助，助力开发能力的提升。

1. 引言：XML数据提取与条件拼接的挑战

在日常开发中，处理XML数据是一项常见任务。xml.etree.ElementTree是Python标准库中一个轻量级且高效的XML解析模块。本教程将聚焦于一个具体的场景：从复杂的XML结构中提取元素的name属性，并根据其内部元素是否包含groups属性，来决定是否将groups的值与name属性进行拼接。例如，如果groups属性存在，我们希望得到inter_name-groupA, inter_name-groupB这样的结果；如果不存在，则只保留inter_name。

2. 示例XML结构分析

为了更好地理解问题，我们先来看一下示例XML的结构：

<abtshop>
    <inters>
        <inter name="nim_turk" first-day="20230301" historical="20220103" market="multi">
            <works>
                <work kind="obopay" run="jbs">
                    <!-- 无 groups 属性 -->
                </work>
            </works>
        </inter>
        <inter name="nim_us_m" first-day="20230301" historical="20220103" market="lone">
            <works>
                <work kind="obopay" run="jbs" groups="groupA,groupB">
                    <!-- 包含 groups 属性 -->
                </work>
                <work kind="silkb" run="jbs" groups="groupA,groupB">
                    <!-- 包含 groups 属性 -->
                </work>
            </works>
        </inter>
    </inters>
</abtshop>

我们的目标是从元素中获取name属性。对于包含子元素且该元素带有groups属性的，我们需要将name与groups中的每个值进行拼接。

3. 使用ElementTree解析XML

xml.etree.ElementTree模块提供了一套直观的API来解析和操作XML文档。

3.1 导入模块与加载XML

首先，我们需要导入xml.etree.ElementTree并加载XML文件。

import xml.etree.ElementTree as ET

# 假设XML内容保存在 'test_xml.xml' 文件中
tree = ET.parse("test_xml.xml")
root = tree.getroot() # 获取XML文档的根元素

或者，如果XML是字符串，可以使用ET.fromstring()。

4. 解决方案策略与实现

核心思路是遍历所有相关的元素，针对每个元素，首先获取其name属性。然后，在其内部查找是否存在带有groups属性的子元素。根据查找结果，执行不同的拼接逻辑。

4.1 遍历目标元素

我们首先需要定位到所有元素。由于它们嵌套在标签下，我们可以先找到，然后遍历其所有子元素。

result = [] # 用于存储最终结果的列表

# 遍历所有 <inters> 元素（通常只有一个）
for inters_elem in root.findall('.//inters'):
    # 遍历 <inters> 下的所有 <inter> 子元素
    for inter_elem in inters_elem:
        # 获取当前 <inter> 元素的 name 属性
        inter_name = inter_elem.get('name')

        # ... 后续处理逻辑 ...

4.2 条件判断与属性提取

在获取到inter_name之后，我们需要判断当前inter_elem内部是否有带有groups属性的元素。Element.find()方法非常适合这个场景，因为它会返回第一个匹配的子元素，如果没有找到则返回None。

XPath表达式".//work[@groups]"的含义是：

• .//：从当前元素（inter_elem）的任意深度子孙节点中查找。
• work：查找标签名为work的元素。
• [@groups]：筛选出带有groups属性的work元素。

        # 在当前 <inter> 元素内部查找第一个带有 'groups' 属性的 <work> 元素
        work_with_groups = inter_elem.find(".//work[@groups]")

        if work_with_groups is not None:
            # 如果找到了带有 'groups' 属性的 <work> 元素
            groups_str = work_with_groups.get('groups') # 获取 groups 属性值，例如 "groupA,groupB"
            group_list = groups_str.split(',')          # 按逗号分割成列表

            # 遍历每个组名，并与 inter_name 拼接
            for group_name in group_list:
                combined_name = inter_name + '-' + group_name
                result.append(combined_name)
        else:
            # 如果没有找到带有 'groups' 属性的 <work> 元素，则只添加 inter_name
            result.append(inter_name)

4.3 完整代码示例

将上述步骤整合起来，形成完整的解决方案：

import xml.etree.ElementTree as ET

# 假设XML内容保存在 'test_xml.xml' 文件中
# 创建一个示例XML文件用于测试
xml_content = """
<abtshop>
    <dDirectory>dub</dDirectory>
    <S>statusd</S>
    <work>worklogs</work>
    <custs>
        <cust>nim-us</cust>
    </custs>
    <mileage>999</mileage>
    <defaults>
        <default type="mercley">
            <user>dairy</user>
            <exec>slm.sh</exec>
            <env>
                <var name="SAN_HOME">youyou-11</var>
            </env>
        </default>
    </defaults>
    <inters>
        <inter name="nim_turk" first-day="20230301" historical="20220103" market="multi">
            <works>
                <work kind="obopay" run="jbs">
                    <args>
                        <arg name="distance">180000</arg>
                    </args>
                </work>
                <work kind="silkb" run="jbs">
                    <args>
                        <arg name="distance">180000</arg>
                    </args>
                </work>
            </works>
        </inter>
        <inter name="nim_us_m" first-day="20230301" historical="20220103" market="lone">
            <works>
                <work kind="obopay" run="jbs" groups="groupA,groupB">
                    <args>
                        <arg name="distance">120000</arg>
                        <arg name="jbsopt">xmas_size=1200000</arg>
                        <arg name="jbsopt">of_obopaying_threads=2</arg>
                    </args>
                </work>
                <work kind="silkb" run="jbs" groups="groupA,groupB">
                    <args>
                        <arg name="distance">120000</arg>
                        <arg name="jbsopt">xmas_size=1200000</arg>
                    </args>
                </work>
            </works>
        </inter>
    </inters>
</abtshop>
"""

# 将XML内容写入文件
with open("test_xml.xml", "w", encoding="utf-8") as f:
    f.write(xml_content)

# 解析XML文件
root = ET.parse("test_xml.xml").getroot()

result = [] # 存储最终结果的列表

# 遍历所有 <inters> 元素（通常只有一个，但使用 findall 更具通用性）
for inters_elem in root.findall('.//inters'):
    # 遍历 <inters> 元素下的所有 <inter> 子元素
    for inter_elem in inters_elem:
        inter_name = inter_elem.get('name') # 获取 <inter> 元素的 name 属性值

        # 在当前 <inter> 元素内部查找第一个带有 'groups' 属性的 <work> 元素
        work_with_groups = inter_elem.find(".//work[@groups]")

        if work_with_groups is not None:
            # 如果找到了带有 'groups' 属性的 <work> 元素
            groups_str = work_with_groups.get('groups') # 获取 groups 属性值
            group_list = groups_str.split(',')          # 按逗号分割成列表

            # 遍历每个组名，并与 inter_name 拼接
            for group_name in group_list:
                combined_name = inter_name + '-' + group_name
                result.append(combined_name)
        else:
            # 如果没有找到带有 'groups' 属性的 <work> 元素，则只添加 inter_name
            result.append(inter_name)

print(result)

预期输出：

['nim_turk', 'nim_us_m-groupA', 'nim_us_m-groupB']

5. ElementTree关键方法解析

• ET.parse(source): 解析一个XML文件（或文件类对象），并返回一个ElementTree对象。
• ElementTree.getroot(): 返回XML树的根元素。
• Element.findall(match): 查找当前元素下所有匹配XPath表达式的子元素（包括任意深度的子孙元素），并返回一个列表。
• Element.find(match): 查找当前元素下第一个匹配XPath表达式的子元素（包括任意深度的子孙元素），如果未找到则返回None。
• Element.get(key, default=None): 获取元素的指定属性值。如果属性不存在，则返回default值（默认为None）。
• XPath表达式:
  • tag：匹配指定标签名的元素。
  • ./tag：匹配当前元素的直接子元素tag。
  • .//tag：匹配当前元素的任意深度子孙元素tag。
  • [@attribute]：匹配带有指定属性的元素。
  • [@attribute='value']：匹配带有指定属性且属性值等于value的元素。

6. 注意事项与最佳实践

• 错误处理: 在实际应用中，应考虑文件不存在、XML格式错误等异常情况，使用try-except块进行处理。
• XPath的精确性: 编写XPath表达式时应尽量精确，避免匹配到不期望的元素，尤其是在大型或结构复杂的XML文档中。
• 性能考量: 对于非常大的XML文件，ElementTree会一次性加载整个文档到内存。如果内存是瓶颈，可以考虑使用iterparse进行增量解析。
• 命名约定: 在代码中，变量名如inter_elem、work_with_groups等，能够清晰地表达其代表的XML元素，提高代码可读性。
• 属性不存在: Element.get('attribute_name')在属性不存在时会返回None，这比直接访问Element.attrib['attribute_name']更安全，后者在属性不存在时会抛出KeyError。

7. 总结

本教程演示了如何利用Python的xml.etree.ElementTree模块高效地解析XML数据，并根据特定条件（是否存在groups属性）从不同层级的元素中提取和拼接数据。通过灵活运用findall()、find()和get()等方法，结合XPath表达式，我们可以精确地定位目标数据并实现复杂的逻辑处理，从而满足多样化的XML数据处理需求。

本篇关于《PythonElementTree：条件提取XML属性技巧》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于文章的相关知识，请关注golang学习网公众号！