52 lines
2.1 KiB
Python
52 lines
2.1 KiB
Python
import os
|
||
import xml.etree.ElementTree as ET
|
||
|
||
def remove_tag_from_element(element, tag):
|
||
"""
|
||
递归遍历element的所有子节点，删除所有匹配指定标签的子节点。
|
||
"""
|
||
for child in list(element): # 用 list() 避免在遍历时修改子节点列表导致问题
|
||
if child.tag == tag:
|
||
element.remove(child)
|
||
else:
|
||
remove_tag_from_element(child, tag)
|
||
|
||
def remove_tag_from_xml(xml_file, tag_to_remove, output_file):
|
||
"""
|
||
解析单个 XML 文件，删除其中所有指定的标签，然后保存为新的 XML 文件。
|
||
"""
|
||
try:
|
||
tree = ET.parse(xml_file)
|
||
root = tree.getroot()
|
||
# 如果根节点本身就是要删除的标签，则提示并跳过处理
|
||
if root.tag == tag_to_remove:
|
||
print(f"警告：{xml_file} 的根标签就是要删除的标签({tag_to_remove})，跳过该文件。")
|
||
return
|
||
|
||
remove_tag_from_element(root, tag_to_remove)
|
||
tree.write(output_file, encoding="utf-8", xml_declaration=True)
|
||
print(f"处理完成: {xml_file} -> {output_file}")
|
||
except Exception as e:
|
||
print(f"处理 {xml_file} 时出错: {e}")
|
||
|
||
def batch_remove_tag(input_dir, output_dir, tag_to_remove):
|
||
"""
|
||
批量处理：遍历 input_dir 下所有 XML 文件，删除指定标签后，将新文件保存到 output_dir。
|
||
"""
|
||
if not os.path.exists(output_dir):
|
||
os.makedirs(output_dir)
|
||
|
||
for file in os.listdir(input_dir):
|
||
if file.lower().endswith(".xml"):
|
||
input_path = os.path.join(input_dir, file)
|
||
output_path = os.path.join(output_dir, file)
|
||
remove_tag_from_xml(input_path, tag_to_remove, output_path)
|
||
|
||
if __name__ == "__main__":
|
||
# 请修改以下路径和要删除的标签名称
|
||
input_directory = r"C:\Users\10561\Downloads\Compressed\xml" # 输入 XML 文件所在目录
|
||
output_directory = "path_to_output_xml_directory" # 处理后 XML 文件保存目录
|
||
tag_to_delete = "segmented" # 例如：删除 <segmented> 标签，可改为你需要删除的标签名
|
||
|
||
batch_remove_tag(input_directory, output_directory, tag_to_delete)
|