Python修改XML文件新增节点

引言

XML（可扩展标记语言）是一种常用的数据存储和传输格式，广泛应用于各种领域。在处理XML文件时，经常需要对其进行修改和更新，以满足不同的需求。本文将介绍如何使用Python语言来修改XML文件并新增节点。

XML文件的组成

XML文件由标签（Element）、属性（Attribute）和文本内容（Text）组成。标签用于定义数据的结构，属性用于描述标签的特征，文本内容用于存储实际数据。下面是一个简单的XML文件示例：

<?xml version="1.0" encoding="UTF-8"?>
<bookstore>
    <book category="children">
        <title lang="en">Harry Potter</title>
        <author>J.K. Rowling</author>
        <year>2005</year>
        <price>29.99</price>
    </book>
    <book category="cooking">
        <title lang="en">Cooking 101</title>
        <author>John Smith</author>
        <year>2010</year>
        <price>19.99</price>
    </book>
</bookstore>

以上XML文件描述了一个图书店的书籍信息，每本书都有一个类别（category），包括标题（title）、作者（author）、出版年份（year）和价格（price）等信息。

Python操作XML文件

Python提供了多种操作XML文件的库，如xml.dom、xml.etree.ElementTree等。本文将使用xml.etree.ElementTree库来演示如何修改XML文件并新增节点。

首先，我们需要导入xml.etree.ElementTree库，并使用ElementTree.parse()方法加载XML文件。代码示例如下：

import xml.etree.ElementTree as ET

tree = ET.parse('books.xml')
root = tree.getroot()

以上代码首先导入了xml.etree.ElementTree库，并使用ET.parse()方法加载名为books.xml的XML文件。然后，通过tree.getroot()方法获取XML文件的根节点，即bookstore节点。

修改XML文件节点

接下来，我们可以通过访问节点的子节点、属性和文本内容来修改XML文件。例如，如果我们想要修改第一本书的标题为《Harry Potter and the Chamber of Secrets》，可以使用以下代码：

book = root.find('book')
title = book.find('title')
title.text = 'Harry Potter and the Chamber of Secrets'

以上代码通过root.find()方法找到名为book的子节点，再通过book.find()方法找到名为title的子节点。最后，通过修改title节点的text属性，将标题修改为《Harry Potter and the Chamber of Secrets》。

新增XML文件节点

在修改XML文件时，有时我们需要新增节点来存储额外的信息。下面是一个示例，展示如何向XML文件中新增一本书的信息：

new_book = ET.Element('book', {'category': 'fiction'})
new_title = ET.SubElement(new_book, 'title')
new_title.text = 'The Great Gatsby'
new_author = ET.SubElement(new_book, 'author')
new_author.text = 'F. Scott Fitzgerald'
new_year = ET.SubElement(new_book, 'year')
new_year.text = '1925'
new_price = ET.SubElement(new_book, 'price')
new_price.text = '9.99'

root.append(new_book)

以上代码首先使用ET.Element()方法创建一个名为new_book的新节点，并使用{'category': 'fiction'}定义节点的属性。然后，通过ET.SubElement()方法创建名为new_title、new_author、new_year和new_price的子节点，并分别给它们赋值。最后，通过root.append()方法将新节点添加到根节点中。

将修改后的XML文件保存

在对XML文件进行修改和新增节点后，我们需要将修改后的内容保存到文件中。可以使用ElementTree.write()方法将修改后的XML文件保存到指定的路径。代码示例如下：

tree.write('updated_books.xml')

以上代码将修改后的XML文件保存为updated_books.xml。

完整示例代码

下面是一个完整的示例代码，演示了如何使用Python修改XML文件并新增节点：

import xml.etree.ElementTree as ET

# Load XML file
tree = ET.parse('books.xml')
root = tree.getroot()