blob: 085f0eb7b10a5d05b6eb6c19a30be016157b3cf7 [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 Drake1f549022000-09-24 05:21:58 +000017import string
Fred Drake4ccf4a12000-11-21 22:02:22 +000018_string = string
19del string
20
Fred Drake49a5d032001-11-30 22:21:58 +000021from xml.dom import HierarchyRequestErr, EMPTY_NAMESPACE
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +000022
Fred Drake4ccf4a12000-11-21 22:02:22 +000023# localize the types, and allow support for Unicode values if available:
Fred Drake1f549022000-09-24 05:21:58 +000024import types
Fred Drake4ccf4a12000-11-21 22:02:22 +000025_TupleType = types.TupleType
26try:
27 _StringTypes = (types.StringType, types.UnicodeType)
28except AttributeError:
29 _StringTypes = (types.StringType,)
30del types
31
Fred Drakef7cf40d2000-12-14 18:16:11 +000032import xml.dom
Fred Drake55c38192000-06-29 19:39:57 +000033
Fred Drake3ac6a092001-09-28 04:33:06 +000034
35if list is type([]):
36 class NodeList(list):
37 def item(self, index):
38 if 0 <= index < len(self):
39 return self[index]
40
Fred Drake575712e2001-09-28 20:25:45 +000041 length = property(lambda self: len(self),
42 doc="The number of nodes in the NodeList.")
Fred Drake3ac6a092001-09-28 04:33:06 +000043
44else:
45 def NodeList():
46 return []
Martin v. Löwis95700f72002-03-15 13:51:59 +000047
Fred Drake3ac6a092001-09-28 04:33:06 +000048
Fred Drake575712e2001-09-28 20:25:45 +000049class Node(xml.dom.Node):
Fred Drake1f549022000-09-24 05:21:58 +000050 allnodes = {}
51 _debug = 0
52 _makeParentNodes = 1
53 debug = None
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +000054 childNodeTypes = ()
Martin v. Löwis126f2f62001-03-13 10:50:13 +000055 namespaceURI = None # this is non-null only for elements and attributes
Fred Drake575712e2001-09-28 20:25:45 +000056 parentNode = None
57 ownerDocument = None
Martin v. Löwis52ce0d02001-01-27 08:47:37 +000058
Fred Drake1f549022000-09-24 05:21:58 +000059 def __init__(self):
Fred Drake3ac6a092001-09-28 04:33:06 +000060 self.childNodes = NodeList()
Fred Drake16f63292000-10-23 18:09:50 +000061 if Node._debug:
Fred Drake1f549022000-09-24 05:21:58 +000062 index = repr(id(self)) + repr(self.__class__)
63 Node.allnodes[index] = repr(self.__dict__)
64 if Node.debug is None:
Fred Drake4ccf4a12000-11-21 22:02:22 +000065 Node.debug = _get_StringIO()
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +000066 #open("debug4.out", "w")
Fred Drake1f549022000-09-24 05:21:58 +000067 Node.debug.write("create %s\n" % index)
Fred Drake55c38192000-06-29 19:39:57 +000068
Fred Drake1f549022000-09-24 05:21:58 +000069 def __nonzero__(self):
70 return 1
Fred Drake55c38192000-06-29 19:39:57 +000071
Fred Drake1f549022000-09-24 05:21:58 +000072 def toxml(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +000073 writer = _get_StringIO()
Fred Drake1f549022000-09-24 05:21:58 +000074 self.writexml(writer)
Fred Drake55c38192000-06-29 19:39:57 +000075 return writer.getvalue()
76
Martin v. Löwis46fa39a2001-02-06 00:14:08 +000077 def toprettyxml(self, indent="\t", newl="\n"):
Martin v. Löwiscb67ea12001-03-31 16:30:40 +000078 # indent = the indentation string to prepend, per level
79 # newl = the newline string to append
80 writer = _get_StringIO()
81 self.writexml(writer, "", indent, newl)
82 return writer.getvalue()
Martin v. Löwis46fa39a2001-02-06 00:14:08 +000083
Fred Drake1f549022000-09-24 05:21:58 +000084 def hasChildNodes(self):
85 if self.childNodes:
86 return 1
87 else:
88 return 0
Fred Drake55c38192000-06-29 19:39:57 +000089
Fred Drake1f549022000-09-24 05:21:58 +000090 def _get_firstChild(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +000091 if self.childNodes:
92 return self.childNodes[0]
Paul Prescod73678da2000-07-01 04:58:47 +000093
Fred Drake1f549022000-09-24 05:21:58 +000094 def _get_lastChild(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +000095 if self.childNodes:
96 return self.childNodes[-1]
Paul Prescod73678da2000-07-01 04:58:47 +000097
Fred Draked1572372001-09-29 04:58:32 +000098 try:
99 property
100 except NameError:
101 def __getattr__(self, key):
102 if key[0:2] == "__":
103 raise AttributeError, key
104 # getattr should never call getattr!
105 if self.__dict__.has_key("inGetAttr"):
106 del self.inGetAttr
107 raise AttributeError, key
108
109 prefix, attrname = key[:5], key[5:]
110 if prefix == "_get_":
111 self.inGetAttr = 1
112 if hasattr(self, attrname):
113 del self.inGetAttr
114 return (lambda self=self, attrname=attrname:
115 getattr(self, attrname))
116 else:
117 del self.inGetAttr
118 raise AttributeError, key
119 else:
120 self.inGetAttr = 1
121 try:
122 func = getattr(self, "_get_" + key)
123 except AttributeError:
124 raise AttributeError, key
125 del self.inGetAttr
126 return func()
127 else:
128 firstChild = property(_get_firstChild,
129 doc="First child node, or None.")
130 lastChild = property(_get_lastChild,
131 doc="Last child node, or None.")
132
Fred Drake1f549022000-09-24 05:21:58 +0000133 def insertBefore(self, newChild, refChild):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000134 if newChild.nodeType == self.DOCUMENT_FRAGMENT_NODE:
Fred Drakee50959a2001-12-06 04:32:18 +0000135 for c in tuple(newChild.childNodes):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000136 self.insertBefore(c, refChild)
137 ### The DOM does not clearly specify what to return in this case
138 return newChild
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000139 if newChild.nodeType not in self.childNodeTypes:
140 raise HierarchyRequestErr, \
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000141 "%s cannot be child of %s" % (repr(newChild), repr(self))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000142 if newChild.parentNode is not None:
143 newChild.parentNode.removeChild(newChild)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000144 if refChild is None:
145 self.appendChild(newChild)
146 else:
147 index = self.childNodes.index(refChild)
148 self.childNodes.insert(index, newChild)
149 newChild.nextSibling = refChild
150 refChild.previousSibling = newChild
151 if index:
152 node = self.childNodes[index-1]
153 node.nextSibling = newChild
154 newChild.previousSibling = node
155 else:
156 newChild.previousSibling = None
157 if self._makeParentNodes:
158 newChild.parentNode = self
159 return newChild
Fred Drake55c38192000-06-29 19:39:57 +0000160
Fred Drake1f549022000-09-24 05:21:58 +0000161 def appendChild(self, node):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000162 if node.nodeType == self.DOCUMENT_FRAGMENT_NODE:
Fred Drakee50959a2001-12-06 04:32:18 +0000163 for c in tuple(node.childNodes):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000164 self.appendChild(c)
165 ### The DOM does not clearly specify what to return in this case
166 return node
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000167 if node.nodeType not in self.childNodeTypes:
168 raise HierarchyRequestErr, \
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000169 "%s cannot be child of %s" % (repr(node), repr(self))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000170 if node.parentNode is not None:
171 node.parentNode.removeChild(node)
Fred Drake13a30692000-10-09 20:04:16 +0000172 if self.childNodes:
173 last = self.lastChild
174 node.previousSibling = last
175 last.nextSibling = node
176 else:
177 node.previousSibling = None
178 node.nextSibling = None
Fred Drake1f549022000-09-24 05:21:58 +0000179 self.childNodes.append(node)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000180 if self._makeParentNodes:
181 node.parentNode = self
Paul Prescod73678da2000-07-01 04:58:47 +0000182 return node
183
Fred Drake1f549022000-09-24 05:21:58 +0000184 def replaceChild(self, newChild, oldChild):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000185 if newChild.nodeType == self.DOCUMENT_FRAGMENT_NODE:
186 refChild = oldChild.nextSibling
187 self.removeChild(oldChild)
188 return self.insertBefore(newChild, refChild)
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000189 if newChild.nodeType not in self.childNodeTypes:
190 raise HierarchyRequestErr, \
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000191 "%s cannot be child of %s" % (repr(newChild), repr(self))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000192 if newChild.parentNode is not None:
193 newChild.parentNode.removeChild(newChild)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000194 if newChild is oldChild:
195 return
Fred Drake1f549022000-09-24 05:21:58 +0000196 index = self.childNodes.index(oldChild)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000197 self.childNodes[index] = newChild
198 if self._makeParentNodes:
199 newChild.parentNode = self
200 oldChild.parentNode = None
201 newChild.nextSibling = oldChild.nextSibling
202 newChild.previousSibling = oldChild.previousSibling
Martin v. Löwis156c3372000-12-28 18:40:56 +0000203 oldChild.nextSibling = None
Fred Drake4ccf4a12000-11-21 22:02:22 +0000204 oldChild.previousSibling = None
Martin v. Löwis156c3372000-12-28 18:40:56 +0000205 if newChild.previousSibling:
206 newChild.previousSibling.nextSibling = newChild
207 if newChild.nextSibling:
208 newChild.nextSibling.previousSibling = newChild
Fred Drake4ccf4a12000-11-21 22:02:22 +0000209 return oldChild
Paul Prescod73678da2000-07-01 04:58:47 +0000210
Fred Drake1f549022000-09-24 05:21:58 +0000211 def removeChild(self, oldChild):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000212 self.childNodes.remove(oldChild)
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000213 if oldChild.nextSibling is not None:
214 oldChild.nextSibling.previousSibling = oldChild.previousSibling
215 if oldChild.previousSibling is not None:
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000216 oldChild.previousSibling.nextSibling = oldChild.nextSibling
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000217 oldChild.nextSibling = oldChild.previousSibling = None
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000218
Fred Drake4ccf4a12000-11-21 22:02:22 +0000219 if self._makeParentNodes:
220 oldChild.parentNode = None
221 return oldChild
222
223 def normalize(self):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000224 L = []
225 for child in self.childNodes:
226 if child.nodeType == Node.TEXT_NODE:
227 data = child.data
228 if data and L and L[-1].nodeType == child.nodeType:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000229 # collapse text node
230 node = L[-1]
231 node.data = node.nodeValue = node.data + child.data
232 node.nextSibling = child.nextSibling
233 child.unlink()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000234 elif data:
235 if L:
236 L[-1].nextSibling = child
237 child.previousSibling = L[-1]
238 else:
239 child.previousSibling = None
240 L.append(child)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000241 else:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000242 # empty text node; discard
243 child.unlink()
244 else:
245 if L:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000246 L[-1].nextSibling = child
247 child.previousSibling = L[-1]
Fred Drakef7cf40d2000-12-14 18:16:11 +0000248 else:
249 child.previousSibling = None
250 L.append(child)
251 if child.nodeType == Node.ELEMENT_NODE:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000252 child.normalize()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000253 self.childNodes[:] = L
Paul Prescod73678da2000-07-01 04:58:47 +0000254
Fred Drake1f549022000-09-24 05:21:58 +0000255 def cloneNode(self, deep):
Paul Prescod73678da2000-07-01 04:58:47 +0000256 import new
Fred Drake4ccf4a12000-11-21 22:02:22 +0000257 clone = new.instance(self.__class__, self.__dict__.copy())
258 if self._makeParentNodes:
259 clone.parentNode = None
Fred Drake3ac6a092001-09-28 04:33:06 +0000260 clone.childNodes = NodeList()
Fred Drake4ccf4a12000-11-21 22:02:22 +0000261 if deep:
262 for child in self.childNodes:
263 clone.appendChild(child.cloneNode(1))
Paul Prescod73678da2000-07-01 04:58:47 +0000264 return clone
Fred Drake55c38192000-06-29 19:39:57 +0000265
Fred Drake25239772001-02-02 19:40:19 +0000266 # DOM Level 3 (Working Draft 2001-Jan-26)
267
268 def isSameNode(self, other):
269 return self is other
270
271 # minidom-specific API:
272
Fred Drake1f549022000-09-24 05:21:58 +0000273 def unlink(self):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000274 self.parentNode = self.ownerDocument = None
Fred Drake4ccf4a12000-11-21 22:02:22 +0000275 for child in self.childNodes:
276 child.unlink()
Fred Drake1f549022000-09-24 05:21:58 +0000277 self.childNodes = None
Paul Prescod4221ff02000-10-13 20:11:42 +0000278 self.previousSibling = None
279 self.nextSibling = None
Paul Prescod73678da2000-07-01 04:58:47 +0000280 if Node._debug:
Fred Drake1f549022000-09-24 05:21:58 +0000281 index = repr(id(self)) + repr(self.__class__)
282 self.debug.write("Deleting: %s\n" % index)
Paul Prescod73678da2000-07-01 04:58:47 +0000283 del Node.allnodes[index]
Fred Drake55c38192000-06-29 19:39:57 +0000284
Fred Drake1f549022000-09-24 05:21:58 +0000285def _write_data(writer, data):
Fred Drake55c38192000-06-29 19:39:57 +0000286 "Writes datachars to writer."
Fred Drake4ccf4a12000-11-21 22:02:22 +0000287 replace = _string.replace
288 data = replace(data, "&", "&amp;")
289 data = replace(data, "<", "&lt;")
290 data = replace(data, "\"", "&quot;")
291 data = replace(data, ">", "&gt;")
Fred Drake55c38192000-06-29 19:39:57 +0000292 writer.write(data)
293
Fred Drake1f549022000-09-24 05:21:58 +0000294def _getElementsByTagNameHelper(parent, name, rc):
Fred Drake55c38192000-06-29 19:39:57 +0000295 for node in parent.childNodes:
Fred Drake1f549022000-09-24 05:21:58 +0000296 if node.nodeType == Node.ELEMENT_NODE and \
297 (name == "*" or node.tagName == name):
298 rc.append(node)
299 _getElementsByTagNameHelper(node, name, rc)
Fred Drake55c38192000-06-29 19:39:57 +0000300 return rc
301
Fred Drake1f549022000-09-24 05:21:58 +0000302def _getElementsByTagNameNSHelper(parent, nsURI, localName, rc):
Fred Drake55c38192000-06-29 19:39:57 +0000303 for node in parent.childNodes:
Fred Drake1f549022000-09-24 05:21:58 +0000304 if node.nodeType == Node.ELEMENT_NODE:
Martin v. Löwised525fb2001-06-03 14:06:42 +0000305 if ((localName == "*" or node.localName == localName) and
Fred Drake1f549022000-09-24 05:21:58 +0000306 (nsURI == "*" or node.namespaceURI == nsURI)):
307 rc.append(node)
Fred Drakef7cf40d2000-12-14 18:16:11 +0000308 _getElementsByTagNameNSHelper(node, nsURI, localName, rc)
309 return rc
Fred Drake55c38192000-06-29 19:39:57 +0000310
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000311class DocumentFragment(Node):
312 nodeType = Node.DOCUMENT_FRAGMENT_NODE
313 nodeName = "#document-fragment"
314 nodeValue = None
315 attributes = None
316 parentNode = None
317 childNodeTypes = (Node.ELEMENT_NODE,
318 Node.TEXT_NODE,
319 Node.CDATA_SECTION_NODE,
320 Node.ENTITY_REFERENCE_NODE,
321 Node.PROCESSING_INSTRUCTION_NODE,
322 Node.COMMENT_NODE,
323 Node.NOTATION_NODE)
324
325
Fred Drake55c38192000-06-29 19:39:57 +0000326class Attr(Node):
Fred Drake1f549022000-09-24 05:21:58 +0000327 nodeType = Node.ATTRIBUTE_NODE
Fred Drake4ccf4a12000-11-21 22:02:22 +0000328 attributes = None
329 ownerElement = None
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000330 childNodeTypes = (Node.TEXT_NODE, Node.ENTITY_REFERENCE_NODE)
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000331
Fred Drake49a5d032001-11-30 22:21:58 +0000332 def __init__(self, qName, namespaceURI=EMPTY_NAMESPACE, localName=None, prefix=None):
Fred Drake55c38192000-06-29 19:39:57 +0000333 # skip setattr for performance
Fred Drake4ccf4a12000-11-21 22:02:22 +0000334 d = self.__dict__
335 d["localName"] = localName or qName
336 d["nodeName"] = d["name"] = qName
337 d["namespaceURI"] = namespaceURI
338 d["prefix"] = prefix
Fred Drake1f549022000-09-24 05:21:58 +0000339 Node.__init__(self)
Paul Prescod73678da2000-07-01 04:58:47 +0000340 # nodeValue and value are set elsewhere
Fred Drake55c38192000-06-29 19:39:57 +0000341
Fred Drake1f549022000-09-24 05:21:58 +0000342 def __setattr__(self, name, value):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000343 d = self.__dict__
Fred Drake1f549022000-09-24 05:21:58 +0000344 if name in ("value", "nodeValue"):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000345 d["value"] = d["nodeValue"] = value
346 elif name in ("name", "nodeName"):
347 d["name"] = d["nodeName"] = value
Fred Drake55c38192000-06-29 19:39:57 +0000348 else:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000349 d[name] = value
Fred Drake55c38192000-06-29 19:39:57 +0000350
Fred Drake4ccf4a12000-11-21 22:02:22 +0000351 def cloneNode(self, deep):
352 clone = Node.cloneNode(self, deep)
353 if clone.__dict__.has_key("ownerElement"):
354 del clone.ownerElement
355 return clone
356
Fred Drakef7cf40d2000-12-14 18:16:11 +0000357
358class NamedNodeMap:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000359 """The attribute list is a transient interface to the underlying
360 dictionaries. Mutations here will change the underlying element's
Fred Drakef7cf40d2000-12-14 18:16:11 +0000361 dictionary.
362
363 Ordering is imposed artificially and does not reflect the order of
364 attributes as found in an input document.
365 """
Fred Drake4ccf4a12000-11-21 22:02:22 +0000366
Fred Drake2998a552001-12-06 18:27:48 +0000367 def __init__(self, attrs, attrsNS, ownerElement):
Fred Drake1f549022000-09-24 05:21:58 +0000368 self._attrs = attrs
369 self._attrsNS = attrsNS
Fred Drake2998a552001-12-06 18:27:48 +0000370 self._ownerElement = ownerElement
Fred Drakef7cf40d2000-12-14 18:16:11 +0000371
Fred Draked1572372001-09-29 04:58:32 +0000372 try:
373 property
374 except NameError:
375 def __getattr__(self, name):
376 if name == "length":
377 return len(self._attrs)
378 raise AttributeError, name
379 else:
380 length = property(lambda self: len(self._attrs),
381 doc="Number of nodes in the NamedNodeMap.")
Fred Drake55c38192000-06-29 19:39:57 +0000382
Fred Drake1f549022000-09-24 05:21:58 +0000383 def item(self, index):
Fred Drake55c38192000-06-29 19:39:57 +0000384 try:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000385 return self[self._attrs.keys()[index]]
Fred Drake55c38192000-06-29 19:39:57 +0000386 except IndexError:
387 return None
Fred Drake55c38192000-06-29 19:39:57 +0000388
Fred Drake1f549022000-09-24 05:21:58 +0000389 def items(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000390 L = []
391 for node in self._attrs.values():
Martin v. Löwisd5fb58f2001-01-27 08:38:34 +0000392 L.append((node.nodeName, node.value))
Fred Drake4ccf4a12000-11-21 22:02:22 +0000393 return L
Fred Drake1f549022000-09-24 05:21:58 +0000394
395 def itemsNS(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000396 L = []
397 for node in self._attrs.values():
Fred Drake49a5d032001-11-30 22:21:58 +0000398 L.append(((node.namespaceURI, node.localName), node.value))
Fred Drake4ccf4a12000-11-21 22:02:22 +0000399 return L
Fred Drake16f63292000-10-23 18:09:50 +0000400
Fred Drake1f549022000-09-24 05:21:58 +0000401 def keys(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000402 return self._attrs.keys()
Fred Drake55c38192000-06-29 19:39:57 +0000403
Fred Drake1f549022000-09-24 05:21:58 +0000404 def keysNS(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000405 return self._attrsNS.keys()
Fred Drake55c38192000-06-29 19:39:57 +0000406
Fred Drake1f549022000-09-24 05:21:58 +0000407 def values(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000408 return self._attrs.values()
Fred Drake55c38192000-06-29 19:39:57 +0000409
Martin v. Löwisd5fb58f2001-01-27 08:38:34 +0000410 def get(self, name, value = None):
411 return self._attrs.get(name, value)
412
Fred Drake1f549022000-09-24 05:21:58 +0000413 def __len__(self):
Fred Drake55c38192000-06-29 19:39:57 +0000414 return self.length
415
Fred Drake1f549022000-09-24 05:21:58 +0000416 def __cmp__(self, other):
417 if self._attrs is getattr(other, "_attrs", None):
Fred Drake55c38192000-06-29 19:39:57 +0000418 return 0
Fred Drake16f63292000-10-23 18:09:50 +0000419 else:
Fred Drake1f549022000-09-24 05:21:58 +0000420 return cmp(id(self), id(other))
Fred Drake55c38192000-06-29 19:39:57 +0000421
422 #FIXME: is it appropriate to return .value?
Fred Drake1f549022000-09-24 05:21:58 +0000423 def __getitem__(self, attname_or_tuple):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000424 if type(attname_or_tuple) is _TupleType:
Paul Prescod73678da2000-07-01 04:58:47 +0000425 return self._attrsNS[attname_or_tuple]
Fred Drake55c38192000-06-29 19:39:57 +0000426 else:
Paul Prescod73678da2000-07-01 04:58:47 +0000427 return self._attrs[attname_or_tuple]
Fred Drake55c38192000-06-29 19:39:57 +0000428
Paul Prescod1e688272000-07-01 19:21:47 +0000429 # same as set
Fred Drake1f549022000-09-24 05:21:58 +0000430 def __setitem__(self, attname, value):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000431 if type(value) in _StringTypes:
Fred Drake1f549022000-09-24 05:21:58 +0000432 node = Attr(attname)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000433 node.value = value
Fred Drake2998a552001-12-06 18:27:48 +0000434 node.ownerDocument = self._ownerElement.ownerDocument
Paul Prescod1e688272000-07-01 19:21:47 +0000435 else:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000436 if not isinstance(value, Attr):
437 raise TypeError, "value must be a string or Attr object"
Fred Drake1f549022000-09-24 05:21:58 +0000438 node = value
Fred Drakef7cf40d2000-12-14 18:16:11 +0000439 self.setNamedItem(node)
440
441 def setNamedItem(self, node):
Andrew M. Kuchlingbc8f72c2001-02-21 01:30:26 +0000442 if not isinstance(node, Attr):
443 raise HierarchyRequestErr, \
444 "%s cannot be child of %s" % (repr(node), repr(self))
Fred Drakef7cf40d2000-12-14 18:16:11 +0000445 old = self._attrs.get(node.name)
Paul Prescod1e688272000-07-01 19:21:47 +0000446 if old:
447 old.unlink()
Fred Drake1f549022000-09-24 05:21:58 +0000448 self._attrs[node.name] = node
449 self._attrsNS[(node.namespaceURI, node.localName)] = node
Fred Drake2998a552001-12-06 18:27:48 +0000450 node.ownerElement = self._ownerElement
Fred Drakef7cf40d2000-12-14 18:16:11 +0000451 return old
452
453 def setNamedItemNS(self, node):
454 return self.setNamedItem(node)
Paul Prescod73678da2000-07-01 04:58:47 +0000455
Fred Drake1f549022000-09-24 05:21:58 +0000456 def __delitem__(self, attname_or_tuple):
457 node = self[attname_or_tuple]
Paul Prescod73678da2000-07-01 04:58:47 +0000458 node.unlink()
459 del self._attrs[node.name]
460 del self._attrsNS[(node.namespaceURI, node.localName)]
Fred Drakef7cf40d2000-12-14 18:16:11 +0000461 self.length = len(self._attrs)
462
463AttributeList = NamedNodeMap
464
Fred Drake1f549022000-09-24 05:21:58 +0000465
Martin v. Löwisa2fda0d2000-10-07 12:10:28 +0000466class Element(Node):
Fred Drake1f549022000-09-24 05:21:58 +0000467 nodeType = Node.ELEMENT_NODE
Fred Drake4ccf4a12000-11-21 22:02:22 +0000468 nextSibling = None
469 previousSibling = None
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000470 childNodeTypes = (Node.ELEMENT_NODE, Node.PROCESSING_INSTRUCTION_NODE,
471 Node.COMMENT_NODE, Node.TEXT_NODE,
472 Node.CDATA_SECTION_NODE, Node.ENTITY_REFERENCE_NODE)
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000473
Fred Drake49a5d032001-11-30 22:21:58 +0000474 def __init__(self, tagName, namespaceURI=EMPTY_NAMESPACE, prefix=None,
Fred Drake1f549022000-09-24 05:21:58 +0000475 localName=None):
476 Node.__init__(self)
Fred Drake55c38192000-06-29 19:39:57 +0000477 self.tagName = self.nodeName = tagName
Fred Drake1f549022000-09-24 05:21:58 +0000478 self.localName = localName or tagName
479 self.prefix = prefix
480 self.namespaceURI = namespaceURI
481 self.nodeValue = None
Fred Drake55c38192000-06-29 19:39:57 +0000482
Fred Drake4ccf4a12000-11-21 22:02:22 +0000483 self._attrs = {} # attributes are double-indexed:
484 self._attrsNS = {} # tagName -> Attribute
485 # URI,localName -> Attribute
486 # in the future: consider lazy generation
487 # of attribute objects this is too tricky
488 # for now because of headaches with
489 # namespaces.
490
491 def cloneNode(self, deep):
492 clone = Node.cloneNode(self, deep)
493 clone._attrs = {}
494 clone._attrsNS = {}
495 for attr in self._attrs.values():
496 node = attr.cloneNode(1)
497 clone._attrs[node.name] = node
498 clone._attrsNS[(node.namespaceURI, node.localName)] = node
499 node.ownerElement = clone
500 return clone
501
502 def unlink(self):
503 for attr in self._attrs.values():
504 attr.unlink()
505 self._attrs = None
506 self._attrsNS = None
507 Node.unlink(self)
Fred Drake55c38192000-06-29 19:39:57 +0000508
Fred Drake1f549022000-09-24 05:21:58 +0000509 def getAttribute(self, attname):
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000510 try:
511 return self._attrs[attname].value
512 except KeyError:
513 return ""
Fred Drake55c38192000-06-29 19:39:57 +0000514
Fred Drake1f549022000-09-24 05:21:58 +0000515 def getAttributeNS(self, namespaceURI, localName):
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000516 try:
517 return self._attrsNS[(namespaceURI, localName)].value
518 except KeyError:
519 return ""
Fred Drake1f549022000-09-24 05:21:58 +0000520
521 def setAttribute(self, attname, value):
522 attr = Attr(attname)
Fred Drake55c38192000-06-29 19:39:57 +0000523 # for performance
Fred Drake2998a552001-12-06 18:27:48 +0000524 d = attr.__dict__
525 d["value"] = d["nodeValue"] = value
526 d["ownerDocument"] = self.ownerDocument
Fred Drake1f549022000-09-24 05:21:58 +0000527 self.setAttributeNode(attr)
Fred Drake55c38192000-06-29 19:39:57 +0000528
Fred Drake1f549022000-09-24 05:21:58 +0000529 def setAttributeNS(self, namespaceURI, qualifiedName, value):
530 prefix, localname = _nssplit(qualifiedName)
Fred Drake55c38192000-06-29 19:39:57 +0000531 # for performance
Fred Drake1f549022000-09-24 05:21:58 +0000532 attr = Attr(qualifiedName, namespaceURI, localname, prefix)
Fred Drake2998a552001-12-06 18:27:48 +0000533 d = attr.__dict__
534 d["value"] = d["nodeValue"] = value
535 d["ownerDocument"] = self.ownerDocument
Fred Drake1f549022000-09-24 05:21:58 +0000536 self.setAttributeNode(attr)
Fred Drake55c38192000-06-29 19:39:57 +0000537
Fred Drake1f549022000-09-24 05:21:58 +0000538 def getAttributeNode(self, attrname):
539 return self._attrs.get(attrname)
Paul Prescod73678da2000-07-01 04:58:47 +0000540
Fred Drake1f549022000-09-24 05:21:58 +0000541 def getAttributeNodeNS(self, namespaceURI, localName):
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000542 return self._attrsNS.get((namespaceURI, localName))
Paul Prescod73678da2000-07-01 04:58:47 +0000543
Fred Drake1f549022000-09-24 05:21:58 +0000544 def setAttributeNode(self, attr):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000545 if attr.ownerElement not in (None, self):
Martin v. Löwisd5fb58f2001-01-27 08:38:34 +0000546 raise xml.dom.InuseAttributeErr("attribute node already owned")
Fred Drake1f549022000-09-24 05:21:58 +0000547 old = self._attrs.get(attr.name, None)
Paul Prescod73678da2000-07-01 04:58:47 +0000548 if old:
549 old.unlink()
Fred Drake1f549022000-09-24 05:21:58 +0000550 self._attrs[attr.name] = attr
551 self._attrsNS[(attr.namespaceURI, attr.localName)] = attr
Fred Drake4ccf4a12000-11-21 22:02:22 +0000552
553 # This creates a circular reference, but Element.unlink()
554 # breaks the cycle since the references to the attribute
555 # dictionaries are tossed.
556 attr.ownerElement = self
557
558 if old is not attr:
559 # It might have already been part of this node, in which case
560 # it doesn't represent a change, and should not be returned.
561 return old
Fred Drake55c38192000-06-29 19:39:57 +0000562
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000563 setAttributeNodeNS = setAttributeNode
564
Fred Drake1f549022000-09-24 05:21:58 +0000565 def removeAttribute(self, name):
Paul Prescod73678da2000-07-01 04:58:47 +0000566 attr = self._attrs[name]
Fred Drake1f549022000-09-24 05:21:58 +0000567 self.removeAttributeNode(attr)
Fred Drake55c38192000-06-29 19:39:57 +0000568
Fred Drake1f549022000-09-24 05:21:58 +0000569 def removeAttributeNS(self, namespaceURI, localName):
Paul Prescod73678da2000-07-01 04:58:47 +0000570 attr = self._attrsNS[(namespaceURI, localName)]
Fred Drake1f549022000-09-24 05:21:58 +0000571 self.removeAttributeNode(attr)
Fred Drake55c38192000-06-29 19:39:57 +0000572
Fred Drake1f549022000-09-24 05:21:58 +0000573 def removeAttributeNode(self, node):
Paul Prescod73678da2000-07-01 04:58:47 +0000574 node.unlink()
575 del self._attrs[node.name]
576 del self._attrsNS[(node.namespaceURI, node.localName)]
Fred Drake16f63292000-10-23 18:09:50 +0000577
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000578 removeAttributeNodeNS = removeAttributeNode
579
Martin v. Löwis156c3372000-12-28 18:40:56 +0000580 def hasAttribute(self, name):
581 return self._attrs.has_key(name)
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000582
Martin v. Löwis156c3372000-12-28 18:40:56 +0000583 def hasAttributeNS(self, namespaceURI, localName):
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000584 return self._attrsNS.has_key((namespaceURI, localName))
585
Fred Drake1f549022000-09-24 05:21:58 +0000586 def getElementsByTagName(self, name):
Martin v. Löwis95700f72002-03-15 13:51:59 +0000587 return _getElementsByTagNameHelper(self, name, NodeList())
Fred Drake55c38192000-06-29 19:39:57 +0000588
Fred Drake1f549022000-09-24 05:21:58 +0000589 def getElementsByTagNameNS(self, namespaceURI, localName):
Fred Drake15b68932002-03-15 14:37:23 +0000590 return _getElementsByTagNameNSHelper(self, namespaceURI, localName,
591 NodeList())
Fred Drake55c38192000-06-29 19:39:57 +0000592
Fred Drake1f549022000-09-24 05:21:58 +0000593 def __repr__(self):
594 return "<DOM Element: %s at %s>" % (self.tagName, id(self))
Fred Drake55c38192000-06-29 19:39:57 +0000595
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000596 def writexml(self, writer, indent="", addindent="", newl=""):
597 # indent = current indentation
598 # addindent = indentation to add to higher levels
599 # newl = newline string
600 writer.write(indent+"<" + self.tagName)
Fred Drake16f63292000-10-23 18:09:50 +0000601
Fred Drake4ccf4a12000-11-21 22:02:22 +0000602 attrs = self._get_attributes()
603 a_names = attrs.keys()
Fred Drake55c38192000-06-29 19:39:57 +0000604 a_names.sort()
605
606 for a_name in a_names:
Fred Drake1f549022000-09-24 05:21:58 +0000607 writer.write(" %s=\"" % a_name)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000608 _write_data(writer, attrs[a_name].value)
Fred Drake55c38192000-06-29 19:39:57 +0000609 writer.write("\"")
610 if self.childNodes:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000611 writer.write(">%s"%(newl))
Fred Drake55c38192000-06-29 19:39:57 +0000612 for node in self.childNodes:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000613 node.writexml(writer,indent+addindent,addindent,newl)
614 writer.write("%s</%s>%s" % (indent,self.tagName,newl))
Fred Drake55c38192000-06-29 19:39:57 +0000615 else:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000616 writer.write("/>%s"%(newl))
Fred Drake55c38192000-06-29 19:39:57 +0000617
Fred Drake1f549022000-09-24 05:21:58 +0000618 def _get_attributes(self):
Fred Drake2998a552001-12-06 18:27:48 +0000619 return NamedNodeMap(self._attrs, self._attrsNS, self)
Fred Drake55c38192000-06-29 19:39:57 +0000620
Fred Draked1572372001-09-29 04:58:32 +0000621 try:
622 property
623 except NameError:
624 pass
625 else:
626 attributes = property(_get_attributes,
627 doc="NamedNodeMap of attributes on the element.")
628
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000629 def hasAttributes(self):
630 if self._attrs or self._attrsNS:
631 return 1
632 else:
633 return 0
634
Fred Drake1f549022000-09-24 05:21:58 +0000635class Comment(Node):
636 nodeType = Node.COMMENT_NODE
Fred Drake4ccf4a12000-11-21 22:02:22 +0000637 nodeName = "#comment"
638 attributes = None
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000639 childNodeTypes = ()
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000640
Fred Drake1f549022000-09-24 05:21:58 +0000641 def __init__(self, data):
642 Node.__init__(self)
643 self.data = self.nodeValue = data
Fred Drake55c38192000-06-29 19:39:57 +0000644
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000645 def writexml(self, writer, indent="", addindent="", newl=""):
646 writer.write("%s<!--%s-->%s" % (indent,self.data,newl))
Fred Drake1f549022000-09-24 05:21:58 +0000647
648class ProcessingInstruction(Node):
649 nodeType = Node.PROCESSING_INSTRUCTION_NODE
Fred Drake4ccf4a12000-11-21 22:02:22 +0000650 attributes = None
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000651 childNodeTypes = ()
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000652
Fred Drake1f549022000-09-24 05:21:58 +0000653 def __init__(self, target, data):
654 Node.__init__(self)
Fred Drake55c38192000-06-29 19:39:57 +0000655 self.target = self.nodeName = target
656 self.data = self.nodeValue = data
Fred Drake55c38192000-06-29 19:39:57 +0000657
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000658 def writexml(self, writer, indent="", addindent="", newl=""):
659 writer.write("%s<?%s %s?>%s" % (indent,self.target, self.data, newl))
Fred Drake55c38192000-06-29 19:39:57 +0000660
Fred Drake87432f42001-04-04 14:09:46 +0000661class CharacterData(Node):
Fred Drake1f549022000-09-24 05:21:58 +0000662 def __init__(self, data):
Fred Drakedaa823a2001-01-08 04:04:34 +0000663 if type(data) not in _StringTypes:
664 raise TypeError, "node contents must be a string"
Fred Drake1f549022000-09-24 05:21:58 +0000665 Node.__init__(self)
Fred Drake55c38192000-06-29 19:39:57 +0000666 self.data = self.nodeValue = data
Fred Drake33d2b842001-04-04 15:15:18 +0000667 self.length = len(data)
Fred Drake87432f42001-04-04 14:09:46 +0000668
Fred Drake55c38192000-06-29 19:39:57 +0000669 def __repr__(self):
Fred Drake1f549022000-09-24 05:21:58 +0000670 if len(self.data) > 10:
671 dotdotdot = "..."
Fred Drake55c38192000-06-29 19:39:57 +0000672 else:
Fred Drake1f549022000-09-24 05:21:58 +0000673 dotdotdot = ""
Fred Drake87432f42001-04-04 14:09:46 +0000674 return "<DOM %s node \"%s%s\">" % (
675 self.__class__.__name__, self.data[0:10], dotdotdot)
676
677 def substringData(self, offset, count):
678 if offset < 0:
679 raise xml.dom.IndexSizeErr("offset cannot be negative")
680 if offset >= len(self.data):
681 raise xml.dom.IndexSizeErr("offset cannot be beyond end of data")
682 if count < 0:
683 raise xml.dom.IndexSizeErr("count cannot be negative")
684 return self.data[offset:offset+count]
685
686 def appendData(self, arg):
687 self.data = self.data + arg
688 self.nodeValue = self.data
Fred Drake33d2b842001-04-04 15:15:18 +0000689 self.length = len(self.data)
Fred Drake87432f42001-04-04 14:09:46 +0000690
691 def insertData(self, offset, arg):
692 if offset < 0:
693 raise xml.dom.IndexSizeErr("offset cannot be negative")
694 if offset >= len(self.data):
695 raise xml.dom.IndexSizeErr("offset cannot be beyond end of data")
696 if arg:
697 self.data = "%s%s%s" % (
698 self.data[:offset], arg, self.data[offset:])
699 self.nodeValue = self.data
Fred Drake33d2b842001-04-04 15:15:18 +0000700 self.length = len(self.data)
Fred Drake87432f42001-04-04 14:09:46 +0000701
702 def deleteData(self, offset, count):
703 if offset < 0:
704 raise xml.dom.IndexSizeErr("offset cannot be negative")
705 if offset >= len(self.data):
706 raise xml.dom.IndexSizeErr("offset cannot be beyond end of data")
707 if count < 0:
708 raise xml.dom.IndexSizeErr("count cannot be negative")
709 if count:
710 self.data = self.data[:offset] + self.data[offset+count:]
711 self.nodeValue = self.data
Fred Drake33d2b842001-04-04 15:15:18 +0000712 self.length = len(self.data)
Fred Drake87432f42001-04-04 14:09:46 +0000713
714 def replaceData(self, offset, count, arg):
715 if offset < 0:
716 raise xml.dom.IndexSizeErr("offset cannot be negative")
717 if offset >= len(self.data):
718 raise xml.dom.IndexSizeErr("offset cannot be beyond end of data")
719 if count < 0:
720 raise xml.dom.IndexSizeErr("count cannot be negative")
721 if count:
722 self.data = "%s%s%s" % (
723 self.data[:offset], arg, self.data[offset+count:])
724 self.nodeValue = self.data
Fred Drake33d2b842001-04-04 15:15:18 +0000725 self.length = len(self.data)
Fred Drake87432f42001-04-04 14:09:46 +0000726
727class Text(CharacterData):
728 nodeType = Node.TEXT_NODE
729 nodeName = "#text"
730 attributes = None
731 childNodeTypes = ()
Fred Drake55c38192000-06-29 19:39:57 +0000732
Fred Drakef7cf40d2000-12-14 18:16:11 +0000733 def splitText(self, offset):
734 if offset < 0 or offset > len(self.data):
Martin v. Löwisd5fb58f2001-01-27 08:38:34 +0000735 raise xml.dom.IndexSizeErr("illegal offset value")
Fred Drakef7cf40d2000-12-14 18:16:11 +0000736 newText = Text(self.data[offset:])
737 next = self.nextSibling
738 if self.parentNode and self in self.parentNode.childNodes:
739 if next is None:
740 self.parentNode.appendChild(newText)
741 else:
742 self.parentNode.insertBefore(newText, next)
743 self.data = self.data[:offset]
Fred Drake33d2b842001-04-04 15:15:18 +0000744 self.nodeValue = self.data
745 self.length = len(self.data)
Fred Drakef7cf40d2000-12-14 18:16:11 +0000746 return newText
747
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000748 def writexml(self, writer, indent="", addindent="", newl=""):
749 _write_data(writer, "%s%s%s"%(indent, self.data, newl))
Fred Drake55c38192000-06-29 19:39:57 +0000750
Fred Drake87432f42001-04-04 14:09:46 +0000751
752class CDATASection(Text):
753 nodeType = Node.CDATA_SECTION_NODE
754 nodeName = "#cdata-section"
755
756 def writexml(self, writer, indent="", addindent="", newl=""):
Guido van Rossum5b5e0b92001-09-19 13:28:25 +0000757 writer.write("<![CDATA[%s]]>" % self.data)
Fred Drake87432f42001-04-04 14:09:46 +0000758
759
Fred Drake1f549022000-09-24 05:21:58 +0000760def _nssplit(qualifiedName):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000761 fields = _string.split(qualifiedName, ':', 1)
Paul Prescod73678da2000-07-01 04:58:47 +0000762 if len(fields) == 2:
763 return fields
764 elif len(fields) == 1:
Fred Drake49a5d032001-11-30 22:21:58 +0000765 return (None, fields[0])
Paul Prescod73678da2000-07-01 04:58:47 +0000766
Fred Drakef7cf40d2000-12-14 18:16:11 +0000767
768class DocumentType(Node):
769 nodeType = Node.DOCUMENT_TYPE_NODE
770 nodeValue = None
771 attributes = None
772 name = None
773 publicId = None
774 systemId = None
Fred Drakedc806702001-04-05 14:41:30 +0000775 internalSubset = None
Fred Drakef7cf40d2000-12-14 18:16:11 +0000776 entities = None
777 notations = None
778
779 def __init__(self, qualifiedName):
780 Node.__init__(self)
781 if qualifiedName:
782 prefix, localname = _nssplit(qualifiedName)
783 self.name = localname
784
785
786class DOMImplementation:
787 def hasFeature(self, feature, version):
788 if version not in ("1.0", "2.0"):
789 return 0
790 feature = _string.lower(feature)
791 return feature == "core"
792
793 def createDocument(self, namespaceURI, qualifiedName, doctype):
794 if doctype and doctype.parentNode is not None:
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000795 raise xml.dom.WrongDocumentErr(
796 "doctype object owned by another DOM tree")
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000797 doc = self._createDocument()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000798 if doctype is None:
799 doctype = self.createDocumentType(qualifiedName, None, None)
Martin v. Löwisb417be22001-02-06 01:16:06 +0000800 if not qualifiedName:
801 # The spec is unclear what to raise here; SyntaxErr
802 # would be the other obvious candidate. Since Xerces raises
803 # InvalidCharacterErr, and since SyntaxErr is not listed
804 # for createDocument, that seems to be the better choice.
805 # XXX: need to check for illegal characters here and in
806 # createElement.
807 raise xml.dom.InvalidCharacterErr("Element with no name")
808 prefix, localname = _nssplit(qualifiedName)
809 if prefix == "xml" \
810 and namespaceURI != "http://www.w3.org/XML/1998/namespace":
811 raise xml.dom.NamespaceErr("illegal use of 'xml' prefix")
812 if prefix and not namespaceURI:
813 raise xml.dom.NamespaceErr(
814 "illegal use of prefix without namespaces")
815 element = doc.createElementNS(namespaceURI, qualifiedName)
816 doc.appendChild(element)
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000817 doctype.parentNode = doctype.ownerDocument = doc
Fred Drakef7cf40d2000-12-14 18:16:11 +0000818 doc.doctype = doctype
819 doc.implementation = self
820 return doc
821
822 def createDocumentType(self, qualifiedName, publicId, systemId):
823 doctype = DocumentType(qualifiedName)
824 doctype.publicId = publicId
825 doctype.systemId = systemId
826 return doctype
827
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000828 # internal
829 def _createDocument(self):
830 return Document()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000831
Fred Drake1f549022000-09-24 05:21:58 +0000832class Document(Node):
833 nodeType = Node.DOCUMENT_NODE
Fred Drake4ccf4a12000-11-21 22:02:22 +0000834 nodeName = "#document"
835 nodeValue = None
836 attributes = None
Fred Drakef7cf40d2000-12-14 18:16:11 +0000837 doctype = None
838 parentNode = None
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000839 previousSibling = nextSibling = None
Fred Drakef7cf40d2000-12-14 18:16:11 +0000840
841 implementation = DOMImplementation()
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000842 childNodeTypes = (Node.ELEMENT_NODE, Node.PROCESSING_INSTRUCTION_NODE,
843 Node.COMMENT_NODE, Node.DOCUMENT_TYPE_NODE)
Fred Drake55c38192000-06-29 19:39:57 +0000844
Fred Drake1f549022000-09-24 05:21:58 +0000845 def appendChild(self, node):
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000846 if node.nodeType not in self.childNodeTypes:
847 raise HierarchyRequestErr, \
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000848 "%s cannot be child of %s" % (repr(node), repr(self))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000849 if node.parentNode is not None:
850 node.parentNode.removeChild(node)
851
Fred Drakef7cf40d2000-12-14 18:16:11 +0000852 if node.nodeType == Node.ELEMENT_NODE \
853 and self._get_documentElement():
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000854 raise xml.dom.HierarchyRequestErr(
855 "two document elements disallowed")
Fred Drake4ccf4a12000-11-21 22:02:22 +0000856 return Node.appendChild(self, node)
Paul Prescod73678da2000-07-01 04:58:47 +0000857
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000858 def removeChild(self, oldChild):
859 self.childNodes.remove(oldChild)
860 oldChild.nextSibling = oldChild.previousSibling = None
861 oldChild.parentNode = None
862 if self.documentElement is oldChild:
863 self.documentElement = None
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000864
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000865 return oldChild
866
Fred Drakef7cf40d2000-12-14 18:16:11 +0000867 def _get_documentElement(self):
868 for node in self.childNodes:
869 if node.nodeType == Node.ELEMENT_NODE:
870 return node
871
Fred Draked1572372001-09-29 04:58:32 +0000872 try:
873 property
874 except NameError:
875 pass
876 else:
877 documentElement = property(_get_documentElement,
878 doc="Top-level element of this document.")
879
Fred Drakef7cf40d2000-12-14 18:16:11 +0000880 def unlink(self):
881 if self.doctype is not None:
882 self.doctype.unlink()
883 self.doctype = None
884 Node.unlink(self)
885
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000886 def createDocumentFragment(self):
887 d = DocumentFragment()
888 d.ownerDoc = self
889 return d
Fred Drake55c38192000-06-29 19:39:57 +0000890
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000891 def createElement(self, tagName):
892 e = Element(tagName)
893 e.ownerDocument = self
894 return e
Fred Drake55c38192000-06-29 19:39:57 +0000895
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000896 def createTextNode(self, data):
897 t = Text(data)
898 t.ownerDocument = self
899 return t
Fred Drake55c38192000-06-29 19:39:57 +0000900
Fred Drake87432f42001-04-04 14:09:46 +0000901 def createCDATASection(self, data):
902 c = CDATASection(data)
903 c.ownerDocument = self
904 return c
905
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000906 def createComment(self, data):
907 c = Comment(data)
908 c.ownerDocument = self
909 return c
Fred Drake55c38192000-06-29 19:39:57 +0000910
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000911 def createProcessingInstruction(self, target, data):
912 p = ProcessingInstruction(target, data)
913 p.ownerDocument = self
914 return p
915
916 def createAttribute(self, qName):
917 a = Attr(qName)
918 a.ownerDocument = self
Martin v. Löwiscb67ea12001-03-31 16:30:40 +0000919 a.value = ""
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000920 return a
Fred Drake55c38192000-06-29 19:39:57 +0000921
922 def createElementNS(self, namespaceURI, qualifiedName):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000923 prefix, localName = _nssplit(qualifiedName)
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000924 e = Element(qualifiedName, namespaceURI, prefix, localName)
925 e.ownerDocument = self
926 return e
Fred Drake55c38192000-06-29 19:39:57 +0000927
928 def createAttributeNS(self, namespaceURI, qualifiedName):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000929 prefix, localName = _nssplit(qualifiedName)
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000930 a = Attr(qualifiedName, namespaceURI, localName, prefix)
931 a.ownerDocument = self
Martin v. Löwiscb67ea12001-03-31 16:30:40 +0000932 a.value = ""
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000933 return a
Fred Drake55c38192000-06-29 19:39:57 +0000934
Fred Drake1f549022000-09-24 05:21:58 +0000935 def getElementsByTagName(self, name):
Martin v. Löwis95700f72002-03-15 13:51:59 +0000936 return _getElementsByTagNameHelper(self, name, NodeList())
Fred Drakefbe7b4f2001-07-04 06:25:53 +0000937
938 def getElementsByTagNameNS(self, namespaceURI, localName):
Fred Drake15b68932002-03-15 14:37:23 +0000939 return _getElementsByTagNameNSHelper(self, namespaceURI, localName,
940 NodeList())
Fred Drake55c38192000-06-29 19:39:57 +0000941
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000942 def writexml(self, writer, indent="", addindent="", newl=""):
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000943 writer.write('<?xml version="1.0" ?>\n')
Fred Drake55c38192000-06-29 19:39:57 +0000944 for node in self.childNodes:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000945 node.writexml(writer, indent, addindent, newl)
Fred Drake55c38192000-06-29 19:39:57 +0000946
Fred Drake4ccf4a12000-11-21 22:02:22 +0000947def _get_StringIO():
Fred Drakef7cf40d2000-12-14 18:16:11 +0000948 # we can't use cStringIO since it doesn't support Unicode strings
949 from StringIO import StringIO
Fred Drake4ccf4a12000-11-21 22:02:22 +0000950 return StringIO()
951
Fred Drake1f549022000-09-24 05:21:58 +0000952def _doparse(func, args, kwargs):
953 events = apply(func, args, kwargs)
954 toktype, rootNode = events.getEvent()
955 events.expandNode(rootNode)
Martin v. Löwisb417be22001-02-06 01:16:06 +0000956 events.clear()
Fred Drake55c38192000-06-29 19:39:57 +0000957 return rootNode
958
Fred Drake1f549022000-09-24 05:21:58 +0000959def parse(*args, **kwargs):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000960 """Parse a file into a DOM by filename or file object."""
Fred Drake4ccf4a12000-11-21 22:02:22 +0000961 from xml.dom import pulldom
Fred Drake1f549022000-09-24 05:21:58 +0000962 return _doparse(pulldom.parse, args, kwargs)
Fred Drake55c38192000-06-29 19:39:57 +0000963
Fred Drake1f549022000-09-24 05:21:58 +0000964def parseString(*args, **kwargs):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000965 """Parse a file into a DOM from a string."""
Fred Drake4ccf4a12000-11-21 22:02:22 +0000966 from xml.dom import pulldom
Fred Drake1f549022000-09-24 05:21:58 +0000967 return _doparse(pulldom.parseString, args, kwargs)
Martin v. Löwis7edbd4f2001-02-22 14:05:50 +0000968
969def getDOMImplementation():
970 return Document.implementation