blob: 02e3b85d4328628d1bad958dc8e013f863cbfad6 [file] [log] [blame]
Fred Drake1f549022000-09-24 05:21:58 +00001"""\
Fred Drakef7cf40d2000-12-14 18:16:11 +00002minidom.py -- a lightweight DOM implementation.
Fred Drake55c38192000-06-29 19:39:57 +00003
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +00004parse("foo.xml")
Paul Prescod623511b2000-07-21 22:05:49 +00005
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +00006parseString("<foo><bar/></foo>")
Paul Prescod623511b2000-07-21 22:05:49 +00007
Fred Drake55c38192000-06-29 19:39:57 +00008Todo:
9=====
10 * convenience methods for getting elements and text.
11 * more testing
12 * bring some of the writer and linearizer code into conformance with this
13 interface
14 * SAX 2 namespaces
15"""
16
Fred Drakefbdeaad2006-07-29 16:56:15 +000017import xml.dom
Fred Drake55c38192000-06-29 19:39:57 +000018
Fred Drakefbdeaad2006-07-29 16:56:15 +000019from xml.dom import EMPTY_NAMESPACE, EMPTY_PREFIX, XMLNS_NAMESPACE, domreg
20from xml.dom.minicompat import *
21from xml.dom.xmlbuilder import DOMImplementationLS, DocumentLS
Fred Drake3ac6a092001-09-28 04:33:06 +000022
Martin v. Löwis787354c2003-01-25 15:28:29 +000023# This is used by the ID-cache invalidation checks; the list isn't
24# actually complete, since the nodes being checked will never be the
25# DOCUMENT_NODE or DOCUMENT_FRAGMENT_NODE. (The node being checked is
26# the node being added or removed, not the node being modified.)
27#
Fred Drakefbdeaad2006-07-29 16:56:15 +000028_nodeTypes_with_children = (xml.dom.Node.ELEMENT_NODE,
29 xml.dom.Node.ENTITY_REFERENCE_NODE)
Martin v. Löwis95700f72002-03-15 13:51:59 +000030
Fred Drake3ac6a092001-09-28 04:33:06 +000031
Fred Drakefbdeaad2006-07-29 16:56:15 +000032class Node(xml.dom.Node):
Martin v. Löwis126f2f62001-03-13 10:50:13 +000033 namespaceURI = None # this is non-null only for elements and attributes
Fred Drake575712e2001-09-28 20:25:45 +000034 parentNode = None
35 ownerDocument = None
Martin v. Löwis787354c2003-01-25 15:28:29 +000036 nextSibling = None
37 previousSibling = None
Martin v. Löwis52ce0d02001-01-27 08:47:37 +000038
Martin v. Löwis787354c2003-01-25 15:28:29 +000039 prefix = EMPTY_PREFIX # non-null only for NS elements and attributes
Fred Drake55c38192000-06-29 19:39:57 +000040
Fred Drake1f549022000-09-24 05:21:58 +000041 def __nonzero__(self):
Martin v. Löwis787354c2003-01-25 15:28:29 +000042 return True
Fred Drake55c38192000-06-29 19:39:57 +000043
Martin v. Löwis7d650ca2002-06-30 15:05:00 +000044 def toxml(self, encoding = None):
45 return self.toprettyxml("", "", encoding)
Fred Drake55c38192000-06-29 19:39:57 +000046
Martin v. Löwis7d650ca2002-06-30 15:05:00 +000047 def toprettyxml(self, indent="\t", newl="\n", encoding = None):
Martin v. Löwiscb67ea12001-03-31 16:30:40 +000048 # indent = the indentation string to prepend, per level
49 # newl = the newline string to append
50 writer = _get_StringIO()
Martin v. Löwis7d650ca2002-06-30 15:05:00 +000051 if encoding is not None:
52 import codecs
53 # Can't use codecs.getwriter to preserve 2.0 compatibility
54 writer = codecs.lookup(encoding)[3](writer)
55 if self.nodeType == Node.DOCUMENT_NODE:
56 # Can pass encoding only to document, to put it into XML header
57 self.writexml(writer, "", indent, newl, encoding)
58 else:
59 self.writexml(writer, "", indent, newl)
Martin v. Löwiscb67ea12001-03-31 16:30:40 +000060 return writer.getvalue()
Martin v. Löwis46fa39a2001-02-06 00:14:08 +000061
Fred Drake1f549022000-09-24 05:21:58 +000062 def hasChildNodes(self):
63 if self.childNodes:
Martin v. Löwis787354c2003-01-25 15:28:29 +000064 return True
Fred Drake1f549022000-09-24 05:21:58 +000065 else:
Martin v. Löwis787354c2003-01-25 15:28:29 +000066 return False
67
68 def _get_childNodes(self):
69 return self.childNodes
Fred Drake55c38192000-06-29 19:39:57 +000070
Fred Drake1f549022000-09-24 05:21:58 +000071 def _get_firstChild(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +000072 if self.childNodes:
73 return self.childNodes[0]
Paul Prescod73678da2000-07-01 04:58:47 +000074
Fred Drake1f549022000-09-24 05:21:58 +000075 def _get_lastChild(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +000076 if self.childNodes:
77 return self.childNodes[-1]
Paul Prescod73678da2000-07-01 04:58:47 +000078
Fred Drake1f549022000-09-24 05:21:58 +000079 def insertBefore(self, newChild, refChild):
Martin v. Löwis126f2f62001-03-13 10:50:13 +000080 if newChild.nodeType == self.DOCUMENT_FRAGMENT_NODE:
Fred Drakee50959a2001-12-06 04:32:18 +000081 for c in tuple(newChild.childNodes):
Martin v. Löwis126f2f62001-03-13 10:50:13 +000082 self.insertBefore(c, refChild)
83 ### The DOM does not clearly specify what to return in this case
84 return newChild
Martin v. Löwis787354c2003-01-25 15:28:29 +000085 if newChild.nodeType not in self._child_node_types:
Fred Drakefbdeaad2006-07-29 16:56:15 +000086 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +000087 "%s cannot be child of %s" % (repr(newChild), repr(self)))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +000088 if newChild.parentNode is not None:
89 newChild.parentNode.removeChild(newChild)
Fred Drake4ccf4a12000-11-21 22:02:22 +000090 if refChild is None:
91 self.appendChild(newChild)
92 else:
Martin v. Löwis787354c2003-01-25 15:28:29 +000093 try:
94 index = self.childNodes.index(refChild)
95 except ValueError:
Fred Drakefbdeaad2006-07-29 16:56:15 +000096 raise xml.dom.NotFoundErr()
Martin v. Löwis787354c2003-01-25 15:28:29 +000097 if newChild.nodeType in _nodeTypes_with_children:
98 _clear_id_cache(self)
Fred Drake4ccf4a12000-11-21 22:02:22 +000099 self.childNodes.insert(index, newChild)
100 newChild.nextSibling = refChild
101 refChild.previousSibling = newChild
102 if index:
103 node = self.childNodes[index-1]
104 node.nextSibling = newChild
105 newChild.previousSibling = node
106 else:
107 newChild.previousSibling = None
Martin v. Löwis787354c2003-01-25 15:28:29 +0000108 newChild.parentNode = self
Fred Drake4ccf4a12000-11-21 22:02:22 +0000109 return newChild
Fred Drake55c38192000-06-29 19:39:57 +0000110
Fred Drake1f549022000-09-24 05:21:58 +0000111 def appendChild(self, node):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000112 if node.nodeType == self.DOCUMENT_FRAGMENT_NODE:
Fred Drakee50959a2001-12-06 04:32:18 +0000113 for c in tuple(node.childNodes):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000114 self.appendChild(c)
115 ### The DOM does not clearly specify what to return in this case
116 return node
Martin v. Löwis787354c2003-01-25 15:28:29 +0000117 if node.nodeType not in self._child_node_types:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000118 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +0000119 "%s cannot be child of %s" % (repr(node), repr(self)))
120 elif node.nodeType in _nodeTypes_with_children:
121 _clear_id_cache(self)
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000122 if node.parentNode is not None:
123 node.parentNode.removeChild(node)
Martin v. Löwis787354c2003-01-25 15:28:29 +0000124 _append_child(self, node)
Fred Drake13a30692000-10-09 20:04:16 +0000125 node.nextSibling = None
Paul Prescod73678da2000-07-01 04:58:47 +0000126 return node
127
Fred Drake1f549022000-09-24 05:21:58 +0000128 def replaceChild(self, newChild, oldChild):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000129 if newChild.nodeType == self.DOCUMENT_FRAGMENT_NODE:
130 refChild = oldChild.nextSibling
131 self.removeChild(oldChild)
132 return self.insertBefore(newChild, refChild)
Martin v. Löwis787354c2003-01-25 15:28:29 +0000133 if newChild.nodeType not in self._child_node_types:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000134 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +0000135 "%s cannot be child of %s" % (repr(newChild), repr(self)))
Fred Drake4ccf4a12000-11-21 22:02:22 +0000136 if newChild is oldChild:
137 return
Andrew M. Kuchling841d25e2005-11-22 19:03:16 +0000138 if newChild.parentNode is not None:
139 newChild.parentNode.removeChild(newChild)
Martin v. Löwis787354c2003-01-25 15:28:29 +0000140 try:
141 index = self.childNodes.index(oldChild)
142 except ValueError:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000143 raise xml.dom.NotFoundErr()
Fred Drake4ccf4a12000-11-21 22:02:22 +0000144 self.childNodes[index] = newChild
Martin v. Löwis787354c2003-01-25 15:28:29 +0000145 newChild.parentNode = self
146 oldChild.parentNode = None
147 if (newChild.nodeType in _nodeTypes_with_children
148 or oldChild.nodeType in _nodeTypes_with_children):
149 _clear_id_cache(self)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000150 newChild.nextSibling = oldChild.nextSibling
151 newChild.previousSibling = oldChild.previousSibling
Martin v. Löwis156c3372000-12-28 18:40:56 +0000152 oldChild.nextSibling = None
Fred Drake4ccf4a12000-11-21 22:02:22 +0000153 oldChild.previousSibling = None
Martin v. Löwis156c3372000-12-28 18:40:56 +0000154 if newChild.previousSibling:
155 newChild.previousSibling.nextSibling = newChild
156 if newChild.nextSibling:
157 newChild.nextSibling.previousSibling = newChild
Fred Drake4ccf4a12000-11-21 22:02:22 +0000158 return oldChild
Paul Prescod73678da2000-07-01 04:58:47 +0000159
Fred Drake1f549022000-09-24 05:21:58 +0000160 def removeChild(self, oldChild):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000161 try:
162 self.childNodes.remove(oldChild)
163 except ValueError:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000164 raise xml.dom.NotFoundErr()
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000165 if oldChild.nextSibling is not None:
166 oldChild.nextSibling.previousSibling = oldChild.previousSibling
167 if oldChild.previousSibling is not None:
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000168 oldChild.previousSibling.nextSibling = oldChild.nextSibling
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000169 oldChild.nextSibling = oldChild.previousSibling = None
Martin v. Löwis787354c2003-01-25 15:28:29 +0000170 if oldChild.nodeType in _nodeTypes_with_children:
171 _clear_id_cache(self)
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000172
Martin v. Löwis787354c2003-01-25 15:28:29 +0000173 oldChild.parentNode = None
Fred Drake4ccf4a12000-11-21 22:02:22 +0000174 return oldChild
175
176 def normalize(self):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000177 L = []
178 for child in self.childNodes:
179 if child.nodeType == Node.TEXT_NODE:
R. David Murray0374a822009-04-09 21:54:50 +0000180 if not child.data:
181 # empty text node; discard
182 if L:
183 L[-1].nextSibling = child.nextSibling
184 if child.nextSibling:
185 child.nextSibling.previousSibling = child.previousSibling
186 child.unlink()
187 elif L and L[-1].nodeType == child.nodeType:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000188 # collapse text node
189 node = L[-1]
Martin v. Löwis787354c2003-01-25 15:28:29 +0000190 node.data = node.data + child.data
Fred Drake4ccf4a12000-11-21 22:02:22 +0000191 node.nextSibling = child.nextSibling
R. David Murray0374a822009-04-09 21:54:50 +0000192 if child.nextSibling:
193 child.nextSibling.previousSibling = node
Fred Drake4ccf4a12000-11-21 22:02:22 +0000194 child.unlink()
R. David Murray0374a822009-04-09 21:54:50 +0000195 else:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000196 L.append(child)
Fred Drakef7cf40d2000-12-14 18:16:11 +0000197 else:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000198 L.append(child)
199 if child.nodeType == Node.ELEMENT_NODE:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000200 child.normalize()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000201 self.childNodes[:] = L
Paul Prescod73678da2000-07-01 04:58:47 +0000202
Fred Drake1f549022000-09-24 05:21:58 +0000203 def cloneNode(self, deep):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000204 return _clone_node(self, deep, self.ownerDocument or self)
Fred Drake55c38192000-06-29 19:39:57 +0000205
Martin v. Löwis787354c2003-01-25 15:28:29 +0000206 def isSupported(self, feature, version):
207 return self.ownerDocument.implementation.hasFeature(feature, version)
208
209 def _get_localName(self):
210 # Overridden in Element and Attr where localName can be Non-Null
211 return None
212
213 # Node interfaces from Level 3 (WD 9 April 2002)
Fred Drake25239772001-02-02 19:40:19 +0000214
215 def isSameNode(self, other):
216 return self is other
217
Martin v. Löwis787354c2003-01-25 15:28:29 +0000218 def getInterface(self, feature):
219 if self.isSupported(feature, None):
220 return self
221 else:
222 return None
223
224 # The "user data" functions use a dictionary that is only present
225 # if some user data has been set, so be careful not to assume it
226 # exists.
227
228 def getUserData(self, key):
229 try:
230 return self._user_data[key][0]
231 except (AttributeError, KeyError):
232 return None
233
234 def setUserData(self, key, data, handler):
235 old = None
236 try:
237 d = self._user_data
238 except AttributeError:
239 d = {}
240 self._user_data = d
Brett Cannon5dd504d2008-08-04 00:23:58 +0000241 if key in d:
Martin v. Löwis787354c2003-01-25 15:28:29 +0000242 old = d[key][0]
243 if data is None:
244 # ignore handlers passed for None
245 handler = None
246 if old is not None:
247 del d[key]
248 else:
249 d[key] = (data, handler)
250 return old
251
252 def _call_user_data_handler(self, operation, src, dst):
253 if hasattr(self, "_user_data"):
254 for key, (data, handler) in self._user_data.items():
255 if handler is not None:
256 handler.handle(operation, key, data, src, dst)
257
Fred Drake25239772001-02-02 19:40:19 +0000258 # minidom-specific API:
259
Fred Drake1f549022000-09-24 05:21:58 +0000260 def unlink(self):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000261 self.parentNode = self.ownerDocument = None
Martin v. Löwis787354c2003-01-25 15:28:29 +0000262 if self.childNodes:
263 for child in self.childNodes:
264 child.unlink()
265 self.childNodes = NodeList()
Paul Prescod4221ff02000-10-13 20:11:42 +0000266 self.previousSibling = None
267 self.nextSibling = None
Martin v. Löwis787354c2003-01-25 15:28:29 +0000268
269defproperty(Node, "firstChild", doc="First child node, or None.")
270defproperty(Node, "lastChild", doc="Last child node, or None.")
271defproperty(Node, "localName", doc="Namespace-local name of this node.")
272
273
274def _append_child(self, node):
275 # fast path with less checks; usable by DOM builders if careful
276 childNodes = self.childNodes
277 if childNodes:
278 last = childNodes[-1]
279 node.__dict__["previousSibling"] = last
280 last.__dict__["nextSibling"] = node
281 childNodes.append(node)
282 node.__dict__["parentNode"] = self
283
284def _in_document(node):
285 # return True iff node is part of a document tree
286 while node is not None:
287 if node.nodeType == Node.DOCUMENT_NODE:
288 return True
289 node = node.parentNode
290 return False
Fred Drake55c38192000-06-29 19:39:57 +0000291
Fred Drake1f549022000-09-24 05:21:58 +0000292def _write_data(writer, data):
Fred Drake55c38192000-06-29 19:39:57 +0000293 "Writes datachars to writer."
Martin v. Löwis787354c2003-01-25 15:28:29 +0000294 data = data.replace("&", "&amp;").replace("<", "&lt;")
295 data = data.replace("\"", "&quot;").replace(">", "&gt;")
Fred Drake55c38192000-06-29 19:39:57 +0000296 writer.write(data)
297
Martin v. Löwis787354c2003-01-25 15:28:29 +0000298def _get_elements_by_tagName_helper(parent, name, rc):
Fred Drake55c38192000-06-29 19:39:57 +0000299 for node in parent.childNodes:
Fred Drake1f549022000-09-24 05:21:58 +0000300 if node.nodeType == Node.ELEMENT_NODE and \
301 (name == "*" or node.tagName == name):
302 rc.append(node)
Martin v. Löwis787354c2003-01-25 15:28:29 +0000303 _get_elements_by_tagName_helper(node, name, rc)
Fred Drake55c38192000-06-29 19:39:57 +0000304 return rc
305
Martin v. Löwis787354c2003-01-25 15:28:29 +0000306def _get_elements_by_tagName_ns_helper(parent, nsURI, localName, rc):
Fred Drake55c38192000-06-29 19:39:57 +0000307 for node in parent.childNodes:
Fred Drake1f549022000-09-24 05:21:58 +0000308 if node.nodeType == Node.ELEMENT_NODE:
Martin v. Löwised525fb2001-06-03 14:06:42 +0000309 if ((localName == "*" or node.localName == localName) and
Fred Drake1f549022000-09-24 05:21:58 +0000310 (nsURI == "*" or node.namespaceURI == nsURI)):
311 rc.append(node)
Martin v. Löwis787354c2003-01-25 15:28:29 +0000312 _get_elements_by_tagName_ns_helper(node, nsURI, localName, rc)
Fred Drakef7cf40d2000-12-14 18:16:11 +0000313 return rc
Fred Drake55c38192000-06-29 19:39:57 +0000314
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000315class DocumentFragment(Node):
316 nodeType = Node.DOCUMENT_FRAGMENT_NODE
317 nodeName = "#document-fragment"
318 nodeValue = None
319 attributes = None
320 parentNode = None
Martin v. Löwis787354c2003-01-25 15:28:29 +0000321 _child_node_types = (Node.ELEMENT_NODE,
322 Node.TEXT_NODE,
323 Node.CDATA_SECTION_NODE,
324 Node.ENTITY_REFERENCE_NODE,
325 Node.PROCESSING_INSTRUCTION_NODE,
326 Node.COMMENT_NODE,
327 Node.NOTATION_NODE)
328
329 def __init__(self):
330 self.childNodes = NodeList()
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000331
332
Fred Drake55c38192000-06-29 19:39:57 +0000333class Attr(Node):
Fred Drake1f549022000-09-24 05:21:58 +0000334 nodeType = Node.ATTRIBUTE_NODE
Fred Drake4ccf4a12000-11-21 22:02:22 +0000335 attributes = None
336 ownerElement = None
Martin v. Löwis787354c2003-01-25 15:28:29 +0000337 specified = False
338 _is_id = False
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000339
Martin v. Löwis787354c2003-01-25 15:28:29 +0000340 _child_node_types = (Node.TEXT_NODE, Node.ENTITY_REFERENCE_NODE)
341
342 def __init__(self, qName, namespaceURI=EMPTY_NAMESPACE, localName=None,
343 prefix=None):
Fred Drake55c38192000-06-29 19:39:57 +0000344 # skip setattr for performance
Fred Drake4ccf4a12000-11-21 22:02:22 +0000345 d = self.__dict__
Fred Drake4ccf4a12000-11-21 22:02:22 +0000346 d["nodeName"] = d["name"] = qName
347 d["namespaceURI"] = namespaceURI
348 d["prefix"] = prefix
Martin v. Löwis787354c2003-01-25 15:28:29 +0000349 d['childNodes'] = NodeList()
350
351 # Add the single child node that represents the value of the attr
352 self.childNodes.append(Text())
353
Paul Prescod73678da2000-07-01 04:58:47 +0000354 # nodeValue and value are set elsewhere
Fred Drake55c38192000-06-29 19:39:57 +0000355
Martin v. Löwis787354c2003-01-25 15:28:29 +0000356 def _get_localName(self):
357 return self.nodeName.split(":", 1)[-1]
358
359 def _get_name(self):
360 return self.name
361
362 def _get_specified(self):
363 return self.specified
364
Fred Drake1f549022000-09-24 05:21:58 +0000365 def __setattr__(self, name, value):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000366 d = self.__dict__
Fred Drake1f549022000-09-24 05:21:58 +0000367 if name in ("value", "nodeValue"):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000368 d["value"] = d["nodeValue"] = value
Martin v. Löwis787354c2003-01-25 15:28:29 +0000369 d2 = self.childNodes[0].__dict__
370 d2["data"] = d2["nodeValue"] = value
371 if self.ownerElement is not None:
372 _clear_id_cache(self.ownerElement)
Fred Drakef7cf40d2000-12-14 18:16:11 +0000373 elif name in ("name", "nodeName"):
374 d["name"] = d["nodeName"] = value
Martin v. Löwis787354c2003-01-25 15:28:29 +0000375 if self.ownerElement is not None:
376 _clear_id_cache(self.ownerElement)
Fred Drake55c38192000-06-29 19:39:57 +0000377 else:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000378 d[name] = value
Fred Drake55c38192000-06-29 19:39:57 +0000379
Martin v. Löwis995359c2003-01-26 08:59:32 +0000380 def _set_prefix(self, prefix):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000381 nsuri = self.namespaceURI
Martin v. Löwis995359c2003-01-26 08:59:32 +0000382 if prefix == "xmlns":
383 if nsuri and nsuri != XMLNS_NAMESPACE:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000384 raise xml.dom.NamespaceErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +0000385 "illegal use of 'xmlns' prefix for the wrong namespace")
386 d = self.__dict__
387 d['prefix'] = prefix
388 if prefix is None:
389 newName = self.localName
390 else:
Martin v. Löwis995359c2003-01-26 08:59:32 +0000391 newName = "%s:%s" % (prefix, self.localName)
Martin v. Löwis787354c2003-01-25 15:28:29 +0000392 if self.ownerElement:
393 _clear_id_cache(self.ownerElement)
394 d['nodeName'] = d['name'] = newName
395
396 def _set_value(self, value):
397 d = self.__dict__
398 d['value'] = d['nodeValue'] = value
399 if self.ownerElement:
400 _clear_id_cache(self.ownerElement)
401 self.childNodes[0].data = value
402
403 def unlink(self):
404 # This implementation does not call the base implementation
405 # since most of that is not needed, and the expense of the
406 # method call is not warranted. We duplicate the removal of
407 # children, but that's all we needed from the base class.
408 elem = self.ownerElement
409 if elem is not None:
410 del elem._attrs[self.nodeName]
411 del elem._attrsNS[(self.namespaceURI, self.localName)]
412 if self._is_id:
413 self._is_id = False
414 elem._magic_id_nodes -= 1
415 self.ownerDocument._magic_id_count -= 1
416 for child in self.childNodes:
417 child.unlink()
418 del self.childNodes[:]
419
420 def _get_isId(self):
421 if self._is_id:
422 return True
423 doc = self.ownerDocument
424 elem = self.ownerElement
425 if doc is None or elem is None:
426 return False
427
428 info = doc._get_elem_info(elem)
429 if info is None:
430 return False
431 if self.namespaceURI:
432 return info.isIdNS(self.namespaceURI, self.localName)
433 else:
434 return info.isId(self.nodeName)
435
436 def _get_schemaType(self):
437 doc = self.ownerDocument
438 elem = self.ownerElement
439 if doc is None or elem is None:
440 return _no_type
441
442 info = doc._get_elem_info(elem)
443 if info is None:
444 return _no_type
445 if self.namespaceURI:
446 return info.getAttributeTypeNS(self.namespaceURI, self.localName)
447 else:
448 return info.getAttributeType(self.nodeName)
449
450defproperty(Attr, "isId", doc="True if this attribute is an ID.")
451defproperty(Attr, "localName", doc="Namespace-local name of this attribute.")
452defproperty(Attr, "schemaType", doc="Schema type for this attribute.")
Fred Drake4ccf4a12000-11-21 22:02:22 +0000453
Fred Drakef7cf40d2000-12-14 18:16:11 +0000454
Fred Drake9ea179f2006-04-06 01:29:04 +0000455class NamedNodeMap(object):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000456 """The attribute list is a transient interface to the underlying
457 dictionaries. Mutations here will change the underlying element's
Fred Drakef7cf40d2000-12-14 18:16:11 +0000458 dictionary.
459
460 Ordering is imposed artificially and does not reflect the order of
461 attributes as found in an input document.
462 """
Fred Drake4ccf4a12000-11-21 22:02:22 +0000463
Martin v. Löwis787354c2003-01-25 15:28:29 +0000464 __slots__ = ('_attrs', '_attrsNS', '_ownerElement')
465
Fred Drake2998a552001-12-06 18:27:48 +0000466 def __init__(self, attrs, attrsNS, ownerElement):
Fred Drake1f549022000-09-24 05:21:58 +0000467 self._attrs = attrs
468 self._attrsNS = attrsNS
Fred Drake2998a552001-12-06 18:27:48 +0000469 self._ownerElement = ownerElement
Fred Drakef7cf40d2000-12-14 18:16:11 +0000470
Martin v. Löwis787354c2003-01-25 15:28:29 +0000471 def _get_length(self):
472 return len(self._attrs)
Fred Drake55c38192000-06-29 19:39:57 +0000473
Fred Drake1f549022000-09-24 05:21:58 +0000474 def item(self, index):
Fred Drake55c38192000-06-29 19:39:57 +0000475 try:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000476 return self[self._attrs.keys()[index]]
Fred Drake55c38192000-06-29 19:39:57 +0000477 except IndexError:
478 return None
Fred Drake55c38192000-06-29 19:39:57 +0000479
Fred Drake1f549022000-09-24 05:21:58 +0000480 def items(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000481 L = []
482 for node in self._attrs.values():
Martin v. Löwisd5fb58f2001-01-27 08:38:34 +0000483 L.append((node.nodeName, node.value))
Fred Drake4ccf4a12000-11-21 22:02:22 +0000484 return L
Fred Drake1f549022000-09-24 05:21:58 +0000485
486 def itemsNS(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000487 L = []
488 for node in self._attrs.values():
Fred Drake49a5d032001-11-30 22:21:58 +0000489 L.append(((node.namespaceURI, node.localName), node.value))
Fred Drake4ccf4a12000-11-21 22:02:22 +0000490 return L
Fred Drake16f63292000-10-23 18:09:50 +0000491
Martin v. Löwis787354c2003-01-25 15:28:29 +0000492 def has_key(self, key):
493 if isinstance(key, StringTypes):
Benjamin Peterson6e3dbbd2009-10-09 22:15:50 +0000494 return key in self._attrs
Martin v. Löwis787354c2003-01-25 15:28:29 +0000495 else:
Benjamin Peterson6e3dbbd2009-10-09 22:15:50 +0000496 return key in self._attrsNS
Martin v. Löwis787354c2003-01-25 15:28:29 +0000497
Fred Drake1f549022000-09-24 05:21:58 +0000498 def keys(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000499 return self._attrs.keys()
Fred Drake55c38192000-06-29 19:39:57 +0000500
Fred Drake1f549022000-09-24 05:21:58 +0000501 def keysNS(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000502 return self._attrsNS.keys()
Fred Drake55c38192000-06-29 19:39:57 +0000503
Fred Drake1f549022000-09-24 05:21:58 +0000504 def values(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000505 return self._attrs.values()
Fred Drake55c38192000-06-29 19:39:57 +0000506
Martin v. Löwis787354c2003-01-25 15:28:29 +0000507 def get(self, name, value=None):
Martin v. Löwisd5fb58f2001-01-27 08:38:34 +0000508 return self._attrs.get(name, value)
509
Martin v. Löwis787354c2003-01-25 15:28:29 +0000510 __len__ = _get_length
Fred Drake55c38192000-06-29 19:39:57 +0000511
Nick Coghlan48361f52008-08-11 15:45:58 +0000512 __hash__ = None # Mutable type can't be correctly hashed
Fred Drake1f549022000-09-24 05:21:58 +0000513 def __cmp__(self, other):
514 if self._attrs is getattr(other, "_attrs", None):
Fred Drake55c38192000-06-29 19:39:57 +0000515 return 0
Fred Drake16f63292000-10-23 18:09:50 +0000516 else:
Fred Drake1f549022000-09-24 05:21:58 +0000517 return cmp(id(self), id(other))
Fred Drake55c38192000-06-29 19:39:57 +0000518
Fred Drake1f549022000-09-24 05:21:58 +0000519 def __getitem__(self, attname_or_tuple):
Fred Drake70d044b2006-04-06 01:32:26 +0000520 if isinstance(attname_or_tuple, tuple):
Paul Prescod73678da2000-07-01 04:58:47 +0000521 return self._attrsNS[attname_or_tuple]
Fred Drake55c38192000-06-29 19:39:57 +0000522 else:
Paul Prescod73678da2000-07-01 04:58:47 +0000523 return self._attrs[attname_or_tuple]
Fred Drake55c38192000-06-29 19:39:57 +0000524
Paul Prescod1e688272000-07-01 19:21:47 +0000525 # same as set
Fred Drake1f549022000-09-24 05:21:58 +0000526 def __setitem__(self, attname, value):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000527 if isinstance(value, StringTypes):
528 try:
529 node = self._attrs[attname]
530 except KeyError:
531 node = Attr(attname)
532 node.ownerDocument = self._ownerElement.ownerDocument
Martin v. Löwis995359c2003-01-26 08:59:32 +0000533 self.setNamedItem(node)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000534 node.value = value
Paul Prescod1e688272000-07-01 19:21:47 +0000535 else:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000536 if not isinstance(value, Attr):
537 raise TypeError, "value must be a string or Attr object"
Fred Drake1f549022000-09-24 05:21:58 +0000538 node = value
Martin v. Löwis787354c2003-01-25 15:28:29 +0000539 self.setNamedItem(node)
540
541 def getNamedItem(self, name):
542 try:
543 return self._attrs[name]
544 except KeyError:
545 return None
546
547 def getNamedItemNS(self, namespaceURI, localName):
548 try:
549 return self._attrsNS[(namespaceURI, localName)]
550 except KeyError:
551 return None
552
553 def removeNamedItem(self, name):
554 n = self.getNamedItem(name)
555 if n is not None:
556 _clear_id_cache(self._ownerElement)
557 del self._attrs[n.nodeName]
558 del self._attrsNS[(n.namespaceURI, n.localName)]
Brett Cannon5dd504d2008-08-04 00:23:58 +0000559 if 'ownerElement' in n.__dict__:
Martin v. Löwis787354c2003-01-25 15:28:29 +0000560 n.__dict__['ownerElement'] = None
561 return n
562 else:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000563 raise xml.dom.NotFoundErr()
Martin v. Löwis787354c2003-01-25 15:28:29 +0000564
565 def removeNamedItemNS(self, namespaceURI, localName):
566 n = self.getNamedItemNS(namespaceURI, localName)
567 if n is not None:
568 _clear_id_cache(self._ownerElement)
569 del self._attrsNS[(n.namespaceURI, n.localName)]
570 del self._attrs[n.nodeName]
Brett Cannon5dd504d2008-08-04 00:23:58 +0000571 if 'ownerElement' in n.__dict__:
Martin v. Löwis787354c2003-01-25 15:28:29 +0000572 n.__dict__['ownerElement'] = None
573 return n
574 else:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000575 raise xml.dom.NotFoundErr()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000576
577 def setNamedItem(self, node):
Andrew M. Kuchlingbc8f72c2001-02-21 01:30:26 +0000578 if not isinstance(node, Attr):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000579 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +0000580 "%s cannot be child of %s" % (repr(node), repr(self)))
Fred Drakef7cf40d2000-12-14 18:16:11 +0000581 old = self._attrs.get(node.name)
Paul Prescod1e688272000-07-01 19:21:47 +0000582 if old:
583 old.unlink()
Fred Drake1f549022000-09-24 05:21:58 +0000584 self._attrs[node.name] = node
585 self._attrsNS[(node.namespaceURI, node.localName)] = node
Fred Drake2998a552001-12-06 18:27:48 +0000586 node.ownerElement = self._ownerElement
Martin v. Löwis787354c2003-01-25 15:28:29 +0000587 _clear_id_cache(node.ownerElement)
Fred Drakef7cf40d2000-12-14 18:16:11 +0000588 return old
589
590 def setNamedItemNS(self, node):
591 return self.setNamedItem(node)
Paul Prescod73678da2000-07-01 04:58:47 +0000592
Fred Drake1f549022000-09-24 05:21:58 +0000593 def __delitem__(self, attname_or_tuple):
594 node = self[attname_or_tuple]
Martin v. Löwis787354c2003-01-25 15:28:29 +0000595 _clear_id_cache(node.ownerElement)
Paul Prescod73678da2000-07-01 04:58:47 +0000596 node.unlink()
Martin v. Löwis787354c2003-01-25 15:28:29 +0000597
598 def __getstate__(self):
599 return self._attrs, self._attrsNS, self._ownerElement
600
601 def __setstate__(self, state):
602 self._attrs, self._attrsNS, self._ownerElement = state
603
604defproperty(NamedNodeMap, "length",
605 doc="Number of nodes in the NamedNodeMap.")
Fred Drakef7cf40d2000-12-14 18:16:11 +0000606
607AttributeList = NamedNodeMap
608
Fred Drake1f549022000-09-24 05:21:58 +0000609
Fred Drake9ea179f2006-04-06 01:29:04 +0000610class TypeInfo(object):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000611 __slots__ = 'namespace', 'name'
612
613 def __init__(self, namespace, name):
614 self.namespace = namespace
615 self.name = name
616
617 def __repr__(self):
618 if self.namespace:
Walter Dörwald70a6b492004-02-12 17:35:32 +0000619 return "<TypeInfo %r (from %r)>" % (self.name, self.namespace)
Martin v. Löwis787354c2003-01-25 15:28:29 +0000620 else:
Walter Dörwald70a6b492004-02-12 17:35:32 +0000621 return "<TypeInfo %r>" % self.name
Martin v. Löwis787354c2003-01-25 15:28:29 +0000622
623 def _get_name(self):
624 return self.name
625
626 def _get_namespace(self):
627 return self.namespace
628
629_no_type = TypeInfo(None, None)
630
Martin v. Löwisa2fda0d2000-10-07 12:10:28 +0000631class Element(Node):
Fred Drake1f549022000-09-24 05:21:58 +0000632 nodeType = Node.ELEMENT_NODE
Martin v. Löwis787354c2003-01-25 15:28:29 +0000633 nodeValue = None
634 schemaType = _no_type
635
636 _magic_id_nodes = 0
637
638 _child_node_types = (Node.ELEMENT_NODE,
639 Node.PROCESSING_INSTRUCTION_NODE,
640 Node.COMMENT_NODE,
641 Node.TEXT_NODE,
642 Node.CDATA_SECTION_NODE,
643 Node.ENTITY_REFERENCE_NODE)
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000644
Fred Drake49a5d032001-11-30 22:21:58 +0000645 def __init__(self, tagName, namespaceURI=EMPTY_NAMESPACE, prefix=None,
Fred Drake1f549022000-09-24 05:21:58 +0000646 localName=None):
Fred Drake55c38192000-06-29 19:39:57 +0000647 self.tagName = self.nodeName = tagName
Fred Drake1f549022000-09-24 05:21:58 +0000648 self.prefix = prefix
649 self.namespaceURI = namespaceURI
Martin v. Löwis787354c2003-01-25 15:28:29 +0000650 self.childNodes = NodeList()
Fred Drake55c38192000-06-29 19:39:57 +0000651
Fred Drake4ccf4a12000-11-21 22:02:22 +0000652 self._attrs = {} # attributes are double-indexed:
653 self._attrsNS = {} # tagName -> Attribute
654 # URI,localName -> Attribute
655 # in the future: consider lazy generation
656 # of attribute objects this is too tricky
657 # for now because of headaches with
658 # namespaces.
659
Martin v. Löwis787354c2003-01-25 15:28:29 +0000660 def _get_localName(self):
661 return self.tagName.split(":", 1)[-1]
662
663 def _get_tagName(self):
664 return self.tagName
Fred Drake4ccf4a12000-11-21 22:02:22 +0000665
666 def unlink(self):
667 for attr in self._attrs.values():
668 attr.unlink()
669 self._attrs = None
670 self._attrsNS = None
671 Node.unlink(self)
Fred Drake55c38192000-06-29 19:39:57 +0000672
Fred Drake1f549022000-09-24 05:21:58 +0000673 def getAttribute(self, attname):
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000674 try:
675 return self._attrs[attname].value
676 except KeyError:
677 return ""
Fred Drake55c38192000-06-29 19:39:57 +0000678
Fred Drake1f549022000-09-24 05:21:58 +0000679 def getAttributeNS(self, namespaceURI, localName):
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000680 try:
681 return self._attrsNS[(namespaceURI, localName)].value
682 except KeyError:
683 return ""
Fred Drake1f549022000-09-24 05:21:58 +0000684
685 def setAttribute(self, attname, value):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000686 attr = self.getAttributeNode(attname)
687 if attr is None:
688 attr = Attr(attname)
689 # for performance
690 d = attr.__dict__
691 d["value"] = d["nodeValue"] = value
692 d["ownerDocument"] = self.ownerDocument
693 self.setAttributeNode(attr)
694 elif value != attr.value:
695 d = attr.__dict__
696 d["value"] = d["nodeValue"] = value
697 if attr.isId:
698 _clear_id_cache(self)
Fred Drake55c38192000-06-29 19:39:57 +0000699
Fred Drake1f549022000-09-24 05:21:58 +0000700 def setAttributeNS(self, namespaceURI, qualifiedName, value):
701 prefix, localname = _nssplit(qualifiedName)
Martin v. Löwis787354c2003-01-25 15:28:29 +0000702 attr = self.getAttributeNodeNS(namespaceURI, localname)
703 if attr is None:
704 # for performance
705 attr = Attr(qualifiedName, namespaceURI, localname, prefix)
706 d = attr.__dict__
707 d["prefix"] = prefix
708 d["nodeName"] = qualifiedName
709 d["value"] = d["nodeValue"] = value
710 d["ownerDocument"] = self.ownerDocument
711 self.setAttributeNode(attr)
712 else:
713 d = attr.__dict__
714 if value != attr.value:
715 d["value"] = d["nodeValue"] = value
716 if attr.isId:
717 _clear_id_cache(self)
718 if attr.prefix != prefix:
719 d["prefix"] = prefix
720 d["nodeName"] = qualifiedName
Fred Drake55c38192000-06-29 19:39:57 +0000721
Fred Drake1f549022000-09-24 05:21:58 +0000722 def getAttributeNode(self, attrname):
723 return self._attrs.get(attrname)
Paul Prescod73678da2000-07-01 04:58:47 +0000724
Fred Drake1f549022000-09-24 05:21:58 +0000725 def getAttributeNodeNS(self, namespaceURI, localName):
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000726 return self._attrsNS.get((namespaceURI, localName))
Paul Prescod73678da2000-07-01 04:58:47 +0000727
Fred Drake1f549022000-09-24 05:21:58 +0000728 def setAttributeNode(self, attr):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000729 if attr.ownerElement not in (None, self):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000730 raise xml.dom.InuseAttributeErr("attribute node already owned")
Martin v. Löwis787354c2003-01-25 15:28:29 +0000731 old1 = self._attrs.get(attr.name, None)
732 if old1 is not None:
733 self.removeAttributeNode(old1)
734 old2 = self._attrsNS.get((attr.namespaceURI, attr.localName), None)
735 if old2 is not None and old2 is not old1:
736 self.removeAttributeNode(old2)
737 _set_attribute_node(self, attr)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000738
Martin v. Löwis787354c2003-01-25 15:28:29 +0000739 if old1 is not attr:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000740 # It might have already been part of this node, in which case
741 # it doesn't represent a change, and should not be returned.
Martin v. Löwis787354c2003-01-25 15:28:29 +0000742 return old1
743 if old2 is not attr:
744 return old2
Fred Drake55c38192000-06-29 19:39:57 +0000745
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000746 setAttributeNodeNS = setAttributeNode
747
Fred Drake1f549022000-09-24 05:21:58 +0000748 def removeAttribute(self, name):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000749 try:
750 attr = self._attrs[name]
751 except KeyError:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000752 raise xml.dom.NotFoundErr()
Fred Drake1f549022000-09-24 05:21:58 +0000753 self.removeAttributeNode(attr)
Fred Drake55c38192000-06-29 19:39:57 +0000754
Fred Drake1f549022000-09-24 05:21:58 +0000755 def removeAttributeNS(self, namespaceURI, localName):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000756 try:
757 attr = self._attrsNS[(namespaceURI, localName)]
758 except KeyError:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000759 raise xml.dom.NotFoundErr()
Fred Drake1f549022000-09-24 05:21:58 +0000760 self.removeAttributeNode(attr)
Fred Drake55c38192000-06-29 19:39:57 +0000761
Fred Drake1f549022000-09-24 05:21:58 +0000762 def removeAttributeNode(self, node):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000763 if node is None:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000764 raise xml.dom.NotFoundErr()
Martin v. Löwis787354c2003-01-25 15:28:29 +0000765 try:
766 self._attrs[node.name]
767 except KeyError:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000768 raise xml.dom.NotFoundErr()
Martin v. Löwis787354c2003-01-25 15:28:29 +0000769 _clear_id_cache(self)
Paul Prescod73678da2000-07-01 04:58:47 +0000770 node.unlink()
Martin v. Löwis787354c2003-01-25 15:28:29 +0000771 # Restore this since the node is still useful and otherwise
772 # unlinked
773 node.ownerDocument = self.ownerDocument
Fred Drake16f63292000-10-23 18:09:50 +0000774
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000775 removeAttributeNodeNS = removeAttributeNode
776
Martin v. Löwis156c3372000-12-28 18:40:56 +0000777 def hasAttribute(self, name):
Benjamin Peterson6e3dbbd2009-10-09 22:15:50 +0000778 return name in self._attrs
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000779
Martin v. Löwis156c3372000-12-28 18:40:56 +0000780 def hasAttributeNS(self, namespaceURI, localName):
Benjamin Peterson6e3dbbd2009-10-09 22:15:50 +0000781 return (namespaceURI, localName) in self._attrsNS
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000782
Fred Drake1f549022000-09-24 05:21:58 +0000783 def getElementsByTagName(self, name):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000784 return _get_elements_by_tagName_helper(self, name, NodeList())
Fred Drake55c38192000-06-29 19:39:57 +0000785
Fred Drake1f549022000-09-24 05:21:58 +0000786 def getElementsByTagNameNS(self, namespaceURI, localName):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000787 return _get_elements_by_tagName_ns_helper(
788 self, namespaceURI, localName, NodeList())
Fred Drake55c38192000-06-29 19:39:57 +0000789
Fred Drake1f549022000-09-24 05:21:58 +0000790 def __repr__(self):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000791 return "<DOM Element: %s at %#x>" % (self.tagName, id(self))
Fred Drake55c38192000-06-29 19:39:57 +0000792
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000793 def writexml(self, writer, indent="", addindent="", newl=""):
794 # indent = current indentation
795 # addindent = indentation to add to higher levels
796 # newl = newline string
797 writer.write(indent+"<" + self.tagName)
Fred Drake16f63292000-10-23 18:09:50 +0000798
Fred Drake4ccf4a12000-11-21 22:02:22 +0000799 attrs = self._get_attributes()
800 a_names = attrs.keys()
Fred Drake55c38192000-06-29 19:39:57 +0000801 a_names.sort()
802
803 for a_name in a_names:
Fred Drake1f549022000-09-24 05:21:58 +0000804 writer.write(" %s=\"" % a_name)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000805 _write_data(writer, attrs[a_name].value)
Fred Drake55c38192000-06-29 19:39:57 +0000806 writer.write("\"")
807 if self.childNodes:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000808 writer.write(">%s"%(newl))
Fred Drake55c38192000-06-29 19:39:57 +0000809 for node in self.childNodes:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000810 node.writexml(writer,indent+addindent,addindent,newl)
811 writer.write("%s</%s>%s" % (indent,self.tagName,newl))
Fred Drake55c38192000-06-29 19:39:57 +0000812 else:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000813 writer.write("/>%s"%(newl))
Fred Drake55c38192000-06-29 19:39:57 +0000814
Fred Drake1f549022000-09-24 05:21:58 +0000815 def _get_attributes(self):
Fred Drake2998a552001-12-06 18:27:48 +0000816 return NamedNodeMap(self._attrs, self._attrsNS, self)
Fred Drake55c38192000-06-29 19:39:57 +0000817
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000818 def hasAttributes(self):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000819 if self._attrs:
820 return True
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000821 else:
Martin v. Löwis787354c2003-01-25 15:28:29 +0000822 return False
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000823
Martin v. Löwis787354c2003-01-25 15:28:29 +0000824 # DOM Level 3 attributes, based on the 22 Oct 2002 draft
825
826 def setIdAttribute(self, name):
827 idAttr = self.getAttributeNode(name)
828 self.setIdAttributeNode(idAttr)
829
830 def setIdAttributeNS(self, namespaceURI, localName):
831 idAttr = self.getAttributeNodeNS(namespaceURI, localName)
832 self.setIdAttributeNode(idAttr)
833
834 def setIdAttributeNode(self, idAttr):
835 if idAttr is None or not self.isSameNode(idAttr.ownerElement):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000836 raise xml.dom.NotFoundErr()
Martin v. Löwis787354c2003-01-25 15:28:29 +0000837 if _get_containing_entref(self) is not None:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000838 raise xml.dom.NoModificationAllowedErr()
Martin v. Löwis787354c2003-01-25 15:28:29 +0000839 if not idAttr._is_id:
840 idAttr.__dict__['_is_id'] = True
841 self._magic_id_nodes += 1
842 self.ownerDocument._magic_id_count += 1
843 _clear_id_cache(self)
844
845defproperty(Element, "attributes",
846 doc="NamedNodeMap of attributes on the element.")
847defproperty(Element, "localName",
848 doc="Namespace-local name of this element.")
849
850
851def _set_attribute_node(element, attr):
852 _clear_id_cache(element)
853 element._attrs[attr.name] = attr
854 element._attrsNS[(attr.namespaceURI, attr.localName)] = attr
855
856 # This creates a circular reference, but Element.unlink()
857 # breaks the cycle since the references to the attribute
858 # dictionaries are tossed.
859 attr.__dict__['ownerElement'] = element
860
861
862class Childless:
863 """Mixin that makes childless-ness easy to implement and avoids
864 the complexity of the Node methods that deal with children.
865 """
866
Fred Drake4ccf4a12000-11-21 22:02:22 +0000867 attributes = None
Martin v. Löwis787354c2003-01-25 15:28:29 +0000868 childNodes = EmptyNodeList()
869 firstChild = None
870 lastChild = None
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000871
Martin v. Löwis787354c2003-01-25 15:28:29 +0000872 def _get_firstChild(self):
873 return None
Fred Drake55c38192000-06-29 19:39:57 +0000874
Martin v. Löwis787354c2003-01-25 15:28:29 +0000875 def _get_lastChild(self):
876 return None
Fred Drake1f549022000-09-24 05:21:58 +0000877
Martin v. Löwis787354c2003-01-25 15:28:29 +0000878 def appendChild(self, node):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000879 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +0000880 self.nodeName + " nodes cannot have children")
881
882 def hasChildNodes(self):
883 return False
884
885 def insertBefore(self, newChild, refChild):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000886 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +0000887 self.nodeName + " nodes do not have children")
888
889 def removeChild(self, oldChild):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000890 raise xml.dom.NotFoundErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +0000891 self.nodeName + " nodes do not have children")
892
893 def replaceChild(self, newChild, oldChild):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000894 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +0000895 self.nodeName + " nodes do not have children")
896
897
898class ProcessingInstruction(Childless, Node):
Fred Drake1f549022000-09-24 05:21:58 +0000899 nodeType = Node.PROCESSING_INSTRUCTION_NODE
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000900
Fred Drake1f549022000-09-24 05:21:58 +0000901 def __init__(self, target, data):
Fred Drake55c38192000-06-29 19:39:57 +0000902 self.target = self.nodeName = target
903 self.data = self.nodeValue = data
Fred Drake55c38192000-06-29 19:39:57 +0000904
Martin v. Löwis787354c2003-01-25 15:28:29 +0000905 def _get_data(self):
906 return self.data
907 def _set_data(self, value):
908 d = self.__dict__
909 d['data'] = d['nodeValue'] = value
910
911 def _get_target(self):
912 return self.target
913 def _set_target(self, value):
914 d = self.__dict__
915 d['target'] = d['nodeName'] = value
916
917 def __setattr__(self, name, value):
918 if name == "data" or name == "nodeValue":
919 self.__dict__['data'] = self.__dict__['nodeValue'] = value
920 elif name == "target" or name == "nodeName":
921 self.__dict__['target'] = self.__dict__['nodeName'] = value
922 else:
923 self.__dict__[name] = value
924
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000925 def writexml(self, writer, indent="", addindent="", newl=""):
926 writer.write("%s<?%s %s?>%s" % (indent,self.target, self.data, newl))
Fred Drake55c38192000-06-29 19:39:57 +0000927
Martin v. Löwis787354c2003-01-25 15:28:29 +0000928
929class CharacterData(Childless, Node):
930 def _get_length(self):
931 return len(self.data)
932 __len__ = _get_length
933
934 def _get_data(self):
935 return self.__dict__['data']
936 def _set_data(self, data):
937 d = self.__dict__
938 d['data'] = d['nodeValue'] = data
939
940 _get_nodeValue = _get_data
941 _set_nodeValue = _set_data
942
943 def __setattr__(self, name, value):
944 if name == "data" or name == "nodeValue":
945 self.__dict__['data'] = self.__dict__['nodeValue'] = value
946 else:
947 self.__dict__[name] = value
Fred Drake87432f42001-04-04 14:09:46 +0000948
Fred Drake55c38192000-06-29 19:39:57 +0000949 def __repr__(self):
Martin v. Löwis787354c2003-01-25 15:28:29 +0000950 data = self.data
951 if len(data) > 10:
Fred Drake1f549022000-09-24 05:21:58 +0000952 dotdotdot = "..."
Fred Drake55c38192000-06-29 19:39:57 +0000953 else:
Fred Drake1f549022000-09-24 05:21:58 +0000954 dotdotdot = ""
Facundo Batistaf5ade632007-10-24 19:11:08 +0000955 return '<DOM %s node "%r%s">' % (
Martin v. Löwis787354c2003-01-25 15:28:29 +0000956 self.__class__.__name__, data[0:10], dotdotdot)
Fred Drake87432f42001-04-04 14:09:46 +0000957
958 def substringData(self, offset, count):
959 if offset < 0:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000960 raise xml.dom.IndexSizeErr("offset cannot be negative")
Fred Drake87432f42001-04-04 14:09:46 +0000961 if offset >= len(self.data):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000962 raise xml.dom.IndexSizeErr("offset cannot be beyond end of data")
Fred Drake87432f42001-04-04 14:09:46 +0000963 if count < 0:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000964 raise xml.dom.IndexSizeErr("count cannot be negative")
Fred Drake87432f42001-04-04 14:09:46 +0000965 return self.data[offset:offset+count]
966
967 def appendData(self, arg):
968 self.data = self.data + arg
Fred Drake87432f42001-04-04 14:09:46 +0000969
970 def insertData(self, offset, arg):
971 if offset < 0:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000972 raise xml.dom.IndexSizeErr("offset cannot be negative")
Fred Drake87432f42001-04-04 14:09:46 +0000973 if offset >= len(self.data):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000974 raise xml.dom.IndexSizeErr("offset cannot be beyond end of data")
Fred Drake87432f42001-04-04 14:09:46 +0000975 if arg:
976 self.data = "%s%s%s" % (
977 self.data[:offset], arg, self.data[offset:])
Fred Drake87432f42001-04-04 14:09:46 +0000978
979 def deleteData(self, offset, count):
980 if offset < 0:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000981 raise xml.dom.IndexSizeErr("offset cannot be negative")
Fred Drake87432f42001-04-04 14:09:46 +0000982 if offset >= len(self.data):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000983 raise xml.dom.IndexSizeErr("offset cannot be beyond end of data")
Fred Drake87432f42001-04-04 14:09:46 +0000984 if count < 0:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000985 raise xml.dom.IndexSizeErr("count cannot be negative")
Fred Drake87432f42001-04-04 14:09:46 +0000986 if count:
987 self.data = self.data[:offset] + self.data[offset+count:]
Fred Drake87432f42001-04-04 14:09:46 +0000988
989 def replaceData(self, offset, count, arg):
990 if offset < 0:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000991 raise xml.dom.IndexSizeErr("offset cannot be negative")
Fred Drake87432f42001-04-04 14:09:46 +0000992 if offset >= len(self.data):
Fred Drakefbdeaad2006-07-29 16:56:15 +0000993 raise xml.dom.IndexSizeErr("offset cannot be beyond end of data")
Fred Drake87432f42001-04-04 14:09:46 +0000994 if count < 0:
Fred Drakefbdeaad2006-07-29 16:56:15 +0000995 raise xml.dom.IndexSizeErr("count cannot be negative")
Fred Drake87432f42001-04-04 14:09:46 +0000996 if count:
997 self.data = "%s%s%s" % (
998 self.data[:offset], arg, self.data[offset+count:])
Martin v. Löwis787354c2003-01-25 15:28:29 +0000999
1000defproperty(CharacterData, "length", doc="Length of the string data.")
1001
Fred Drake87432f42001-04-04 14:09:46 +00001002
1003class Text(CharacterData):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001004 # Make sure we don't add an instance __dict__ if we don't already
1005 # have one, at least when that's possible:
Martin v. Löwis995359c2003-01-26 08:59:32 +00001006 # XXX this does not work, CharacterData is an old-style class
1007 # __slots__ = ()
Martin v. Löwis787354c2003-01-25 15:28:29 +00001008
Fred Drake87432f42001-04-04 14:09:46 +00001009 nodeType = Node.TEXT_NODE
1010 nodeName = "#text"
1011 attributes = None
Fred Drake55c38192000-06-29 19:39:57 +00001012
Fred Drakef7cf40d2000-12-14 18:16:11 +00001013 def splitText(self, offset):
1014 if offset < 0 or offset > len(self.data):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001015 raise xml.dom.IndexSizeErr("illegal offset value")
Martin v. Löwis787354c2003-01-25 15:28:29 +00001016 newText = self.__class__()
1017 newText.data = self.data[offset:]
1018 newText.ownerDocument = self.ownerDocument
Fred Drakef7cf40d2000-12-14 18:16:11 +00001019 next = self.nextSibling
1020 if self.parentNode and self in self.parentNode.childNodes:
1021 if next is None:
1022 self.parentNode.appendChild(newText)
1023 else:
1024 self.parentNode.insertBefore(newText, next)
1025 self.data = self.data[:offset]
1026 return newText
1027
Martin v. Löwis46fa39a2001-02-06 00:14:08 +00001028 def writexml(self, writer, indent="", addindent="", newl=""):
1029 _write_data(writer, "%s%s%s"%(indent, self.data, newl))
Fred Drake55c38192000-06-29 19:39:57 +00001030
Martin v. Löwis787354c2003-01-25 15:28:29 +00001031 # DOM Level 3 (WD 9 April 2002)
1032
1033 def _get_wholeText(self):
1034 L = [self.data]
1035 n = self.previousSibling
1036 while n is not None:
1037 if n.nodeType in (Node.TEXT_NODE, Node.CDATA_SECTION_NODE):
1038 L.insert(0, n.data)
1039 n = n.previousSibling
1040 else:
1041 break
1042 n = self.nextSibling
1043 while n is not None:
1044 if n.nodeType in (Node.TEXT_NODE, Node.CDATA_SECTION_NODE):
1045 L.append(n.data)
1046 n = n.nextSibling
1047 else:
1048 break
1049 return ''.join(L)
1050
1051 def replaceWholeText(self, content):
1052 # XXX This needs to be seriously changed if minidom ever
1053 # supports EntityReference nodes.
1054 parent = self.parentNode
1055 n = self.previousSibling
1056 while n is not None:
1057 if n.nodeType in (Node.TEXT_NODE, Node.CDATA_SECTION_NODE):
1058 next = n.previousSibling
1059 parent.removeChild(n)
1060 n = next
1061 else:
1062 break
1063 n = self.nextSibling
1064 if not content:
1065 parent.removeChild(self)
1066 while n is not None:
1067 if n.nodeType in (Node.TEXT_NODE, Node.CDATA_SECTION_NODE):
1068 next = n.nextSibling
1069 parent.removeChild(n)
1070 n = next
1071 else:
1072 break
1073 if content:
1074 d = self.__dict__
1075 d['data'] = content
1076 d['nodeValue'] = content
1077 return self
1078 else:
1079 return None
1080
1081 def _get_isWhitespaceInElementContent(self):
1082 if self.data.strip():
1083 return False
1084 elem = _get_containing_element(self)
1085 if elem is None:
1086 return False
1087 info = self.ownerDocument._get_elem_info(elem)
1088 if info is None:
1089 return False
1090 else:
1091 return info.isElementContent()
1092
1093defproperty(Text, "isWhitespaceInElementContent",
1094 doc="True iff this text node contains only whitespace"
1095 " and is in element content.")
1096defproperty(Text, "wholeText",
1097 doc="The text of all logically-adjacent text nodes.")
1098
1099
1100def _get_containing_element(node):
1101 c = node.parentNode
1102 while c is not None:
1103 if c.nodeType == Node.ELEMENT_NODE:
1104 return c
1105 c = c.parentNode
1106 return None
1107
1108def _get_containing_entref(node):
1109 c = node.parentNode
1110 while c is not None:
1111 if c.nodeType == Node.ENTITY_REFERENCE_NODE:
1112 return c
1113 c = c.parentNode
1114 return None
1115
1116
1117class Comment(Childless, CharacterData):
1118 nodeType = Node.COMMENT_NODE
1119 nodeName = "#comment"
1120
1121 def __init__(self, data):
1122 self.data = self.nodeValue = data
1123
1124 def writexml(self, writer, indent="", addindent="", newl=""):
Martin v. Löwis27e4a172008-05-23 15:18:28 +00001125 if "--" in self.data:
1126 raise ValueError("'--' is not allowed in a comment node")
Martin v. Löwis787354c2003-01-25 15:28:29 +00001127 writer.write("%s<!--%s-->%s" % (indent, self.data, newl))
1128
Fred Drake87432f42001-04-04 14:09:46 +00001129
1130class CDATASection(Text):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001131 # Make sure we don't add an instance __dict__ if we don't already
1132 # have one, at least when that's possible:
Martin v. Löwis995359c2003-01-26 08:59:32 +00001133 # XXX this does not work, Text is an old-style class
1134 # __slots__ = ()
Martin v. Löwis787354c2003-01-25 15:28:29 +00001135
Fred Drake87432f42001-04-04 14:09:46 +00001136 nodeType = Node.CDATA_SECTION_NODE
1137 nodeName = "#cdata-section"
1138
1139 def writexml(self, writer, indent="", addindent="", newl=""):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001140 if self.data.find("]]>") >= 0:
1141 raise ValueError("']]>' not allowed in a CDATA section")
Guido van Rossum5b5e0b92001-09-19 13:28:25 +00001142 writer.write("<![CDATA[%s]]>" % self.data)
Fred Drake87432f42001-04-04 14:09:46 +00001143
1144
Fred Drake9ea179f2006-04-06 01:29:04 +00001145class ReadOnlySequentialNamedNodeMap(object):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001146 __slots__ = '_seq',
1147
1148 def __init__(self, seq=()):
1149 # seq should be a list or tuple
1150 self._seq = seq
1151
1152 def __len__(self):
1153 return len(self._seq)
1154
1155 def _get_length(self):
1156 return len(self._seq)
1157
1158 def getNamedItem(self, name):
1159 for n in self._seq:
1160 if n.nodeName == name:
1161 return n
1162
1163 def getNamedItemNS(self, namespaceURI, localName):
1164 for n in self._seq:
1165 if n.namespaceURI == namespaceURI and n.localName == localName:
1166 return n
1167
1168 def __getitem__(self, name_or_tuple):
Fred Drake70d044b2006-04-06 01:32:26 +00001169 if isinstance(name_or_tuple, tuple):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001170 node = self.getNamedItemNS(*name_or_tuple)
1171 else:
1172 node = self.getNamedItem(name_or_tuple)
1173 if node is None:
1174 raise KeyError, name_or_tuple
1175 return node
1176
1177 def item(self, index):
1178 if index < 0:
1179 return None
1180 try:
1181 return self._seq[index]
1182 except IndexError:
1183 return None
1184
1185 def removeNamedItem(self, name):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001186 raise xml.dom.NoModificationAllowedErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001187 "NamedNodeMap instance is read-only")
1188
1189 def removeNamedItemNS(self, namespaceURI, localName):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001190 raise xml.dom.NoModificationAllowedErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001191 "NamedNodeMap instance is read-only")
1192
1193 def setNamedItem(self, node):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001194 raise xml.dom.NoModificationAllowedErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001195 "NamedNodeMap instance is read-only")
1196
1197 def setNamedItemNS(self, node):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001198 raise xml.dom.NoModificationAllowedErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001199 "NamedNodeMap instance is read-only")
1200
1201 def __getstate__(self):
1202 return [self._seq]
1203
1204 def __setstate__(self, state):
1205 self._seq = state[0]
1206
1207defproperty(ReadOnlySequentialNamedNodeMap, "length",
1208 doc="Number of entries in the NamedNodeMap.")
Paul Prescod73678da2000-07-01 04:58:47 +00001209
Fred Drakef7cf40d2000-12-14 18:16:11 +00001210
Martin v. Löwis787354c2003-01-25 15:28:29 +00001211class Identified:
1212 """Mix-in class that supports the publicId and systemId attributes."""
1213
Martin v. Löwis995359c2003-01-26 08:59:32 +00001214 # XXX this does not work, this is an old-style class
1215 # __slots__ = 'publicId', 'systemId'
Martin v. Löwis787354c2003-01-25 15:28:29 +00001216
1217 def _identified_mixin_init(self, publicId, systemId):
1218 self.publicId = publicId
1219 self.systemId = systemId
1220
1221 def _get_publicId(self):
1222 return self.publicId
1223
1224 def _get_systemId(self):
1225 return self.systemId
1226
1227class DocumentType(Identified, Childless, Node):
Fred Drakef7cf40d2000-12-14 18:16:11 +00001228 nodeType = Node.DOCUMENT_TYPE_NODE
1229 nodeValue = None
Fred Drakef7cf40d2000-12-14 18:16:11 +00001230 name = None
1231 publicId = None
1232 systemId = None
Fred Drakedc806702001-04-05 14:41:30 +00001233 internalSubset = None
Fred Drakef7cf40d2000-12-14 18:16:11 +00001234
1235 def __init__(self, qualifiedName):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001236 self.entities = ReadOnlySequentialNamedNodeMap()
1237 self.notations = ReadOnlySequentialNamedNodeMap()
Fred Drakef7cf40d2000-12-14 18:16:11 +00001238 if qualifiedName:
1239 prefix, localname = _nssplit(qualifiedName)
1240 self.name = localname
Martin v. Löwis787354c2003-01-25 15:28:29 +00001241 self.nodeName = self.name
1242
1243 def _get_internalSubset(self):
1244 return self.internalSubset
1245
1246 def cloneNode(self, deep):
1247 if self.ownerDocument is None:
1248 # it's ok
1249 clone = DocumentType(None)
1250 clone.name = self.name
1251 clone.nodeName = self.name
Fred Drakefbdeaad2006-07-29 16:56:15 +00001252 operation = xml.dom.UserDataHandler.NODE_CLONED
Martin v. Löwis787354c2003-01-25 15:28:29 +00001253 if deep:
1254 clone.entities._seq = []
1255 clone.notations._seq = []
1256 for n in self.notations._seq:
1257 notation = Notation(n.nodeName, n.publicId, n.systemId)
1258 clone.notations._seq.append(notation)
1259 n._call_user_data_handler(operation, n, notation)
1260 for e in self.entities._seq:
1261 entity = Entity(e.nodeName, e.publicId, e.systemId,
1262 e.notationName)
1263 entity.actualEncoding = e.actualEncoding
1264 entity.encoding = e.encoding
1265 entity.version = e.version
1266 clone.entities._seq.append(entity)
1267 e._call_user_data_handler(operation, n, entity)
1268 self._call_user_data_handler(operation, self, clone)
1269 return clone
1270 else:
1271 return None
1272
1273 def writexml(self, writer, indent="", addindent="", newl=""):
1274 writer.write("<!DOCTYPE ")
1275 writer.write(self.name)
1276 if self.publicId:
Georg Brandl175a7dc2005-08-25 22:02:43 +00001277 writer.write("%s PUBLIC '%s'%s '%s'"
1278 % (newl, self.publicId, newl, self.systemId))
Martin v. Löwis787354c2003-01-25 15:28:29 +00001279 elif self.systemId:
Georg Brandl175a7dc2005-08-25 22:02:43 +00001280 writer.write("%s SYSTEM '%s'" % (newl, self.systemId))
Martin v. Löwis787354c2003-01-25 15:28:29 +00001281 if self.internalSubset is not None:
1282 writer.write(" [")
1283 writer.write(self.internalSubset)
1284 writer.write("]")
Georg Brandl175a7dc2005-08-25 22:02:43 +00001285 writer.write(">"+newl)
Martin v. Löwis787354c2003-01-25 15:28:29 +00001286
1287class Entity(Identified, Node):
1288 attributes = None
1289 nodeType = Node.ENTITY_NODE
1290 nodeValue = None
1291
1292 actualEncoding = None
1293 encoding = None
1294 version = None
1295
1296 def __init__(self, name, publicId, systemId, notation):
1297 self.nodeName = name
1298 self.notationName = notation
1299 self.childNodes = NodeList()
1300 self._identified_mixin_init(publicId, systemId)
1301
1302 def _get_actualEncoding(self):
1303 return self.actualEncoding
1304
1305 def _get_encoding(self):
1306 return self.encoding
1307
1308 def _get_version(self):
1309 return self.version
1310
1311 def appendChild(self, newChild):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001312 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001313 "cannot append children to an entity node")
1314
1315 def insertBefore(self, newChild, refChild):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001316 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001317 "cannot insert children below an entity node")
1318
1319 def removeChild(self, oldChild):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001320 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001321 "cannot remove children from an entity node")
1322
1323 def replaceChild(self, newChild, oldChild):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001324 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001325 "cannot replace children of an entity node")
1326
1327class Notation(Identified, Childless, Node):
1328 nodeType = Node.NOTATION_NODE
1329 nodeValue = None
1330
1331 def __init__(self, name, publicId, systemId):
1332 self.nodeName = name
1333 self._identified_mixin_init(publicId, systemId)
Fred Drakef7cf40d2000-12-14 18:16:11 +00001334
1335
Martin v. Löwis787354c2003-01-25 15:28:29 +00001336class DOMImplementation(DOMImplementationLS):
1337 _features = [("core", "1.0"),
1338 ("core", "2.0"),
1339 ("core", "3.0"),
1340 ("core", None),
1341 ("xml", "1.0"),
1342 ("xml", "2.0"),
1343 ("xml", "3.0"),
1344 ("xml", None),
1345 ("ls-load", "3.0"),
1346 ("ls-load", None),
1347 ]
1348
Fred Drakef7cf40d2000-12-14 18:16:11 +00001349 def hasFeature(self, feature, version):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001350 if version == "":
1351 version = None
1352 return (feature.lower(), version) in self._features
Fred Drakef7cf40d2000-12-14 18:16:11 +00001353
1354 def createDocument(self, namespaceURI, qualifiedName, doctype):
1355 if doctype and doctype.parentNode is not None:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001356 raise xml.dom.WrongDocumentErr(
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +00001357 "doctype object owned by another DOM tree")
Martin v. Löwis787354c2003-01-25 15:28:29 +00001358 doc = self._create_document()
1359
1360 add_root_element = not (namespaceURI is None
1361 and qualifiedName is None
1362 and doctype is None)
1363
1364 if not qualifiedName and add_root_element:
Martin v. Löwisb417be22001-02-06 01:16:06 +00001365 # The spec is unclear what to raise here; SyntaxErr
1366 # would be the other obvious candidate. Since Xerces raises
1367 # InvalidCharacterErr, and since SyntaxErr is not listed
1368 # for createDocument, that seems to be the better choice.
1369 # XXX: need to check for illegal characters here and in
1370 # createElement.
Martin v. Löwis787354c2003-01-25 15:28:29 +00001371
1372 # DOM Level III clears this up when talking about the return value
1373 # of this function. If namespaceURI, qName and DocType are
1374 # Null the document is returned without a document element
1375 # Otherwise if doctype or namespaceURI are not None
1376 # Then we go back to the above problem
Fred Drakefbdeaad2006-07-29 16:56:15 +00001377 raise xml.dom.InvalidCharacterErr("Element with no name")
Martin v. Löwis787354c2003-01-25 15:28:29 +00001378
1379 if add_root_element:
1380 prefix, localname = _nssplit(qualifiedName)
1381 if prefix == "xml" \
1382 and namespaceURI != "http://www.w3.org/XML/1998/namespace":
Fred Drakefbdeaad2006-07-29 16:56:15 +00001383 raise xml.dom.NamespaceErr("illegal use of 'xml' prefix")
Martin v. Löwis787354c2003-01-25 15:28:29 +00001384 if prefix and not namespaceURI:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001385 raise xml.dom.NamespaceErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001386 "illegal use of prefix without namespaces")
1387 element = doc.createElementNS(namespaceURI, qualifiedName)
1388 if doctype:
1389 doc.appendChild(doctype)
1390 doc.appendChild(element)
1391
1392 if doctype:
1393 doctype.parentNode = doctype.ownerDocument = doc
1394
Fred Drakef7cf40d2000-12-14 18:16:11 +00001395 doc.doctype = doctype
1396 doc.implementation = self
1397 return doc
1398
1399 def createDocumentType(self, qualifiedName, publicId, systemId):
1400 doctype = DocumentType(qualifiedName)
1401 doctype.publicId = publicId
1402 doctype.systemId = systemId
1403 return doctype
1404
Martin v. Löwis787354c2003-01-25 15:28:29 +00001405 # DOM Level 3 (WD 9 April 2002)
1406
1407 def getInterface(self, feature):
1408 if self.hasFeature(feature, None):
1409 return self
1410 else:
1411 return None
1412
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001413 # internal
Martin v. Löwis787354c2003-01-25 15:28:29 +00001414 def _create_document(self):
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001415 return Document()
Fred Drakef7cf40d2000-12-14 18:16:11 +00001416
Fred Drake9ea179f2006-04-06 01:29:04 +00001417class ElementInfo(object):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001418 """Object that represents content-model information for an element.
1419
1420 This implementation is not expected to be used in practice; DOM
1421 builders should provide implementations which do the right thing
1422 using information available to it.
1423
1424 """
1425
1426 __slots__ = 'tagName',
1427
1428 def __init__(self, name):
1429 self.tagName = name
1430
1431 def getAttributeType(self, aname):
1432 return _no_type
1433
1434 def getAttributeTypeNS(self, namespaceURI, localName):
1435 return _no_type
1436
1437 def isElementContent(self):
1438 return False
1439
1440 def isEmpty(self):
1441 """Returns true iff this element is declared to have an EMPTY
1442 content model."""
1443 return False
1444
1445 def isId(self, aname):
1446 """Returns true iff the named attribte is a DTD-style ID."""
1447 return False
1448
1449 def isIdNS(self, namespaceURI, localName):
1450 """Returns true iff the identified attribute is a DTD-style ID."""
1451 return False
1452
1453 def __getstate__(self):
1454 return self.tagName
1455
1456 def __setstate__(self, state):
1457 self.tagName = state
1458
1459def _clear_id_cache(node):
1460 if node.nodeType == Node.DOCUMENT_NODE:
1461 node._id_cache.clear()
1462 node._id_search_stack = None
1463 elif _in_document(node):
1464 node.ownerDocument._id_cache.clear()
1465 node.ownerDocument._id_search_stack= None
1466
1467class Document(Node, DocumentLS):
1468 _child_node_types = (Node.ELEMENT_NODE, Node.PROCESSING_INSTRUCTION_NODE,
1469 Node.COMMENT_NODE, Node.DOCUMENT_TYPE_NODE)
1470
Fred Drake1f549022000-09-24 05:21:58 +00001471 nodeType = Node.DOCUMENT_NODE
Fred Drake4ccf4a12000-11-21 22:02:22 +00001472 nodeName = "#document"
1473 nodeValue = None
1474 attributes = None
Fred Drakef7cf40d2000-12-14 18:16:11 +00001475 doctype = None
1476 parentNode = None
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001477 previousSibling = nextSibling = None
Fred Drakef7cf40d2000-12-14 18:16:11 +00001478
1479 implementation = DOMImplementation()
Martin v. Löwis787354c2003-01-25 15:28:29 +00001480
1481 # Document attributes from Level 3 (WD 9 April 2002)
1482
1483 actualEncoding = None
1484 encoding = None
1485 standalone = None
1486 version = None
1487 strictErrorChecking = False
1488 errorHandler = None
1489 documentURI = None
1490
1491 _magic_id_count = 0
1492
1493 def __init__(self):
1494 self.childNodes = NodeList()
1495 # mapping of (namespaceURI, localName) -> ElementInfo
1496 # and tagName -> ElementInfo
1497 self._elem_info = {}
1498 self._id_cache = {}
1499 self._id_search_stack = None
1500
1501 def _get_elem_info(self, element):
1502 if element.namespaceURI:
1503 key = element.namespaceURI, element.localName
1504 else:
1505 key = element.tagName
1506 return self._elem_info.get(key)
1507
1508 def _get_actualEncoding(self):
1509 return self.actualEncoding
1510
1511 def _get_doctype(self):
1512 return self.doctype
1513
1514 def _get_documentURI(self):
1515 return self.documentURI
1516
1517 def _get_encoding(self):
1518 return self.encoding
1519
1520 def _get_errorHandler(self):
1521 return self.errorHandler
1522
1523 def _get_standalone(self):
1524 return self.standalone
1525
1526 def _get_strictErrorChecking(self):
1527 return self.strictErrorChecking
1528
1529 def _get_version(self):
1530 return self.version
Fred Drake55c38192000-06-29 19:39:57 +00001531
Fred Drake1f549022000-09-24 05:21:58 +00001532 def appendChild(self, node):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001533 if node.nodeType not in self._child_node_types:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001534 raise xml.dom.HierarchyRequestErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001535 "%s cannot be child of %s" % (repr(node), repr(self)))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +00001536 if node.parentNode is not None:
Martin v. Löwis787354c2003-01-25 15:28:29 +00001537 # This needs to be done before the next test since this
1538 # may *be* the document element, in which case it should
1539 # end up re-ordered to the end.
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +00001540 node.parentNode.removeChild(node)
1541
Fred Drakef7cf40d2000-12-14 18:16:11 +00001542 if node.nodeType == Node.ELEMENT_NODE \
1543 and self._get_documentElement():
Fred Drakefbdeaad2006-07-29 16:56:15 +00001544 raise xml.dom.HierarchyRequestErr(
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +00001545 "two document elements disallowed")
Fred Drake4ccf4a12000-11-21 22:02:22 +00001546 return Node.appendChild(self, node)
Paul Prescod73678da2000-07-01 04:58:47 +00001547
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +00001548 def removeChild(self, oldChild):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001549 try:
1550 self.childNodes.remove(oldChild)
1551 except ValueError:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001552 raise xml.dom.NotFoundErr()
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +00001553 oldChild.nextSibling = oldChild.previousSibling = None
1554 oldChild.parentNode = None
1555 if self.documentElement is oldChild:
1556 self.documentElement = None
Martin v. Löwis52ce0d02001-01-27 08:47:37 +00001557
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +00001558 return oldChild
1559
Fred Drakef7cf40d2000-12-14 18:16:11 +00001560 def _get_documentElement(self):
1561 for node in self.childNodes:
1562 if node.nodeType == Node.ELEMENT_NODE:
1563 return node
1564
1565 def unlink(self):
1566 if self.doctype is not None:
1567 self.doctype.unlink()
1568 self.doctype = None
1569 Node.unlink(self)
1570
Martin v. Löwis787354c2003-01-25 15:28:29 +00001571 def cloneNode(self, deep):
1572 if not deep:
1573 return None
1574 clone = self.implementation.createDocument(None, None, None)
1575 clone.encoding = self.encoding
1576 clone.standalone = self.standalone
1577 clone.version = self.version
1578 for n in self.childNodes:
1579 childclone = _clone_node(n, deep, clone)
1580 assert childclone.ownerDocument.isSameNode(clone)
1581 clone.childNodes.append(childclone)
1582 if childclone.nodeType == Node.DOCUMENT_NODE:
1583 assert clone.documentElement is None
1584 elif childclone.nodeType == Node.DOCUMENT_TYPE_NODE:
1585 assert clone.doctype is None
1586 clone.doctype = childclone
1587 childclone.parentNode = clone
Fred Drakefbdeaad2006-07-29 16:56:15 +00001588 self._call_user_data_handler(xml.dom.UserDataHandler.NODE_CLONED,
Martin v. Löwis787354c2003-01-25 15:28:29 +00001589 self, clone)
1590 return clone
1591
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001592 def createDocumentFragment(self):
1593 d = DocumentFragment()
Martin v. Löwis787354c2003-01-25 15:28:29 +00001594 d.ownerDocument = self
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001595 return d
Fred Drake55c38192000-06-29 19:39:57 +00001596
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001597 def createElement(self, tagName):
1598 e = Element(tagName)
1599 e.ownerDocument = self
1600 return e
Fred Drake55c38192000-06-29 19:39:57 +00001601
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001602 def createTextNode(self, data):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001603 if not isinstance(data, StringTypes):
1604 raise TypeError, "node contents must be a string"
1605 t = Text()
1606 t.data = data
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001607 t.ownerDocument = self
1608 return t
Fred Drake55c38192000-06-29 19:39:57 +00001609
Fred Drake87432f42001-04-04 14:09:46 +00001610 def createCDATASection(self, data):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001611 if not isinstance(data, StringTypes):
1612 raise TypeError, "node contents must be a string"
1613 c = CDATASection()
1614 c.data = data
Fred Drake87432f42001-04-04 14:09:46 +00001615 c.ownerDocument = self
1616 return c
1617
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001618 def createComment(self, data):
1619 c = Comment(data)
1620 c.ownerDocument = self
1621 return c
Fred Drake55c38192000-06-29 19:39:57 +00001622
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001623 def createProcessingInstruction(self, target, data):
1624 p = ProcessingInstruction(target, data)
1625 p.ownerDocument = self
1626 return p
1627
1628 def createAttribute(self, qName):
1629 a = Attr(qName)
1630 a.ownerDocument = self
Martin v. Löwiscb67ea12001-03-31 16:30:40 +00001631 a.value = ""
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001632 return a
Fred Drake55c38192000-06-29 19:39:57 +00001633
1634 def createElementNS(self, namespaceURI, qualifiedName):
Fred Drake4ccf4a12000-11-21 22:02:22 +00001635 prefix, localName = _nssplit(qualifiedName)
Martin v. Löwis787354c2003-01-25 15:28:29 +00001636 e = Element(qualifiedName, namespaceURI, prefix)
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001637 e.ownerDocument = self
1638 return e
Fred Drake55c38192000-06-29 19:39:57 +00001639
1640 def createAttributeNS(self, namespaceURI, qualifiedName):
Fred Drake4ccf4a12000-11-21 22:02:22 +00001641 prefix, localName = _nssplit(qualifiedName)
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001642 a = Attr(qualifiedName, namespaceURI, localName, prefix)
1643 a.ownerDocument = self
Martin v. Löwiscb67ea12001-03-31 16:30:40 +00001644 a.value = ""
Martin v. Löwis126f2f62001-03-13 10:50:13 +00001645 return a
Fred Drake55c38192000-06-29 19:39:57 +00001646
Martin v. Löwis787354c2003-01-25 15:28:29 +00001647 # A couple of implementation-specific helpers to create node types
1648 # not supported by the W3C DOM specs:
1649
1650 def _create_entity(self, name, publicId, systemId, notationName):
1651 e = Entity(name, publicId, systemId, notationName)
1652 e.ownerDocument = self
1653 return e
1654
1655 def _create_notation(self, name, publicId, systemId):
1656 n = Notation(name, publicId, systemId)
1657 n.ownerDocument = self
1658 return n
1659
1660 def getElementById(self, id):
Brett Cannon5dd504d2008-08-04 00:23:58 +00001661 if id in self._id_cache:
Martin v. Löwis787354c2003-01-25 15:28:29 +00001662 return self._id_cache[id]
1663 if not (self._elem_info or self._magic_id_count):
1664 return None
1665
1666 stack = self._id_search_stack
1667 if stack is None:
1668 # we never searched before, or the cache has been cleared
1669 stack = [self.documentElement]
1670 self._id_search_stack = stack
1671 elif not stack:
1672 # Previous search was completed and cache is still valid;
1673 # no matching node.
1674 return None
1675
1676 result = None
1677 while stack:
1678 node = stack.pop()
1679 # add child elements to stack for continued searching
1680 stack.extend([child for child in node.childNodes
1681 if child.nodeType in _nodeTypes_with_children])
1682 # check this node
1683 info = self._get_elem_info(node)
1684 if info:
1685 # We have to process all ID attributes before
1686 # returning in order to get all the attributes set to
1687 # be IDs using Element.setIdAttribute*().
1688 for attr in node.attributes.values():
1689 if attr.namespaceURI:
1690 if info.isIdNS(attr.namespaceURI, attr.localName):
1691 self._id_cache[attr.value] = node
1692 if attr.value == id:
1693 result = node
1694 elif not node._magic_id_nodes:
1695 break
1696 elif info.isId(attr.name):
1697 self._id_cache[attr.value] = node
1698 if attr.value == id:
1699 result = node
1700 elif not node._magic_id_nodes:
1701 break
1702 elif attr._is_id:
1703 self._id_cache[attr.value] = node
1704 if attr.value == id:
1705 result = node
1706 elif node._magic_id_nodes == 1:
1707 break
1708 elif node._magic_id_nodes:
1709 for attr in node.attributes.values():
1710 if attr._is_id:
1711 self._id_cache[attr.value] = node
1712 if attr.value == id:
1713 result = node
1714 if result is not None:
1715 break
1716 return result
1717
Fred Drake1f549022000-09-24 05:21:58 +00001718 def getElementsByTagName(self, name):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001719 return _get_elements_by_tagName_helper(self, name, NodeList())
Fred Drakefbe7b4f2001-07-04 06:25:53 +00001720
1721 def getElementsByTagNameNS(self, namespaceURI, localName):
Martin v. Löwis787354c2003-01-25 15:28:29 +00001722 return _get_elements_by_tagName_ns_helper(
1723 self, namespaceURI, localName, NodeList())
1724
1725 def isSupported(self, feature, version):
1726 return self.implementation.hasFeature(feature, version)
1727
1728 def importNode(self, node, deep):
1729 if node.nodeType == Node.DOCUMENT_NODE:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001730 raise xml.dom.NotSupportedErr("cannot import document nodes")
Martin v. Löwis787354c2003-01-25 15:28:29 +00001731 elif node.nodeType == Node.DOCUMENT_TYPE_NODE:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001732 raise xml.dom.NotSupportedErr("cannot import document type nodes")
Martin v. Löwis787354c2003-01-25 15:28:29 +00001733 return _clone_node(node, deep, self)
Fred Drake55c38192000-06-29 19:39:57 +00001734
Martin v. Löwis7d650ca2002-06-30 15:05:00 +00001735 def writexml(self, writer, indent="", addindent="", newl="",
1736 encoding = None):
1737 if encoding is None:
Georg Brandl175a7dc2005-08-25 22:02:43 +00001738 writer.write('<?xml version="1.0" ?>'+newl)
Martin v. Löwis7d650ca2002-06-30 15:05:00 +00001739 else:
Georg Brandl175a7dc2005-08-25 22:02:43 +00001740 writer.write('<?xml version="1.0" encoding="%s"?>%s' % (encoding, newl))
Fred Drake55c38192000-06-29 19:39:57 +00001741 for node in self.childNodes:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +00001742 node.writexml(writer, indent, addindent, newl)
Fred Drake55c38192000-06-29 19:39:57 +00001743
Martin v. Löwis787354c2003-01-25 15:28:29 +00001744 # DOM Level 3 (WD 9 April 2002)
1745
1746 def renameNode(self, n, namespaceURI, name):
1747 if n.ownerDocument is not self:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001748 raise xml.dom.WrongDocumentErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001749 "cannot rename nodes from other documents;\n"
1750 "expected %s,\nfound %s" % (self, n.ownerDocument))
1751 if n.nodeType not in (Node.ELEMENT_NODE, Node.ATTRIBUTE_NODE):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001752 raise xml.dom.NotSupportedErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001753 "renameNode() only applies to element and attribute nodes")
1754 if namespaceURI != EMPTY_NAMESPACE:
1755 if ':' in name:
1756 prefix, localName = name.split(':', 1)
1757 if ( prefix == "xmlns"
Fred Drakefbdeaad2006-07-29 16:56:15 +00001758 and namespaceURI != xml.dom.XMLNS_NAMESPACE):
1759 raise xml.dom.NamespaceErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001760 "illegal use of 'xmlns' prefix")
1761 else:
1762 if ( name == "xmlns"
Fred Drakefbdeaad2006-07-29 16:56:15 +00001763 and namespaceURI != xml.dom.XMLNS_NAMESPACE
Martin v. Löwis787354c2003-01-25 15:28:29 +00001764 and n.nodeType == Node.ATTRIBUTE_NODE):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001765 raise xml.dom.NamespaceErr(
Martin v. Löwis787354c2003-01-25 15:28:29 +00001766 "illegal use of the 'xmlns' attribute")
1767 prefix = None
1768 localName = name
1769 else:
1770 prefix = None
1771 localName = None
1772 if n.nodeType == Node.ATTRIBUTE_NODE:
1773 element = n.ownerElement
1774 if element is not None:
1775 is_id = n._is_id
1776 element.removeAttributeNode(n)
1777 else:
1778 element = None
1779 # avoid __setattr__
1780 d = n.__dict__
1781 d['prefix'] = prefix
1782 d['localName'] = localName
1783 d['namespaceURI'] = namespaceURI
1784 d['nodeName'] = name
1785 if n.nodeType == Node.ELEMENT_NODE:
1786 d['tagName'] = name
1787 else:
1788 # attribute node
1789 d['name'] = name
1790 if element is not None:
1791 element.setAttributeNode(n)
1792 if is_id:
1793 element.setIdAttributeNode(n)
1794 # It's not clear from a semantic perspective whether we should
1795 # call the user data handlers for the NODE_RENAMED event since
1796 # we're re-using the existing node. The draft spec has been
1797 # interpreted as meaning "no, don't call the handler unless a
1798 # new node is created."
1799 return n
1800
1801defproperty(Document, "documentElement",
1802 doc="Top-level element of this document.")
1803
1804
1805def _clone_node(node, deep, newOwnerDocument):
1806 """
1807 Clone a node and give it the new owner document.
1808 Called by Node.cloneNode and Document.importNode
1809 """
1810 if node.ownerDocument.isSameNode(newOwnerDocument):
Fred Drakefbdeaad2006-07-29 16:56:15 +00001811 operation = xml.dom.UserDataHandler.NODE_CLONED
Martin v. Löwis787354c2003-01-25 15:28:29 +00001812 else:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001813 operation = xml.dom.UserDataHandler.NODE_IMPORTED
Martin v. Löwis787354c2003-01-25 15:28:29 +00001814 if node.nodeType == Node.ELEMENT_NODE:
1815 clone = newOwnerDocument.createElementNS(node.namespaceURI,
1816 node.nodeName)
1817 for attr in node.attributes.values():
1818 clone.setAttributeNS(attr.namespaceURI, attr.nodeName, attr.value)
1819 a = clone.getAttributeNodeNS(attr.namespaceURI, attr.localName)
1820 a.specified = attr.specified
1821
1822 if deep:
1823 for child in node.childNodes:
1824 c = _clone_node(child, deep, newOwnerDocument)
1825 clone.appendChild(c)
1826
1827 elif node.nodeType == Node.DOCUMENT_FRAGMENT_NODE:
1828 clone = newOwnerDocument.createDocumentFragment()
1829 if deep:
1830 for child in node.childNodes:
1831 c = _clone_node(child, deep, newOwnerDocument)
1832 clone.appendChild(c)
1833
1834 elif node.nodeType == Node.TEXT_NODE:
1835 clone = newOwnerDocument.createTextNode(node.data)
1836 elif node.nodeType == Node.CDATA_SECTION_NODE:
1837 clone = newOwnerDocument.createCDATASection(node.data)
1838 elif node.nodeType == Node.PROCESSING_INSTRUCTION_NODE:
1839 clone = newOwnerDocument.createProcessingInstruction(node.target,
1840 node.data)
1841 elif node.nodeType == Node.COMMENT_NODE:
1842 clone = newOwnerDocument.createComment(node.data)
1843 elif node.nodeType == Node.ATTRIBUTE_NODE:
1844 clone = newOwnerDocument.createAttributeNS(node.namespaceURI,
1845 node.nodeName)
1846 clone.specified = True
1847 clone.value = node.value
1848 elif node.nodeType == Node.DOCUMENT_TYPE_NODE:
1849 assert node.ownerDocument is not newOwnerDocument
Fred Drakefbdeaad2006-07-29 16:56:15 +00001850 operation = xml.dom.UserDataHandler.NODE_IMPORTED
Martin v. Löwis787354c2003-01-25 15:28:29 +00001851 clone = newOwnerDocument.implementation.createDocumentType(
1852 node.name, node.publicId, node.systemId)
1853 clone.ownerDocument = newOwnerDocument
1854 if deep:
1855 clone.entities._seq = []
1856 clone.notations._seq = []
1857 for n in node.notations._seq:
1858 notation = Notation(n.nodeName, n.publicId, n.systemId)
1859 notation.ownerDocument = newOwnerDocument
1860 clone.notations._seq.append(notation)
1861 if hasattr(n, '_call_user_data_handler'):
1862 n._call_user_data_handler(operation, n, notation)
1863 for e in node.entities._seq:
1864 entity = Entity(e.nodeName, e.publicId, e.systemId,
1865 e.notationName)
1866 entity.actualEncoding = e.actualEncoding
1867 entity.encoding = e.encoding
1868 entity.version = e.version
1869 entity.ownerDocument = newOwnerDocument
1870 clone.entities._seq.append(entity)
1871 if hasattr(e, '_call_user_data_handler'):
1872 e._call_user_data_handler(operation, n, entity)
1873 else:
1874 # Note the cloning of Document and DocumentType nodes is
1875 # implemenetation specific. minidom handles those cases
1876 # directly in the cloneNode() methods.
Fred Drakefbdeaad2006-07-29 16:56:15 +00001877 raise xml.dom.NotSupportedErr("Cannot clone node %s" % repr(node))
Martin v. Löwis787354c2003-01-25 15:28:29 +00001878
1879 # Check for _call_user_data_handler() since this could conceivably
1880 # used with other DOM implementations (one of the FourThought
1881 # DOMs, perhaps?).
1882 if hasattr(node, '_call_user_data_handler'):
1883 node._call_user_data_handler(operation, node, clone)
1884 return clone
1885
1886
1887def _nssplit(qualifiedName):
1888 fields = qualifiedName.split(':', 1)
1889 if len(fields) == 2:
1890 return fields
1891 else:
1892 return (None, fields[0])
1893
1894
Fred Drake4ccf4a12000-11-21 22:02:22 +00001895def _get_StringIO():
Fred Drakef7cf40d2000-12-14 18:16:11 +00001896 # we can't use cStringIO since it doesn't support Unicode strings
1897 from StringIO import StringIO
Fred Drake4ccf4a12000-11-21 22:02:22 +00001898 return StringIO()
1899
Martin v. Löwis787354c2003-01-25 15:28:29 +00001900def _do_pulldom_parse(func, args, kwargs):
Raymond Hettingerff41c482003-04-06 09:01:11 +00001901 events = func(*args, **kwargs)
Fred Drake1f549022000-09-24 05:21:58 +00001902 toktype, rootNode = events.getEvent()
1903 events.expandNode(rootNode)
Martin v. Löwisb417be22001-02-06 01:16:06 +00001904 events.clear()
Fred Drake55c38192000-06-29 19:39:57 +00001905 return rootNode
1906
Martin v. Löwis787354c2003-01-25 15:28:29 +00001907def parse(file, parser=None, bufsize=None):
Fred Drakef7cf40d2000-12-14 18:16:11 +00001908 """Parse a file into a DOM by filename or file object."""
Martin v. Löwis787354c2003-01-25 15:28:29 +00001909 if parser is None and not bufsize:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001910 from xml.dom import expatbuilder
Martin v. Löwis787354c2003-01-25 15:28:29 +00001911 return expatbuilder.parse(file)
1912 else:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001913 from xml.dom import pulldom
Raymond Hettingerff41c482003-04-06 09:01:11 +00001914 return _do_pulldom_parse(pulldom.parse, (file,),
Martin v. Löwis787354c2003-01-25 15:28:29 +00001915 {'parser': parser, 'bufsize': bufsize})
Fred Drake55c38192000-06-29 19:39:57 +00001916
Martin v. Löwis787354c2003-01-25 15:28:29 +00001917def parseString(string, parser=None):
Fred Drakef7cf40d2000-12-14 18:16:11 +00001918 """Parse a file into a DOM from a string."""
Martin v. Löwis787354c2003-01-25 15:28:29 +00001919 if parser is None:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001920 from xml.dom import expatbuilder
Martin v. Löwis787354c2003-01-25 15:28:29 +00001921 return expatbuilder.parseString(string)
1922 else:
Fred Drakefbdeaad2006-07-29 16:56:15 +00001923 from xml.dom import pulldom
Martin v. Löwis787354c2003-01-25 15:28:29 +00001924 return _do_pulldom_parse(pulldom.parseString, (string,),
1925 {'parser': parser})
Martin v. Löwis7edbd4f2001-02-22 14:05:50 +00001926
Martin v. Löwis787354c2003-01-25 15:28:29 +00001927def getDOMImplementation(features=None):
1928 if features:
1929 if isinstance(features, StringTypes):
1930 features = domreg._parse_feature_string(features)
1931 for f, v in features:
1932 if not Document.implementation.hasFeature(f, v):
1933 return None
Martin v. Löwis7edbd4f2001-02-22 14:05:50 +00001934 return Document.implementation