Python XML标签替换脚本仅匹配部分实例问题求助

问题描述 #

我编写了一段Python脚本，旨在查找XML文档中的指定<line>标签并替换为更具描述性的新标签。但运行后发现脚本仅能匹配到部分目标文本实例，即便使用精确匹配文本也无法解决该问题。相关脚本及示例XML文档如下：

原脚本 #

import xml.etree.ElementTree as ET
from lxml import etree

def replace_specific_line_tags(input_file, output_file, replacements):
    # Parse the XML file using lxml
    tree = etree.parse(input_file)
    root = tree.getroot()

    for target_text, replacement_tag in replacements:
        # Find all <line> tags with the specific target text under <content> and replace them with the new tag
        for line_tag in root.xpath('.//content/page/line[contains(., "{}")]'.format(target_text)):
            parent = line_tag.getparent()

            # Create the new tag with the desired tag name
            new_tag = etree.Element(replacement_tag)

            # Copy the attributes of the original <line> tag to the new tag
            for attr, value in line_tag.attrib.items():
                new_tag.set(attr, value)

            # Copy the text of the original <line> tag to the new tag
            new_tag.text = line_tag.text

            # Replace the original <line> tag with the new tag
            parent.replace(line_tag, new_tag)

    # Write the updated XML back to the file
    with open(output_file, 'wb') as f:
        tree.write(f, encoding='utf-8', xml_declaration=True)

if __name__ == '__main__':
    input_file_name = 'beforeTagEdits.xml'
    output_file_name = 'afterTagEdits.xml'
    
    # List of target texts and their corresponding replacement tags
    replacements = [
        ('The Washington Post', 'title'),

        # Add more target texts and their replacement tags as needed
    ]
    
    replace_specific_line_tags(input_file_name, output_file_name, replacements)

示例XML文档 #

<root>
     <content>
          <line>The Washington Post</line>
          <line>The Washington Post</line>
     </content>
</root>

问题原因分析 #

• XPath路径错误：脚本中使用的XPath表达式.//content/page/line[contains(., "{}")]多了一层/page节点，但示例XML里<line>直接隶属于<content>，这会导致所有目标节点无法被匹配；如果实际XML中部分<line>在<page>下、部分不在，就会出现仅匹配到部分实例的情况。
• 匹配逻辑不精准：contains(., "目标文本")会匹配所有包含目标文本的节点，而非精确匹配；若目标文本前后存在缩进、换行等空白字符，也会导致匹配失败。
• 冗余导入：导入了xml.etree.ElementTree as ET但未实际使用，属于无效代码。

修复后的脚本 #

from lxml import etree

def replace_specific_line_tags(input_file, output_file, replacements):
    tree = etree.parse(input_file)
    root = tree.getroot()

    for target_text, replacement_tag in replacements:
        # 修正XPath路径，使用normalize-space处理空白后实现精确匹配
        xpath_expr = f'.//content/line[normalize-space(.) = "{target_text}"]'
        for line_tag in root.xpath(xpath_expr):
            parent = line_tag.getparent()
            new_tag = etree.Element(replacement_tag)
            
            # 复制原标签的所有属性
            for attr, value in line_tag.attrib.items():
                new_tag.set(attr, value)
            
            # 保留原标签的文本内容
            new_tag.text = line_tag.text
            
            parent.replace(line_tag, new_tag)

    with open(output_file, 'wb') as f:
        tree.write(f, encoding='utf-8', xml_declaration=True)

if __name__ == '__main__':
    input_file_name = 'beforeTagEdits.xml'
    output_file_name = 'afterTagEdits.xml'
    
    replacements = [
        ('The Washington Post', 'title'),
        # 可添加更多替换规则
    ]
    
    replace_specific_line_tags(input_file_name, output_file_name, replacements)

关键修复点说明 #

• 修正XPath路径：根据实际XML结构，移除多余的/page节点，确保能准确定位到目标<line>标签。
• 精确匹配文本：使用normalize-space(.) = "{target_text}"，normalize-space会自动去除文本前后空白并合并中间空白，避免因XML缩进、换行导致的匹配失败，同时实现精确匹配。
• 精简代码：删除未使用的xml.etree.ElementTree导入，减少冗余依赖。

内容的提问来源于stack exchange，提问作者E-Flo