Python中使用dom模块生成XML文件示例

yipeiwu_com5年前Python基础

在Python中解析XML文件也有Dom和Sax两种方式,这里先介绍如何是使用Dom解析XML,这一篇文章是Dom生成XML文件,下一篇文章再继续介绍Dom解析XML文件。

在生成XML文件中,我们主要使用下面的方法来完成。

主要方法

1、生成XML节点(node)

复制代码 代码如下:

createElement("node_name")

2、给节点添加属性值(Attribute)
复制代码 代码如下:

node.setAttribute("att_name", "arr_value")

3、节点的标签值(data)
复制代码 代码如下:

createTextNode("node_value")

其中第1、3点在创建完节点(节点值)之后,还需使用下面的方法添加到指点的节点的位置下面:
复制代码 代码如下:

prev_node.appendChild(cur_node)

这里的prev_node要添加节点的上一层节点,而cur_node即为当前要添加的节点了。

代码演示

下面用代码来演示下如何使用Dom来生成XML,这个是简单版本,如下:

复制代码 代码如下:

'''
Created on 2012-8-28 
 
@author:  walfred
@module: domxml.genXML 
@description:
''' 
import xml.dom.minidom as Dom 
 
if __name__ == "__main__": 
    doc = Dom.Document() 
    root_node = doc.createElement("book_store") 
    root_node.setAttribute("name", "newhua") 
    root_node.setAttribute("website", "//www.jb51.net") 
    doc.appendChild(root_node) 
 
    book_node = doc.createElement("book1") 
 
    book_name_node = doc.createElement("name") 
    book_name_value = doc.createTextNode("hamlet") 
    book_name_node.appendChild(book_name_value) 
    book_node.appendChild(book_name_node) 
 
    book_author_node = doc.createElement("author") 
    book_author_value = doc.createTextNode("William Shakespeare") 
    book_author_node.appendChild(book_author_value) 
    book_node.appendChild(book_author_node) 
 
    root_node.appendChild(book_node) 
 
    f = open("book_store.xml", "w") 
    f.write(doc.toprettyxml(indent = "\t", newl = "\n", encoding = "utf-8")) 
    f.close()

这个代码将在当前目录下生成一个book_store.xml文件,我把它也贴在下面:

复制代码 代码如下:

<?xml version="1.0" encoding="utf-8"?> 
<book_store name="newhua" website="//www.jb51.net"> 
    <book1> 
        <name>hamlet</name> 
        <author>William Shakespeare</author> 
    </book1> 
</book_store>

当然一旦你掌握了这些基本方法之后,我们可以用一个类来更好的完成,这个类我们称之为XMLGenerator,代码如下:

复制代码 代码如下:

'''
Created on 2012-8-28 
 
@author:  walfred
@module: domxml.wXMLbyDom 
@description:
''' 
import xml.dom.minidom as Dom 
 
class XMLGenerator: 
    def __init__(self, xml_name): 
        self.doc = Dom.Document() 
        self.xml_name = xml_name 
 
    def createNode(self, node_name): 
        return self.doc.createElement(node_name) 
 
    def addNode(self, node, prev_node = None): 
        cur_node = node 
        if prev_node is not None: 
            prev_node.appendChild(cur_node) 
        else: 
            self.doc.appendChild(cur_node) 
        return cur_node 
 
    def setNodeAttr(self, node, att_name, value): 
        cur_node = node 
        cur_node.setAttribute(att_name, value) 
 
    def setNodeValue(self, cur_node, value): 
        node_data = self.doc.createTextNode(value) 
        cur_node.appendChild(node_data) 
 
    def genXml(self): 
        f = open(self.xml_name, "w") 
        f.write(self.doc.toprettyxml(indent = "\t", newl = "\n", encoding = "utf-8")) 
        f.close() 
 
