【轉】python XML 操作總結(創建、保存和刪除，支持utf-8和gb2312)

原文地址：http://hi.baidu.com/tbjmnvbagkfgike/item/6743ab10af43bb24f6625cc5

最近寫程序需要用到xml操作，看了看python.org上面的幾個xml類庫，還是一頭霧水，感覺太學術化了，都那么吝惜寫幾個例子。所以自己整理了一下，算是個小總結，和大家分享一下吧。

對於簡單的操作xml文件來說，xml.dom.minidom足以，可以寫可以讀的。

先給出示例程序，然后簡單注釋一下

1.示例程序：

-----------------------------------------------------------------------------------------------------------------

 

# Author: Nonove. nonove[at]msn[dot]com
# XML simple operation Examples and functions
# encoding = gbk

from xml.dom import minidom
import codecs


def write_xml_file(path, xmlDom, option = {'encoding':'utf-8'}):
    """ Generate xml file with writer
    params:
    string path xml file path
    Dom xmlDom xml dom
    dictionary option writer option {'indent': '', 'addindent':' ', 'newl':'\n', 'encoding':'utf-8'}
    returns:
    bool success return True else False
    """
    defaultOption = {'indent': '', 'addindent':' ', 'newl':'\n', 'encoding':'utf-8'}
    for k, v in defaultOption.iteritems():
        if k not in option:
            option[k] = v

    try:
        f=file(path, 'wb')
        writer = codecs.lookup(option['encoding'])[3](f)
        xmlDom.writexml(writer, encoding = option['encoding'], indent = option['indent'], \
        addindent = option['addindent'], newl = option['newl'])
        writer.close()
        return True
    except:
        print('Write xml file failed.... file:{0}'.format(path))
        return False



if __name__ == "__main__":
    # Create a xml dom
    xmlDom = minidom.Document()
    nonove = xmlDom.createElement('nonove')
    xmlDom.appendChild(nonove)

    # Generate a xml dom
    # Create child node, textnode, set attribute, appendChild
    for i in range(3):
        node = xmlDom.createElement('node')
        node.setAttribute('id', str(i))
        node.setAttribute('status', 'alive')
        textNode = xmlDom.createTextNode('node value ' + str(i))
        node.appendChild(textNode)
        nonove.appendChild(node)

    # Print xml dom
    # Print simple xml
    ## print(xmlDom.toxml())
    # Print pretty xml
    print('\n' + xmlDom.toprettyxml(indent=' '))

    # Save xml file with encoding utf-8
    option = {'indent': '', 'addindent':'', 'newl':'', 'encoding':'utf-8'}
    write_xml_file('nonove.xml', xmlDom, option)

    # Load xml dom from file
    xmlDom = minidom.parse('nonove.xml')
    # Get nonove node
    nonove = xmlDom.getElementsByTagName('nonove')[0]
    # Get node list
    nodes = xmlDom.getElementsByTagName('node')
    for node in nodes:
        # Get node attribute id
        nodeid = node.getAttribute('id')
        # Print node id and textnode value
        print('Node id: {0} textnode value: {1}'.format(nodeid, node.firstChild.nodeValue))

    for node in nodes:
        # Set attribute or remove attribute
        node.setAttribute('author', 'nonove')
        node.removeAttribute('status')

    # Remove node 1
    nonove.removeChild(nodes[1])

    print('\n' + xmlDom.toprettyxml(indent=' '))

 

------------------------------------------------------------------------------------------------------------------

2.注釋：

#讀取xml方式有兩種 從文件 和 從字符串
xmlDom = minidom.parse('nonove.xml')
xmlDom = minidom.parseString(xmlstring)

#創建xml dom
主要是用到了minidom.Document()
xmlDom = minidom.Document()

#創建/刪除節點
node = xmlDom.createElement('node')
root.removeChild(node)

#創建Text Node，獲取 node value
textnode = xmlDom.createTextNode('set value here')
value = textnode.nodeValue

#添加/刪除node屬性
node.setAttribute('author', 'nonove')
node.removeAttribute('author')

#保存xml文件
用到了codecs類庫和writexml()函數，例子里面我寫了個函數，把略顯復雜的操作封裝了一下，以后可以方便重用。

write_xml_file(path, xmlDom, option)
path：xml文件保存地址
xmlDom： xml document
option： 是dictionary類型變量，包括縮進和編碼等，這個可以方便的把xml文件按utf-8或者gb2312保存，方便。

3.備注：

關於CharacterData的解析不出來的問題解決辦法：<![CDATA[]]>標簽和兩面的節點不能夠有間隔字符，否則就解析為空。

<nodename><![CDATA[my data data]]></nodename>

 

就先總結了這么多，有什么不對的地方或者更好的方法請賜教啊……

轉載請注明原文地址啊，辛辛苦苦的總結出來的，別人的勞動成果不容易。