Python中处理XML的相关函数库及使用方法

Python中处理XML的相关函数库及使用方法

XML是一种常用的数据交换格式，因为它具有良好的可读性和可扩展性。Python中有很多处理XML的函数库，包括ElementTree、lxml、xml.sax等。本文将详细介绍这些函数库的使用方法。

1.ElementTree

ElementTree是Python标准库中一个用于解析和操作XML文档的模块，它使用简单，易于掌握。下面介绍ElementTree的常用函数：

1). Element：创建XML元素节点

使用Element()函数可以创建一个XML的元素节点，该函数需要传入一个元素的标签名，如下所示：

import xml.etree.ElementTree as ET
root = ET.Element('root')

2). SubElement：创建子节点

使用SubElement()函数可以在创建好的元素节点下创建子节点，该函数需要传入父节点对象和子节点标签名，如下所示：

import xml.etree.ElementTree as ET
root = ET.Element('root')
child = ET.SubElement(root, 'child')

3). text：设置文本内容

使用text属性可以设置XML元素节点的文本内容，如下所示：

import xml.etree.ElementTree as ET
root = ET.Element('root')
child = ET.SubElement(root, 'child')
child.text = 'This is child node.'

4). attrib：设置属性

使用attrib属性可以设置XML元素节点的属性，如下所示：

import xml.etree.ElementTree as ET
root = ET.Element('root')
child = ET.SubElement(root, 'child')
child.attrib = {'name': 'John', 'age': '20'}

5). ElementTree：创建XML树

使用ElementTree()函数可以创建一个XML的树形结构，该函数需要传入根节点对象，如下所示：

import xml.etree.ElementTree as ET
root = ET.Element('root')
child = ET.SubElement(root, 'child')
child.text = 'This is child node.'
tree = ET.ElementTree(root)

6). write：输出XML文档

使用write()函数可以将创建好的XML树输出到文件或者字符串中，该函数需要传入输出路径或者文件对象，如下所示：

import xml.etree.ElementTree as ET
root = ET.Element('root')
child = ET.SubElement(root, 'child')
child.text = 'This is child node.'
tree = ET.ElementTree(root)
tree.write('test.xml')

2.lxml

lxml是一个基于libxml2的Python库，提供了一种高效的方式来解析和处理XML文档。下面介绍lxml的常用函数：

1). Element：创建XML元素节点

lxml中与ElementTree的Element()函数类似，可以使用Element()函数创建XML元素节点，如下所示：

from lxml import etree
root = etree.Element('root')

2). SubElement：创建子节点

使用SubElement()函数可以在创建好的元素节点下创建子节点，该函数需要传入父节点对象和子节点标签名，如下所示：

from lxml import etree
root = etree.Element('root')
child = etree.SubElement(root, 'child')

3). Text：设置文本内容

使用Text()函数可以设置XML元素节点的文本内容，如下所示：

from lxml import etree
root = etree.Element('root')
child = etree.SubElement(root, 'child')
child.text = etree.CDATA('This is child node.')

4). set：设置属性

使用set()函数可以设置XML元素节点的属性，如下所示：

from lxml import etree
root = etree.Element('root')
child = etree.SubElement(root, 'child')
child.set('name', 'John')
child.set('age', '20')

5). ElementTree：创建XML树

使用ElementTree()函数可以创建一个XML的树形结构，该函数需要传入根节点对象，如下所示：

from lxml import etree
root = etree.Element('root')
child = etree.SubElement(root, 'child')
child.text = etree.CDATA('This is child node.')
tree = etree.ElementTree(root)

6). tostring：输出XML文档

使用tostring()函数可以将创建好的XML树输出到字符串中，该函数可以传入pretty_print参数来控制输出效果，如下所示：

from lxml import etree
root = etree.Element('root')
child = etree.SubElement(root, 'child')
child.text = etree.CDATA('This is child node.')
tree = etree.ElementTree(root)
xml_str = etree.tostring(root, pretty_print=True, encoding='utf-8').decode()
print(xml_str)

3.xml.sax

xml.sax是Python标准库中一个处理XML数据的低级API，它可以解析XML文档并将其转换为Python对象，通过回调函数的方式来处理XML文档中的数据。下面介绍xml.sax的常用函数：

1). ContentHandler：处理XML元素

使用ContentHandler类可以处理XML文档中的元素，它主要包括以下方法：

startDocument() # 开始解析XML文档
endDocument() # 结束解析XML文档
startElement(tag, attributes) # 开始解析XML元素
endElement(tag) # 结束解析XML元素
characters(content) # 处理XML元素间的文本内容

2). parse：解析XML文档

使用parse()函数可以解析XML文档并产生一个事件流，该函数需要传入一个XML文件或者一个文件对象和一个回调函数对象，如下所示：

import xml.sax

class MyContentHandler(xml.sax.ContentHandler):
    def startElement(self, tag, attributes):
        print('<{}>'.format(tag))

    def endElement(self, tag):
        print('</{}>'.format(tag))

    def characters(self, content):
        print(content)

handler = MyContentHandler()
parser = xml.sax.make_parser()
parser.setContentHandler(handler)
parser.parse('test.xml')

以上就是Python中常用的处理XML的函数库及使用方法，希望对大家有所帮助。