python通过ElementTree操作XML获取结点读取属性美化XML
1.引入库
需要用到3个类,elementtree,element以及建立子类的包装类subelement
from xml.etree.elementtree import elementtree
from xml.etree.elementtree import element
from xml.etree.elementtree import subelement as se
2.读入并解析
tree = elementtree(file=xmlfile)
root = tree.getroot()
读入后,tree是elementtree的类型,获取xml根结点使用getroot()方法;
xml示例文件:
<item sid='1712' name = '大cc' >
<a id=1></a>
<a id=2></a>
</item>
3.获取儿子结点
查找element的所有子结点:
aarry = item.findall('a')
也可使用getchildren():
childs = item.getchildren()
for subitem in childs:
print subitem.get('id')
4.插入儿子结点
方法一:
item = element("item", {'sid' : '1713', 'name' : 'ityouhui'})
root.append(item)
方法二:
se(root,'item',{'sid':'1713','name':'ityouhui'})
法一的好处是插入之后可以对item继续操作。法二是写法上简单,其中se就是subelement,在引入处做了声明;
5.操作属性
获取element的某个属性值(eg:获取item的 name)
print root.find('item/name').text
print item.get('name')
获取element所有属性
print item.items() # [('sid', '1712'), ('name', '大cc')]
print item.attrib # {'sid': '1712', 'name': '大cc'}
6.美化xml
在写入之前,传入root调用此函数,写入的xml文件格式整齐美观:
indent(root)
book.write(xmlfile,'utf-8')
## get pretty look
def indent( elem, level=0):
i = "\n" + level*" "
if len(elem):
if not elem.text or not elem.text.strip():
elem.text = i + " "
for e in elem:
indent(e, level+1)
if not e.tail or not e.tail.strip():
e.tail = i
if level and (not elem.tail or not elem.tail.strip()):
elem.tail = i
return elem
上一篇: 一个简单的python程序实例(通讯录)
下一篇: Python创建xml的方法