一个使用ElementTree的xml添加元素,删除,更新,读取的类

yangward

在搜索引擎找到很多python操作xml的文章，但都不完整，我这里这个整理了一个，希望大家给点意见  
problem.xml

1. <?xml version='1.0' encoding='UTF-8'?>
   
2. <book>
   
3. <date>2010.3.10</date>
   
4. <version>Three</version>
   
5. <dbcm_version>good</dbcm_version></book>

复制代码

dbcm_xml.py

1. 
   
2. #!/usr/bin/env python
   
3. #coding=UTF-8
   
4. #用于某个xml的属性值
   
5. import os,sys,xml.etree.ElementTree as ET
   
6. class DbcmXml:
   
7.         def __init__(self,filePath):
   
8.                 self.filePath = filePath
   
9.                 print '文件路径 %s' %filePath
   
10.                 print ET.parse(filePath)
    
11.                 self.xmlFile = os.path.abspath(__file__)
    
12.                 self.xmlFile = os.path.dirname(self.xmlFile)
    
13.                 self.xmlFile = os.path.join(self.xmlFile,self.filePath)
    
14.                 try:
    
15.                         self.dom = ET.parse(filePath)
    
16.                         self.root = self.dom.getroot()
    
17.                 except Exception,inst:
    
18.                         print "打开%s 文件时候遇到错误" %self.filePath
    
19.         ##输出某个属性的text内容
    
20.         def getElementText(self,elementName):
    
21.                 self.elementName = elementName
    
22.                 if self.root.find(self.elementName) ==None:
    
23.                         print ' %s None' %self.elementName
    
24.         ## 更新某个已经存在的元素的text内容
    
25.         def updateElementText(self,elementName,textValue):
    
26.                 for self.item in self.root:
    
27.                         self.oldElement = self.root.find(elementName)
    
28.                         self.root.remove(self.oldElement)
    
29.                         self.newElement = ET.SubElement(self.root,elementName)
    
30.                         self.newElement.text = textValue
    
31.                 tree = ET.ElementTree(self.root)
    
32.                 tree.write("problem.xml","UTF-8")
    
33.         ## 添加某个元素进去
    
34.         def addElement(self,elementName,textValue):
    
35.                 if self.root.find(elementName) ==None:
    
36.                         child = ET.SubElement(self.root,elementName)
    
37.                         child.text = textValue
    
38.                 try:
    
39.                         tree = ET.ElementTree(self.root)
    
40.                         tree.write("problem.xml",encoding="UTF-8")
    
41.                 except        Exception,inst:
    
42.                         print "写入文件出错 %s " %inst
    
43.         ##删除某个元素
    
44.         def delElement(self,elementName,textValue):
    
45.                 #print self.root.find(elementName).text
    
46.                 for self.item in self.root:
    
47.                         self.oldElement == elementName
    
48.                         self.oldElement = self.root.find(elementName)
    
49.                         if self.oldElement !=None:
    
50.                                 self.root.remove(self.oldElement)
    
51.                                 tree = ET.ElementTree(self.root)
    
52.                                 tree.write("problem.xml","UTF-8")
    
53.                         #self.newElement = ET.SubElement(self.root,elementName)
    
54.                         #self.newElement.text = textValue
    
55.                 tree = ET.ElementTree(self.root)
    
56.                 tree.write("problem.xml","UTF-8")
    
57. 
    
58. test = DbcmXml('problem.xml')
    
59. test.getElementText('dbcm_version')
    
60. test.updateElementText('dbcm_version','good')
    
61. #test.addElement('google','man')
    
62. test.delElement('google','man')

复制代码