if __name__ == "__main__": 
    myXMLGenerator = XMLGenerator("book_store.xml") 
 
    #xml root node 
    node_book_store = myXMLGenerator.createNode("book_store") 
    myXMLGenerator.setNodeAttr(node_book_store, "name", "new hua") 
    myXMLGenerator.setNodeAttr(node_book_store, "website", "//www.jb51.net") 
    myXMLGenerator.addNode(node = node_book_store) 
 
    #book01 
    node_book_01 = myXMLGenerator.createNode("book") 
 
    node_book_01_name = myXMLGenerator.createNode("name") 
    myXMLGenerator.setNodeValue(node_book_01_name, "Hamlet") 
    myXMLGenerator.addNode(node_book_01_name, node_book_01) 
 
    node_book_01_author = myXMLGenerator.createNode("author") 
    myXMLGenerator.setNodeValue(node_book_01_author, "William Shakespeare") 
    myXMLGenerator.addNode(node_book_01_author, node_book_01) 
 
    node_book_01_price = myXMLGenerator.createNode("price") 
    myXMLGenerator.setNodeValue(node_book_01_price, "$20") 
    myXMLGenerator.addNode(node_book_01_price, node_book_01) 
 
    node_book_01_grade = myXMLGenerator.createNode("grade") 
    myXMLGenerator.setNodeValue(node_book_01_grade, "good") 
    myXMLGenerator.addNode(node_book_01_grade, node_book_01) 
 
    myXMLGenerator.addNode(node_book_01, node_book_store) 
 
    #book 02 
    node_book_02 = myXMLGenerator.createNode("book") 
 
    node_book_02_name = myXMLGenerator.createNode("name") 
    myXMLGenerator.setNodeValue(node_book_02_name, "shuihu") 
    myXMLGenerator.addNode(node_book_02_name, node_book_02) 
 
    node_book_02_author = myXMLGenerator.createNode("author") 
    myXMLGenerator.setNodeValue(node_book_02_author, "naian shi") 
    myXMLGenerator.addNode(node_book_02_author, node_book_02) 
 
    node_book_02_price = myXMLGenerator.createNode("price") 
    myXMLGenerator.setNodeValue(node_book_02_price, "$200") 
    myXMLGenerator.addNode(node_book_02_price, node_book_02) 
 
    node_book_02_grade = myXMLGenerator.createNode("grade") 
    myXMLGenerator.setNodeValue(node_book_02_grade, "good") 
    myXMLGenerator.addNode(node_book_02_grade, node_book_02) 
 
    myXMLGenerator.addNode(node_book_02, node_book_store) 
 
    #gen 
    myXMLGenerator.genXml()

同样这个方法会在本目录下生成一个book_store.xml文件,如下:

复制代码 代码如下:

<?xml version="1.0" encoding="utf-8"?> 
<book_store name="new hua" website="//www.jb51.net"> 
    <book> 
        <name>Hamlet</name> 
        <author>William Shakespeare</author> 
        <price>$20</price> 
        <grade>good</grade> 
    </book> 
    <book> 
        <name>shuihu</name> 
        <author>naian shi</author> 
        <price>$200</price> 
        <grade>good</grade> 
    </book> 
</book_store>

这个版本算一个稍微高级的版本,但由于时间有限还很显得粗糙,读者可以发挥出更好的。

相关文章

Pandas读取MySQL数据到DataFrame的方法

方法一: #-*- coding:utf-8 -*- from sqlalchemy import create_engine class mysql_engine(): us...

Python 实现训练集、测试集随机划分

Python 实现训练集、测试集随机划分

随机从列表中取出元素: import random dataSet = [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]...

Python列表解析配合if else的方法

用习惯列表解析之后会觉得超级酷,所以在尝试使用列表解析,把循环什么的写在一行里面。使用if的时候什么时候必须要有else,什么时候可以没有else一直没搞明白,直到今天!待我缓缓道来:...

在PyCharm下使用 ipython 交互式编程的方法

目的:方便调试,查看中间结果,因为觉得设断点调试相对麻烦。 【运行环境:macOS 10.13.3,PyCharm 2017.2.4】 老手: 选中代码行,Alt+Shift+E。 或选...

机器学习python实战之决策树

机器学习python实战之决策树

决策树原理:从数据集中找出决定性的特征对数据集进行迭代划分,直到某个分支下的数据都属于同一类型,或者已经遍历了所有划分数据集的特征,停止决策树算法。   每次划分数据集的特征都有很多,那...