6.6 解析和修改XML

問(wèn)題

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

解決方案

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

下面是一個(gè)利用 ElementTree 來(lái)讀取這個(gè)文檔并對(duì)它做一些修改的例子：

>>> 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é)果是一個(gè)像下面這樣新的XML文件：

討論

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

如果你需要?jiǎng)?chuàng)建新的元素，可以使用本節(jié)方案中演示的 Element 類。我們?cè)?.5小節(jié)已經(jīng)詳細(xì)討論過(guò)了。