xml模块

# Author:Sure Feng

'''
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>
'''

'''
IndentationError: unexpected indent:缩进错误
'''



# 操作xml的模块，太长了，起个别名ET
import xml.etree.ElementTree as ET

tree = ET.parse("xmltest.xml")
# <xml.etree.ElementTree.ElementTree object at 0x0000000001DC3358>
print(tree)
# 获取Element 'data'对象
root = tree.getroot()
# <Element 'data' at 0x0000000001DD2D68>
# 获取根标签
print(root.tag)

# 遍历xml文档
for child in root:
# attrib以字典形式展示
    print(child.tag, child.attrib)
# 获取国家排名
    rank = int(child.find('rank').text)
# 删除排名50名后的国家
    if rank > 50:
        root.remove(child)
    else:
        for i in child:
            print(i.tag, i.attrib)

# 只遍历year节点
for node in root.iter('year'):
    # 把年份内容改为该大一年
    n_text = int(node.text)
    node.text = str(n_text + 1)
    # 添加新属性
    node.set("update", "yes")

tree.write('output.xml')



'''

import xml.etree.ElementTree as ET


new_xml = ET.Element("namelist")
name = ET.SubElement(new_xml,"name",attrib={"enrolled":"yes"})
age = ET.SubElement(name,"age",attrib={"checked":"no"})
sex = ET.SubElement(name,"sex")
sex.text = '33'
name2 = ET.SubElement(new_xml,"name",attrib={"enrolled":"no"})
age = ET.SubElement(name2,"age")
age.text = '19'

et = ET.ElementTree(new_xml) #生成文档对象
et.write("test.xml", encoding="utf-8",xml_declaration=True)

ET.dump(new_xml) #打印生成的格式
'''