python minidom 处理XML

原文出处 http://hi.baidu.com/tbjmnvbagkfgike/item/6743ab10af43bb24f6625cc5

'''
Created on Dec 12, 2013

@author: bzhux
'''
from xml.dom import minidom
import codecs
import traceback


def write_xml_file(path, xmlDom, option={'encoding':'utf-8'}):
    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')
#         print codecs.lookup(option['encoding'])[3]
#         return
        writer  = codecs.lookup(option['encoding'])[3](f)
        xmlDom.writexml(writer, encoding=option['encoding'], indent=option['indent'],\
                        addindent=option['addindent'], newl=option['newl'])
        writer.close()
        f.close()
        return True
    except:
        print 'Failed'
        traceback.print_exc()
        return False

def main():
    xmlDom = minidom.Document()
    root = xmlDom.createElement('root')
    xmlDom.appendChild(root)
    
    for i in range(1,4):
        node = xmlDom.createElement('person')
        node.setAttribute('id', str(i))
        node.setAttribute('a','b')
        root.appendChild(node)
        for j in range(1, 3):
            subNode = xmlDom.createElement('name')
            nodeValue = "Node%d" %(j)
            textNode = xmlDom.createTextNode(nodeValue)
            subNode.appendChild(textNode)
            node.appendChild(subNode)
            

    xmlDom.toprettyxml(indent='')
    option = {'indent':'', 'newl':'\n', 'encoding':'utf-8'}
    write_xml_file('test.xml', xmlDom, option)

if __name__ == '__main__':
#     write_xml_file(None, None, option={'encoding':'utf-9'})
    main()

生成的xml如下

<?xml version="1.0" encoding="utf-8"?>
<root>
    <person a="b" id="1">
        <name>Node1</name>
        <name>Node2</name>
    </person>
    <person a="b" id="2">
        <name>Node1</name>
        <name>Node2</name>
    </person>
    <person a="b" id="3">
        <name>Node1</name>
        <name>Node2</name>
    </person>
</root>