day5-xml模块

一、简述

xml即可扩展标记语言，它可以用来标记数据、定义数据类型，是一种允许用户对自己的标记语言进行定义的源语言。它用于不同语言或者程序之间进行数据交换，从这点上讲与json差不多，只不过json看起来更美观、可读性更强。另外json诞生的时间并不是很久，在json出现以前，数据交换只能选择xml，即便是json已经在大面积使用的现在，xml依然被广泛使用，java项目中随处可见啊。

二、xml的结构

先来看一个栗子把：

<?xml version="1.0"?>
<data>
    <country name="Liechtenstein">
        <rank updated="yes">2</rank>
        <year>2008</year>
        <gdppc>141100</gdppc>
        <neighbor name="Austria" direction="E"/>
        <neighbor name="Switzerland" direction="W"/>
    </country>
    <country name="Singapore">
        <rank updated="yes">5</rank>
        <year>2011</year>
        <gdppc>59900</gdppc>
        <neighbor name="Malaysia" direction="N"/>
    </country>
    <country name="Panama">
        <rank updated="yes">69</rank>
        <year>2011</year>
        <gdppc>13600</gdppc>
        <neighbor name="Costa Rica" direction="W"/>
        <neighbor name="Colombia" direction="E"/>
    </country>
</data>

OK,从结构上，它很像我们常见的HTML超文本标记语言。但他们被设计的目的是不同的，超文本标记语言被设计用来显示数据，其焦点是数据的外观。它被设计用来传输和存储数据，其焦点是数据的内容。
结构特征解读如下：

• 它由成对的标签组成 <data></data>
  一级标签称为根节点，其他级别的标签称为节点
• 标签可以有属性 <country name="Panama">
• 标签对可以嵌入数据 <year>2011</year>
  嵌入的数据即为节点的值
• 标签可以嵌入子标签（具有层级关系）

三、通过python操作xml文件

就以上面的xml文件为例来看看怎么通过python操作xml文件把。

3.1 读取xml文件内容

import xml.etree.ElementTree as et
tree = et.parse('test.xml')
root = tree.getroot() #获取根节点
print(root.tag) #打印根节点


for child in root:
    print('-----')
    print('	', child.tag, child.attrib)  #分别打印子节点名称和子节点属性
    #遍历子节点下的所有节点
    for i in child:
        print('		', i.tag, i.text, i.attrib) #打印子节点下节点的节点名、节点值和属性
    #只遍历year节点
    for i in child.iter('year'):
        print('			', i.tag, i.text)

print('')
for node in root.iter('year'):   #从根节点上直接遍历year节点
    print(node.tag, node.text)

程序输出：
data
-----
	 country {'name': 'Liechtenstein'}
		 rank 2 {'updated': 'yes'}
		 year 2008 {}
		 gdppc 141100 {}
		 neighbor None {'name': 'Austria', 'direction': 'E'}
		 neighbor None {'name': 'Switzerland', 'direction': 'W'}
			 year 2008
-----
	 country {'name': 'Singapore'}
		 rank 5 {'updated': 'yes'}
		 year 2011 {}
		 gdppc 59900 {}
		 neighbor None {'name': 'Malaysia', 'direction': 'N'}
			 year 2011
-----
	 country {'name': 'Panama'}
		 rank 69 {'updated': 'yes'}
		 year 2011 {}
		 gdppc 13600 {}
		 neighbor None {'name': 'Costa Rica', 'direction': 'W'}
		 neighbor None {'name': 'Colombia', 'direction': 'E'}
			 year 2011

year 2008
year 2011
year 2011

说明：

1. getroot()用于返回根节点，tag返回节点名，attrib返回节点属性，text返回节点的值

2. 只返回某个节点的信息，使用iter(节点名)即可

3.2 修改xml文件内容

import xml.etree.ElementTree as et
tree = et.parse('test.xml')
root = tree.getroot()

print(type(root.iter('year')))
for node in root.iter('year'):
    new_year = int(node.text) + 1
    node.text = str(new_year) #修改节点值
    node.tag = 'next_year' #修改节点名称
    node.set('Maxwell','handsome') #修改节点属性

tree.write('test2.xml') #保存文件

注意最后一步保存操作不能漏掉！
修改的实际效果如下：

3.3 删除xml节点

import xml.etree.ElementTree as et

tree = et.parse('test.xml')
root = tree.getroot()


for country in root.findall('country'):  #查找第一层子节点
    rank = int(country.find('rank').text) #查找子节点下的子节点
    if rank > 50:
        root.remove(country)  #删除符合条件的节点

tree.write('test2.xml')

注意：

1. findall()从根节点只能根据第一层的子节点名查找，并且返回第一层子节点的内存地址
2. 删除节点用remove()方法
3. 删除后需要write保存

3.4 创建新的xml文件

import xml.etree.ElementTree as et

#创建根节点
new_xml = et.Element('profile')
#创建根节点的第一层子节点，参数依次表示父节点，子节点名称，子节点属性
name = et.SubElement(new_xml, 'name', attrib={'Luzhishen':'HuaHeSang'})
age = et.SubElement(name, 'age', attrib={'adult':'yes'})
#设置子节点的值
age.text = '22'
gender = et.SubElement(name, 'gender')
gender.text = 'man'
#创建第二个根节点的第一层子节点
name2 = et.SubElement(new_xml, 'name', attrib={'WuYong':'Zhiduoxing'})
age2 = et.SubElement(name2, 'age')
age2.text = '23'

#生成新的xml文档
ET = et.ElementTree(new_xml)
#保存文档
ET.write('my.xml', encoding='utf-8', xml_declaration='true')
#打印文档格式
et.dump(new_xml)

创建的xml文档格式：

可以看出与一般的xml文件相比就差缩进了，不过不影响数据交换啦。

注意：

SubElement()方法用于创建新的节点，它的第一个参数决定了新节点属于什么节点的子节点。