在 Python 中,可以使用 xml.etree.ElementTree
模块来读取和编辑 XML 文件。下面是一个例子,演示如何编辑 XML 文件中的文本字段并保存更改。
1、问题背景
我有一个包含一些数据的 XML 文件,如下所示:
<?xml version="1.0" encoding="UTF-8" ?>
- <ParameterData><CreationInfo date="10/28/2009 03:05:14 PM" user="manoj" />
- <ParameterList count="85">
- <Parameter name="Spec 2 Included" type="boolean" mode="both"><Value>n/a</Value> <Result>n/a</Result> </Parameter>
- <Parameter name="Spec 2 Label" type="string" mode="both"><Value>n/a</Value> <Result>n/a</Result> </Parameter>
- <Parameter name="Spec 3 Included" type="boolean" mode="both"><Value>n/a</Value> <Result>n/a</Result> </Parameter>
- <Parameter name="Spec 3 Label" type="string" mode="both"><Value>n/a</Value> <Result>n/a</Result> </Parameter></ParameterList></ParameterData>
我还有一份以分行形式存储键值对的文本文件,如下所示:
Spec 2 Included : TRUE
Spec 2 Label: 19-Flat2-HS3
Spec 3 Included : FALSE
Spec 3 Label: 4-1-Bead1-HS3
我想使用 Python 将 XML 文件中的字段值(n/a)替换为文本文件中的相应值,使 XML 文件看起来像这样:
<?xml version="1.0" encoding="UTF-8" ?>
- <ParameterData><CreationInfo date="10/28/2009 03:05:14 PM" user="manoj" />
- <ParameterList count="85">
- <Parameter name="Spec 2 Included" type="boolean" mode="both"><Value>TRUE</Value> <Result>TRUE</Result> </Parameter>
- <Parameter name="Spec 2 Label" type="string" mode="both"><Value>19-Flat2-HS3</Value> <Result>19-Flat2-HS3</Result> </Parameter>
- <Parameter name="Spec 3 Included" type="boolean" mode="both"><Value>FALSE</Value> <Result>FALSE</Result> </Parameter>
- <Parameter name="Spec 3 Label" type="string" mode="both"><Value>4-1-Bead1-HS3</Value> <Result>4-1-Bead1-HS3</Result> </Parameter></ParameterList></ParameterData>
2、解决方案
以下是用 Python 实现的解决方案:
import xml.etree.ElementTree as ET
import re# 解析 XML 文件
tree = ET.parse('input.xml')
root = tree.getroot()# 读取文本文件中的键值对
with open('values.txt', 'r') as f:values = dict(re.findall('(Spec \d+ (?:Included|Label))\s*:\s*(\S+)', f.read()))# 修改 XML 文件中的字段值
for parameter in root.findall('ParameterList/Parameter'):name = parameter.get('name')if name in values:parameter.find('Value').text = values[name]parameter.find('Result').text = values[name]# 保存修改后的 XML 文件
tree.write('output.xml')
这个解决方案使用 ElementTree 库来解析 XML 文件,并使用正则表达式来读取文本文件中的键值对。然后,它迭代 XML 文件中的 Parameter 元素,并使用 values 字典来查找每个 Parameter 的新值。最后,它将修改后的 XML 文件写入一个新的文件中。
以下是一些可以使用的代码示例:
# 读取 XML 文件并创建一个 ElementTree 对象
tree = ET.parse('input.xml')# 获取根元素
root = tree.getroot()# 找到所有名为 'Parameter' 的子元素
parameters = root.findall('ParameterList/Parameter')# 遍历 'Parameter' 元素并打印它们的 'name' 和 'Value' 属性
for parameter in parameters:print(parameter.get('name'), parameter.find('Value').text)
这个代码示例将从 input.xml 文件中读取所有名为 ‘Parameter’ 的元素,并打印它们的 ‘name’ 和 ‘Value’ 属性。
# 创建一个新的 ElementTree 对象并添加一个根元素
tree = ET.ElementTree()
root = ET.Element('root')
tree._setroot(root)# 向根元素添加一个子元素
child = ET.SubElement(root, 'child')# 设置子元素的 'name' 和 'value' 属性
child.set('name', 'my_name')
child.text = 'my_value'# 将 ElementTree 对象保存到一个 XML 文件
tree.write('output.xml')
这个代码示例将创建一个新的 ElementTree 对象,并添加一个根元素。然后,它向根元素添加一个子元素,并设置子元素的 ‘name’ 和 ‘value’ 属性。最后,它将 ElementTree 对象保存到一个 XML 文件中。
注意事项
- 保存编码:
tree.write()
的encoding="utf-8"
参数指定保存的文件编码为 UTF-8,这在处理非 ASCII 字符时很重要。 - 备份文件:在编辑 XML 文件前,建议先备份文件,以防修改错误。
这样,你可以轻松地编辑 XML 文件中的文本字段并保存更改。