The dom module is used in Python to generate a sample XML file

  • 2020-05-05 11:25:51
  • OfStack

There are also two ways to parse XML files in Python, Dom and Sax. Here is how to parse XML using Dom.

In generating the XML file, we mainly use the following methods.

main method

1. Generate XML node (node)


createElement("node_name")

2. Add attribute value (Attribute)
to the node

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

3. Label value of the node (data)

createTextNode("node_value")

Points 1 and 3, after creating the node (node value), need to be added to the location of the pointing node using the following method:

prev_node.appendChild(cur_node)

Here prev_node is adding the next layer of nodes, and cur_node is the current node to be added.

The code demonstrates

The following code demonstrates how to use Dom to generate XML. This is the simple version, as follows:


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

This code will generate an book_store.xml file in the current directory


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

Of course, once you've mastered these basic methods, you can do it better with a class we call XMLGenerator, which looks like this:


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

Again, this method generates an book_store.xml file in this directory, as follows:


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

This is a slightly more advanced version, but it's still rough and ready to go, so the reader can do better.


Related articles: