ElementTree处理简单XML

# -*- coding:gb2312 -*-
import os
# 最好使用C实现的哪种，速度快，消耗内存少
try:
import xml.etree.cElementTree as ET # use
except ImportError:
import xml.etree.ElementTree as ET

g_path_XML = '../../XML/'

class XmlTest(object):
def __init__(self):
self.tree = None #（树）
self.root = None # 根节点，root是一个Element元素

# 从创建文件夹->xml->tree
def startSetPath(self, s_dir=None):
'''
更新XML存放路径，并开始新的存放记录
参数：
s_dir: 具体到路径
'''
if s_dir:
global g_path_XML
g_path_XML = os.path.abspath(s_dir)
else:
g_path_XML = os.path.abspath(g_path_XML)
# 如果文件夹存在，则根据文件夹数量新建，+1.xml
if os.path.exists(g_path_XML):
g_path_XML = os.path.join(g_path_XML, str(len(os.listdir(g_path_XML))+1)+'.xml')
# 第一次肯定新建文件夹，包含'1.xml'
else:
os.mkdir(g_path_XML)
g_path_XML = os.path.join(g_path_XML, '1.xml')
self.tree = ET.ElementTree()       #
self.root = ET.Element('root')    # 新创建一个元素,根节点，root是一个Element元素
self.tree._setroot(self.root)           # def _setroot(self, element): 注意：参数element是一个元素

# 获取XML内容
def getXmlContent(self, s_file=None):
'''
参数：
s_file: 具体到某个文件
'''
if s_file:
global g_path_XML
g_path_XML = os.path.abspath(s_file)
else:
g_path_XML = os.path.join(os.path.abspath(g_path_XML), '1.xml')
if os.path.exists(g_path_XML):
# tree = ET.ElementTree('root', g_path_XML)
# ElementTree 有两个paramer, 第一个是根节点，第二个是需解析的xml文件
# 加载并解析这个XML，形成一个tree(树)
self.tree = ET.ElementTree(file=g_path_XML)
self.root = self.tree.getroot()
# ET.dump(tree)
else:
print "路径%s不存在" % g_path_XML

def writeXml(self):
# tree.write(sys.stdout) # 写入缓存
self.tree.write(g_path_XML, 'UTF-8') # 写入XML

# 删除XML节点
def delXmlNode(self):
for child_of_root in self.tree.iter(tag='Basic'):
# del child_of_root[4]  # 删除节点
# set可以修改和增加元素中的（字典中的值）
child_of_root[0].set('xionda', '888')
child_of_root[0].set('Value', '111')

# 增加新XML节点
def addNewXmlNode(self):
# 增加  <My><Person Age="24" Name="liuyc" Sex="1" /></My>
a = ET.SubElement(self.root, 'My')
b = ET.SubElement(a, 'Person')
b.set('Name', 'liuyc')
b.set('Sex', 'man')
b.set('Age', '24')

def creatNewTree(self):
'''
新建一棵树
<new_root><elem_a><child01>c: some thing</child01><child02 /></elem_a><elem_b>b: some text</elem_b></new_root>
'''
a = ET.Element('elem_a')            # 新建一个树的节点a
c = ET.SubElement(a, 'child01')     # a点增加一个新节点
c.text = 'c: some thing'           # c节点中间增加文体
d = ET.SubElement(a, 'child02')     # a增加第二个节点
b = ET.Element('elem_b')            # 新建和a同级的节点
b.text = 'b: some text'
new_root = ET.Element('new_root')
new_root.extend((a, b))              # （a,b）合并到new_root树节点
new_tree = ET.ElementTree(new_root)  # 增加一颗树
ET.dump(new_tree)

def printXml(self):
print '----------print_xml-----------'
i_num = 0
for child_of_root in self.tree.iter():
i_num += 1
print child_of_root.tag, child_of_root.attrib
# print "[%s] (Name=%s) (Type=%s) (Value=%s)" % \
#        (i_num, child_of_root.attrib['Name'], child_of_root.attrib['Type'], child_of_root.attrib['Value'])

if __name__ == "__main__":
o_xml_test = XmlTest()
o_xml_test.startSetPath(r'D:\note\Test\DEAL_XML\MY_TEST\test_xml')
o_xml_test.addNewXmlNode()
o_xml_test.writeXml()
o_xml_test.printXml()

o_xml_test.getXmlContent(r'D:\note\Test\DEAL_XML\MY_TEST\test_xml\1.xml')
o_xml_test.printXml()