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.