1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
|
########################################################################
#
# File Name: Sax.py
#
#
"""
Components for reading XML files from a SAX producer.
WWW: http://4suite.com/4DOM e-mail: support@4suite.com
Copyright (c) 2000 Fourthought Inc, USA. All Rights Reserved.
See http://4suite.com/COPYRIGHT for license and copyright information
"""
import sys, string, cStringIO, urllib2
from xml.sax import saxlib, saxexts, drivers
from xml.dom import Entity, DocumentType, Document
from xml.dom import DocumentType, Document
from xml.dom import implementation
from xml.dom.ext import SplitQName, ReleaseNode
from xml.dom.ext import reader
class XmlDomGenerator(saxlib.HandlerBase):
def __init__(self, keepAllWs=0):
self._keepAllWs = keepAllWs
return
def initState(self, ownerDoc=None):
"""
If None is passed in as the doc, set up an empty document to act
as owner and also add all elements to this document
"""
if ownerDoc == None:
dt = implementation.createDocumentType('', '', '')
self._ownerDoc = implementation.createDocument('', None, dt)
self._rootNode = self._ownerDoc
else:
self._ownerDoc = ownerDoc
#Create a docfrag to hold all the generated nodes.
self._rootNode = self._ownerDoc.createDocumentFragment()
#Set up the stack which keeps track of the nesting of DOM nodes.
self._nodeStack = []
self._nodeStack.append(self._rootNode)
self._currText = ''
return
def getRootNode(self):
self._completeTextNode()
return self._rootNode
def _completeTextNode(self):
if self._currText:
new_text = self._ownerDoc.createTextNode(self._currText)
self._nodeStack[-1].appendChild(new_text)
self._currText = ''
#Overridden DTDHandler methods
def notationDecl (self, name, publicId, systemId):
new_notation = self._ownerDoc.createNotation(self._ownerDoc, publicId, systemId, name)
self._ownerDoc.documentType.notations.setNamedItem(new_notation)
def unparsedEntityDecl (self, name, publicId, systemId, notationName):
new_notation = implementation.createEntity(self._ownerDoc, publicId, systemId, notationName)
self._ownerDoc.documentType.entities.setNamedItem(new_notation)
#Overridden DocumentHandler methods
def processingInstruction (self, target, data):
self._completeTextNode()
p = self._ownerDoc.createProcessingInstruction(target,data);
self._nodeStack[-1].appendChild(p)
def startElement(self, name, attribs):
self._completeTextNode()
new_element = self._ownerDoc.createElement(name)
for curr_attrib_key in attribs.keys():
new_element.setAttribute(
curr_attrib_key,
attribs[curr_attrib_key]
)
self._nodeStack.append(new_element)
def endElement(self, name):
self._completeTextNode()
new_element = self._nodeStack[-1]
del self._nodeStack[-1]
self._nodeStack[-1].appendChild(new_element)
def ignorableWhitespace(self, ch, start, length):
"""
If 'keepAllWs' permits, add ignorable white-space as a text node.
A Document node cannot contain text nodes directly.
If the white-space occurs outside the root element, there is no place
for it in the DOM and it must be discarded.
"""
if self._keepAllWs:
self._currText = self._currText + ch[start:start+length]
def characters(self, ch, start, length):
self._currText = self._currText + ch[start:start+length]
#Overridden ErrorHandler methods
#def warning(self, exception):
# raise exception
def error(self, exception):
raise exception
def fatalError(self, exception):
raise exception
class Reader(reader.Reader):
def __init__(self, validate=0, keepAllWs=0, catName=None,
saxHandlerClass=XmlDomGenerator, parser=None):
#Create an XML DOM from SAX events
self.parser = parser or (validate and saxexts.XMLValParserFactory.make_parser()) or saxexts.XMLParserFactory.make_parser()
if catName:
#set up the catalog, if there is one
from xml.parsers.xmlproc import catalog
cat_handler = catalog.SAX_catalog(catName, catalog.CatParserFactory())
self.parser.setEntityResolver(cat_handler)
self.handler = saxHandlerClass(keepAllWs)
self.parser.setDocumentHandler(self.handler)
self.parser.setDTDHandler(self.handler)
self.parser.setErrorHandler(self.handler)
return
def releaseNode(self, node):
ReleaseNode(node)
def fromStream(self, stream, ownerDocument=None):
self.handler.initState(ownerDoc=ownerDocument)
self.parser.parseFile(stream)
return self.handler.getRootNode()
########################## Deprecated ##############################
def FromXmlStream(stream, ownerDocument=None, validate=0, keepAllWs=0,
catName=None, saxHandlerClass=XmlDomGenerator, parser=None):
reader = Reader(validate, keepAllWs, catName, saxHandlerClass, parser)
return reader.fromStream(stream, ownerDocument)
def FromXml(text, ownerDocument=None, validate=0, keepAllWs=0,
catName=None, saxHandlerClass=XmlDomGenerator, parser=None):
fp = cStringIO.StringIO(text)
rv = FromXmlStream(fp, ownerDocument, validate, keepAllWs, catName,
saxHandlerClass, parser)
return rv
def FromXmlFile(fileName, ownerDocument=None, validate=0, keepAllWs=0,
catName=None, saxHandlerClass=XmlDomGenerator, parser=None):
fp = open(fileName, 'r')
try:
rv = FromXmlStream(fp, ownerDocument, validate, keepAllWs, catName,
saxHandlerClass, parser)
finally:
fp.close()
return rv
def FromXmlUrl(url, ownerDocument=None, validate=0, keepAllWs=0,
catName=None, saxHandlerClass=XmlDomGenerator, parser=None):
fp = urllib2.urlopen(url)
try:
rv = FromXmlStream(fp, ownerDocument, validate, keepAllWs, catName,
saxHandlerClass, parser)
finally:
fp.close()
return rv
|
