6.6 解析和修改XML

問題

你想讀取一個XML文檔，對它最一些修改，然后將結(jié)果寫回XML文檔。

解決方案

使用 xml.etree.ElementTree 模塊可以很容易的處理這些任務。第一步是以通常的方式來解析這個文檔。例如，假設(shè)你有一個名為 pred.xml 的文檔，類似下面這樣：

下面是一個利用 ElementTree 來讀取這個文檔并對它做一些修改的例子：

>>> from xml.etree.ElementTree import parse, Element
>>> doc = parse('pred.xml')
>>> root = doc.getroot()
>>> root
<Element 'stop' at 0x100770cb0>

>>> # Remove a few elements
>>> root.remove(root.find('sri'))
>>> root.remove(root.find('cr'))
>>> # Insert a new element after <nm>...</nm>
>>> root.getchildren().index(root.find('nm'))
1
>>> e = Element('spam')
>>> e.text = 'This is a test'
>>> root.insert(2, e)

>>> # Write back to a file
>>> doc.write('newpred.xml', xml_declaration=True)
>>>

處理結(jié)果是一個像下面這樣新的XML文件：

討論

修改一個XML文檔結(jié)構(gòu)是很容易的，但是你必須牢記的是所有的修改都是針對父節(jié)點元素，將它作為一個列表來處理。例如，如果你刪除某個元素，通過調(diào)用父節(jié)點的 remove() 方法從它的直接父節(jié)點中刪除。如果你插入或增加新的元素，你同樣使用父節(jié)點元素的 insert() 和 append() 方法。還能對元素使用索引和切片操作，比如 element[i] 或 element[i:j]

如果你需要創(chuàng)建新的元素，可以使用本節(jié)方案中演示的 Element 類。我們在6.5小節(jié)已經(jīng)詳細討論過了。