Python 标准库之 xml.etree.ElementTree

Python 标准库之 xml.etree.ElementTree

Python中有多种xml处理API，常用的有xml.dom.*模块、xml.sax.*模块、xml.parser.expat模块和xml.etree.ElementTree模块（以下简称ET）。本文将主要介绍ET的使用，以及它的常用函数。其它模块的简介，请参照文献[1]。

ET使用Element表示xml中的节点、文本、注释等。其主要属性如下：

tag：string对象，表示数据代表的种类，当为节点时为节点名称。

text：string对象，表示element的内容。

attrib：dictionary对象，表示附有的属性。

tail：string对象，表示element闭合之后的尾迹。

若干子元素（child elements）。

<tag attrib1=1>text</tag>tail
    1     3            2             4

PS: ET模块对于那些恶意构造的数据并不是安全的，如果需要解析数据最好了一下xml的弱点[2]

 

一、导入ET

在Python标准库中，ElementTree有两种实现方式：一种是纯Python的实现xml.etree.ElementTree，另一种是速度更快一点的xml.etree.cElementTree。如果不确定环境中是否有cElementTree，可以使用如下的方式导入：

try:
    import xml.etree.cElementTree as ET
except ImportError:
    import xml.etree.ElementTree as ET

但从Python 3.3开始，会默认使用cElementTree来加快速度，但是之前的版本最好使用如上的代码，以提高代码的兼容性。

 

二、解析xml

假设我们现在有如下的xml

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

有两种方式可以解析xml

1. 从文件中解析xml文件

>>> import xml.etree.cElementTree as ET
>>> tree = ET.ElementTree(file='doc1.xml')     #载入数据
>>> root = tree.getroot()     #获取根节点
<Element 'doc' at 0x11eb780>

2. 从内存字符串中解析xml

root = ET.fromstring(country_data_as_string)

其中第二种方式可以直接过去根节点。

 

三、数据的访问

>>> for child in root:
...   print child.tag, child.attrib
...
country {'name': 'Liechtenstein'}
country {'name': 'Singapore'}
country {'name': 'Panama'}

或者直接使用索引寻找子节点：

>>> root[0][1].text
''
 

或者使用xpath方式进行遍历，但其支持部分xpath方法，其支持的xpath请参照[4]

>>> root.findall("./country/neighbor")
[<Element 'neighbor' at 0x14fa0f0>, <Element 'neighbor' at 0x14fa150>, <Element 'neighbor' at 0x14fa3f0>, <Element 'neighbor' at 0x14fa6c0>, <Element 'neighbor' at 0x14fa750>]

 

四、 处理xml流

如下的程序将从xml文件中一边读入xml一边解析，并将在遇到标签开始或标签结束的时候返回相应的事件。

for event, elem in ET.iterparse(sys.argv[2]):
    if event == 'end':
        if elem.tag == 'location' and elem.text == 'Zimbabwe':
            count += 1
    elem.clear() # discard the element

这个程序将在检查到标签结束的时候，对指定的标签进行计数。最后的elem.clear()保证了，在解析的之后尽快的释放内存。

 

五、Element对象

class xml.etree.ElementTree.Element(tag, attrib={}, **extra)

  tag:string      元素代表的数据种类。
  text:string     元素的内容。
  tail:string      元素的尾形。
  attrib:dictionary     元素的属性字典。

  ＃针对属性的操作
  clear()          清空元素的后代、属性、text和tail也设置为None。
  get(key, default=None)     获取key对应的属性值，如该属性不存在则返回default值。
  items()         根据属性字典返回一个列表，列表元素为(key, value）。
  keys()           返回包含所有元素属性键的列表。
  set(key, value)     设置新的属性键与值。

  ＃针对后代的操作
  append(subelement)     添加直系子元素。
  extend(subelements)    增加一串元素对象作为子元素。＃python2.7新特性
  find(match)                  寻找第一个匹配子元素，匹配对象可以为tag或path。
  findall(match)               寻找所有匹配子元素，匹配对象可以为tag或path。
  findtext(match)             寻找第一个匹配子元素，返回其text值。匹配对象可以为tag或path。
  insert(index, element)   在指定位置插入子元素。
  iter(tag=None)              生成遍历当前元素所有后代或者给定tag的后代的迭代器。＃python2.7新特性
  iterfind(match)              根据tag或path查找所有的后代。
  itertext()                       遍历所有后代并返回text值。
  remove(subelement)      删除子元素。

 

六、ElementTree对象

class xml.etree.ElementTree.ElementTree(element=None, file=None)
  element如果给定，则为新的ElementTree的根节点。

  _setroot(element)：用给定的element替换当前的根节点。慎用。

  ＃ 以下方法与Element类中同名方法近似，区别在于它们指定以根节点作为操作对象。
  find(match)
  findall(match)
  findtext(match, default=None)
  iter(tag=None)
  iterfind(match)
  parse(source, parser=None)     装载xml对象，source可以为文件名或文件类型对象
  getroot()     获取根节点
  write(file, encoding="us-ascii", xml_declaration=None, default_namespace=None,method="xml") 

七、模块方法-->用于生成xml文件

1. 创建一个特别的element，通过标准序列化使其代表了一个comment。comment可以为bytestring或unicode。

     ET.Comment(text=None)

2. 生成一个element tree，通过sys.stdout输出，elem可以是元素树或单个元素。这个方法最好只用于debug。

     ET.dump(elem)

3. text是一个包含XML数据的字符串，与XML()方法类似，返回一个Element实例。

     ET.fromstring(text)

4. 从字符串的序列对象中解析xml文档。缺省parser为XMLParser，返回Element实例。V2.7中新加属性

     ET.fromstringlist(sequence, parser=None)

5. 检查是否是一个element对象。

     ET.iselement(element)

6. 将文件或包含xml数据的文件对象递增解析为element tree，并且报告进度。events是一个汇报列表，如果忽略，将只有end事件会汇报出来。
     注意，iterparse()只会在看见开始标签的">"符号时才会抛出start事件，因此届时属性是已经定义了，但是text和tail属性在那时还没有定义，同样子元素也没有定义，因此他们可能不能被显示出来。如果你想要完整的元素，请查找end事件。

     ET.iterparse(source, events=None, parser=None)

7. 将一个文件或者字符串解析为element tree。

     ET.parse(source, parser=None)

8. 这个方法会创建一个特别的element，该element被序列化为一个xml处理命令。

     ET.ProcessingInstruction(target, text=None)

9. 注册命名空间前缀。这个注册是全局有效，任何已经给出的前缀或者命名空间uri的映射关系会被删除。 V2.7新加属性

     ET.register_namespace(prefix, uri)

10. 子元素工厂，创建一个Element实例并追加到已知的节点。

     ET.SubElement(parent, tag, attrib={}, **extra)

11. 生成一个字符串来表示表示xml的element，包括所有子元素。element是Element实例，method为"xml", "html", "text"。 返回包含了xml数据的字符串。

     ET.tostring(element, encoding="us-ascii", method="xml")

12. 生成一个字符串来表示表示xml的element，包括所有子元素。element是Element实例，method为"xml","html","text"。返回包含了xml数据的字符串列表。V2.7新添加属性

     ET.tostringlist(element, encoding="us-ascii", method="xml")

13. 从一个字符串常量中解析出xml片段。返回Element实例。

     ET.XML(text, parser=None)

14. 从字符串常量解析出xml片段，同时返回一个字典，用以映射element的id到其自身。

     ET.XMLID(text, parser=None)