Python中XML的读写总结

  • 基于DOM
    • 写入
    • 读取
  • 基于ElementTree
    • 写入
    • 读取

近期项目有用到python处理较多的xml数据,在这里总结下pythonxml的读写方法。在python中,主要有三种方法读写xml:基于DOM,基于ELementTree,基于SAX,这里主要总结前两种方式。

基于DOM

导入: import xml.dom.minidom as minidom

写入

创建文档:
dom = minidom.getDOMImplementation().createDocument(None,'Root',None)
获得根节点:root = dom.documentElement
创建节点:element = dom.createElement('Name')
给这个节点添加文本:element.appendChild(dom.createTextNode('default'))
注意:这里的节点文本值是存成另外一个节点的,是createTextNode
设置属性:element.setAttribute('age', '20')
添加到节点:root.appendChild(element)
完整代码:

# -*- coding:utf-8 -*-
import xml.dom.minidom as minidom
dom = minidom.getDOMImplementation().createDocument(None,'Root',None)
root = dom.documentElement

for i in range(5):
    element = dom.createElement('Name')
    element.appendChild(dom.createTextNode('default'))
    element.setAttribute('age', str(i))
    root.appendChild(element)
# 保存文件
with open('default.xml', 'w', encoding='utf-8') as f:
    dom.writexml(f, addindent='\t', newl='\n',encoding='utf-8')

# 文档内容
"1.0" encoding="utf-8"?>

    "0">default
    "1">default
    "2">default
    "3">default
    "4">default

读取

读取文档:dom = minidom.parse('default.xml')
获得根节点:root = dom.documentElement
按照名称查找子节点,注意这里会递归查找所有子节点:names = root.getElementsByTagName('Name')
所有的子节点:root.childNodes
注意:每个节点的文本值存在TextNode节点中,也就是最后一个节点的第一个子节点
查看是否含有属性:name.hasAttribute('age')
查看属性:name.getAttribute('age')
完整代码:

dom = minidom.parse('default.xml')
root = dom.documentElement
names = root.getElementsByTagName('Name')
for name in names:
    # 它的第一个子节点是一个textnode,存取的是真正的节点值
    print(name.childNodes[0].nodeValue, end='\t')
    if name.hasAttribute('age'):
        print(name.getAttribute('age'), end='\t')
    print('')
# 输出
default 0   
default 1   
default 2   
default 3   
default 4   

基于ElementTree

导入:import xml.etree.ElementTree as ET

写入

创建节点:root = ET.Element('Root')
创建文档:tree = ET.ElementTree(root)
设置文本值:element.text = 'default'
设置属性:element.set('age', str(i))
添加节点:root.append(element)
写入文档:tree.write('default.xml', encoding='utf-8', xml_declaration=True)

但这样写入会有个问题,写入的XML会在同一行,缺少换行符,Etree本身并没有提供换行的选项,翻看了国外大神的回答,对root处理以后,再次写入将有换行符。完整代码如下:

# -*- coding:utf-8 -*-

import xml.etree.ElementTree as ET
# 增加换行符
def __indent(elem, level=0):
    i = "\n" + level*"\t"
    if len(elem):
        if not elem.text or not elem.text.strip():
            elem.text = i + "\t"
        if not elem.tail or not elem.tail.strip():
            elem.tail = i
        for elem in elem:
            __indent(elem, level+1)
        if not elem.tail or not elem.tail.strip():
            elem.tail = i
    else:
        if level and (not elem.tail or not elem.tail.strip()):
            elem.tail = i

root = ET.Element('Root')       # 创建节点
tree = ET.ElementTree(root)     # 创建文档

for i in range(5):
    element = ET.Element('Name')
    element.set('age', str(i))
    element.text = 'default'
    root.append(element)

__indent(root)          # 增加换行符
tree.write('default.xml', encoding='utf-8', xml_declaration=True)

# 文档内容
'1.0' encoding='utf-8'?>

    "0">default
    "1">default
    "2">default
    "3">default
    "4">default

读取

读取文档:tree = ET.parse('default.xml')
获得根节点:root = tree.getroot()
获得所有子节点:list(root)
查找子节点,注意这里不会递归查找所有子节点:root.findall('Name')
查找子节点,递归查找所有子节点:root.iter('Name')
查看节点名称:root.tag

全部代码:

# -*- coding:utf-8 -*-

import xml.etree.ElementTree as ET

tree = ET.parse('default.xml')
root = tree.getroot()
for node in list(root):
    print(node.text, node.tag, node.get('age'))

for node in root.findall('Name'):
    print(node.text, node.tag, node.get('age'))

# 输出
default Name 0
default Name 1
default Name 2
default Name 3
default Name 4
default Name 0
default Name 1
default Name 2
default Name 3
default Name 4

如有错误,欢迎指正~

你可能感兴趣的:(python,日常总结)