blob: 967320f5aa5fc1cfb086b230a07f3f934717c98d [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 Drake49a5d032001-11-30 22:21:58 +000017from xml.dom import HierarchyRequestErr, EMPTY_NAMESPACE
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +000018
Fred Drake4ccf4a12000-11-21 22:02:22 +000019# localize the types, and allow support for Unicode values if available:
Fred Drake1f549022000-09-24 05:21:58 +000020import types
Fred Drake4ccf4a12000-11-21 22:02:22 +000021_TupleType = types.TupleType
22try:
23 _StringTypes = (types.StringType, types.UnicodeType)
24except AttributeError:
25 _StringTypes = (types.StringType,)
26del types
27
Fred Drakef7cf40d2000-12-14 18:16:11 +000028import xml.dom
Fred Drake55c38192000-06-29 19:39:57 +000029
Fred Drake3ac6a092001-09-28 04:33:06 +000030
31if list is type([]):
32 class NodeList(list):
33 def item(self, index):
34 if 0 <= index < len(self):
35 return self[index]
36
Fred Drake575712e2001-09-28 20:25:45 +000037 length = property(lambda self: len(self),
38 doc="The number of nodes in the NodeList.")
Fred Drake3ac6a092001-09-28 04:33:06 +000039
40else:
41 def NodeList():
42 return []
Martin v. Löwis95700f72002-03-15 13:51:59 +000043
Fred Drake3ac6a092001-09-28 04:33:06 +000044
Fred Drake575712e2001-09-28 20:25:45 +000045class Node(xml.dom.Node):
Fred Drake1f549022000-09-24 05:21:58 +000046 allnodes = {}
47 _debug = 0
48 _makeParentNodes = 1
49 debug = None
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +000050 childNodeTypes = ()
Martin v. Löwis126f2f62001-03-13 10:50:13 +000051 namespaceURI = None # this is non-null only for elements and attributes
Fred Drake575712e2001-09-28 20:25:45 +000052 parentNode = None
53 ownerDocument = None
Martin v. Löwis52ce0d02001-01-27 08:47:37 +000054
Fred Drake1f549022000-09-24 05:21:58 +000055 def __init__(self):
Fred Drake3ac6a092001-09-28 04:33:06 +000056 self.childNodes = NodeList()
Fred Drake16f63292000-10-23 18:09:50 +000057 if Node._debug:
Fred Drake1f549022000-09-24 05:21:58 +000058 index = repr(id(self)) + repr(self.__class__)
59 Node.allnodes[index] = repr(self.__dict__)
60 if Node.debug is None:
Fred Drake4ccf4a12000-11-21 22:02:22 +000061 Node.debug = _get_StringIO()
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +000062 #open("debug4.out", "w")
Fred Drake1f549022000-09-24 05:21:58 +000063 Node.debug.write("create %s\n" % index)
Fred Drake55c38192000-06-29 19:39:57 +000064
Fred Drake1f549022000-09-24 05:21:58 +000065 def __nonzero__(self):
66 return 1
Fred Drake55c38192000-06-29 19:39:57 +000067
Martin v. Löwis7d650ca2002-06-30 15:05:00 +000068 def toxml(self, encoding = None):
69 return self.toprettyxml("", "", encoding)
Fred Drake55c38192000-06-29 19:39:57 +000070
Martin v. Löwis7d650ca2002-06-30 15:05:00 +000071 def toprettyxml(self, indent="\t", newl="\n", encoding = None):
Martin v. Löwiscb67ea12001-03-31 16:30:40 +000072 # indent = the indentation string to prepend, per level
73 # newl = the newline string to append
74 writer = _get_StringIO()
Martin v. Löwis7d650ca2002-06-30 15:05:00 +000075 if encoding is not None:
76 import codecs
77 # Can't use codecs.getwriter to preserve 2.0 compatibility
78 writer = codecs.lookup(encoding)[3](writer)
79 if self.nodeType == Node.DOCUMENT_NODE:
80 # Can pass encoding only to document, to put it into XML header
81 self.writexml(writer, "", indent, newl, encoding)
82 else:
83 self.writexml(writer, "", indent, newl)
Martin v. Löwiscb67ea12001-03-31 16:30:40 +000084 return writer.getvalue()
Martin v. Löwis46fa39a2001-02-06 00:14:08 +000085
Fred Drake1f549022000-09-24 05:21:58 +000086 def hasChildNodes(self):
87 if self.childNodes:
88 return 1
89 else:
90 return 0
Fred Drake55c38192000-06-29 19:39:57 +000091
Fred Drake1f549022000-09-24 05:21:58 +000092 def _get_firstChild(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +000093 if self.childNodes:
94 return self.childNodes[0]
Paul Prescod73678da2000-07-01 04:58:47 +000095
Fred Drake1f549022000-09-24 05:21:58 +000096 def _get_lastChild(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +000097 if self.childNodes:
98 return self.childNodes[-1]
Paul Prescod73678da2000-07-01 04:58:47 +000099
Fred Draked1572372001-09-29 04:58:32 +0000100 try:
101 property
102 except NameError:
103 def __getattr__(self, key):
104 if key[0:2] == "__":
105 raise AttributeError, key
106 # getattr should never call getattr!
107 if self.__dict__.has_key("inGetAttr"):
108 del self.inGetAttr
109 raise AttributeError, key
110
111 prefix, attrname = key[:5], key[5:]
112 if prefix == "_get_":
113 self.inGetAttr = 1
114 if hasattr(self, attrname):
115 del self.inGetAttr
116 return (lambda self=self, attrname=attrname:
117 getattr(self, attrname))
118 else:
119 del self.inGetAttr
120 raise AttributeError, key
121 else:
122 self.inGetAttr = 1
123 try:
124 func = getattr(self, "_get_" + key)
125 except AttributeError:
126 raise AttributeError, key
127 del self.inGetAttr
128 return func()
129 else:
130 firstChild = property(_get_firstChild,
131 doc="First child node, or None.")
132 lastChild = property(_get_lastChild,
133 doc="Last child node, or None.")
134
Fred Drake1f549022000-09-24 05:21:58 +0000135 def insertBefore(self, newChild, refChild):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000136 if newChild.nodeType == self.DOCUMENT_FRAGMENT_NODE:
Fred Drakee50959a2001-12-06 04:32:18 +0000137 for c in tuple(newChild.childNodes):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000138 self.insertBefore(c, refChild)
139 ### The DOM does not clearly specify what to return in this case
140 return newChild
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000141 if newChild.nodeType not in self.childNodeTypes:
142 raise HierarchyRequestErr, \
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000143 "%s cannot be child of %s" % (repr(newChild), repr(self))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000144 if newChild.parentNode is not None:
145 newChild.parentNode.removeChild(newChild)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000146 if refChild is None:
147 self.appendChild(newChild)
148 else:
149 index = self.childNodes.index(refChild)
150 self.childNodes.insert(index, newChild)
151 newChild.nextSibling = refChild
152 refChild.previousSibling = newChild
153 if index:
154 node = self.childNodes[index-1]
155 node.nextSibling = newChild
156 newChild.previousSibling = node
157 else:
158 newChild.previousSibling = None
159 if self._makeParentNodes:
160 newChild.parentNode = self
161 return newChild
Fred Drake55c38192000-06-29 19:39:57 +0000162
Fred Drake1f549022000-09-24 05:21:58 +0000163 def appendChild(self, node):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000164 if node.nodeType == self.DOCUMENT_FRAGMENT_NODE:
Fred Drakee50959a2001-12-06 04:32:18 +0000165 for c in tuple(node.childNodes):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000166 self.appendChild(c)
167 ### The DOM does not clearly specify what to return in this case
168 return node
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000169 if node.nodeType not in self.childNodeTypes:
170 raise HierarchyRequestErr, \
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000171 "%s cannot be child of %s" % (repr(node), repr(self))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000172 if node.parentNode is not None:
173 node.parentNode.removeChild(node)
Fred Drake13a30692000-10-09 20:04:16 +0000174 if self.childNodes:
175 last = self.lastChild
176 node.previousSibling = last
177 last.nextSibling = node
178 else:
179 node.previousSibling = None
180 node.nextSibling = None
Fred Drake1f549022000-09-24 05:21:58 +0000181 self.childNodes.append(node)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000182 if self._makeParentNodes:
183 node.parentNode = self
Paul Prescod73678da2000-07-01 04:58:47 +0000184 return node
185
Fred Drake1f549022000-09-24 05:21:58 +0000186 def replaceChild(self, newChild, oldChild):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000187 if newChild.nodeType == self.DOCUMENT_FRAGMENT_NODE:
188 refChild = oldChild.nextSibling
189 self.removeChild(oldChild)
190 return self.insertBefore(newChild, refChild)
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000191 if newChild.nodeType not in self.childNodeTypes:
192 raise HierarchyRequestErr, \
Guido van Rossum9e1fe1e2001-02-05 19:17:50 +0000193 "%s cannot be child of %s" % (repr(newChild), repr(self))
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000194 if newChild.parentNode is not None:
195 newChild.parentNode.removeChild(newChild)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000196 if newChild is oldChild:
197 return
Fred Drake1f549022000-09-24 05:21:58 +0000198 index = self.childNodes.index(oldChild)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000199 self.childNodes[index] = newChild
200 if self._makeParentNodes:
201 newChild.parentNode = self
202 oldChild.parentNode = None
203 newChild.nextSibling = oldChild.nextSibling
204 newChild.previousSibling = oldChild.previousSibling
Martin v. Löwis156c3372000-12-28 18:40:56 +0000205 oldChild.nextSibling = None
Fred Drake4ccf4a12000-11-21 22:02:22 +0000206 oldChild.previousSibling = None
Martin v. Löwis156c3372000-12-28 18:40:56 +0000207 if newChild.previousSibling:
208 newChild.previousSibling.nextSibling = newChild
209 if newChild.nextSibling:
210 newChild.nextSibling.previousSibling = newChild
Fred Drake4ccf4a12000-11-21 22:02:22 +0000211 return oldChild
Paul Prescod73678da2000-07-01 04:58:47 +0000212
Fred Drake1f549022000-09-24 05:21:58 +0000213 def removeChild(self, oldChild):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000214 self.childNodes.remove(oldChild)
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000215 if oldChild.nextSibling is not None:
216 oldChild.nextSibling.previousSibling = oldChild.previousSibling
217 if oldChild.previousSibling is not None:
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000218 oldChild.previousSibling.nextSibling = oldChild.nextSibling
Andrew M. Kuchling04a45e92000-12-20 14:47:24 +0000219 oldChild.nextSibling = oldChild.previousSibling = None
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000220
Fred Drake4ccf4a12000-11-21 22:02:22 +0000221 if self._makeParentNodes:
222 oldChild.parentNode = None
223 return oldChild
224
225 def normalize(self):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000226 L = []
227 for child in self.childNodes:
228 if child.nodeType == Node.TEXT_NODE:
229 data = child.data
230 if data and L and L[-1].nodeType == child.nodeType:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000231 # collapse text node
232 node = L[-1]
233 node.data = node.nodeValue = node.data + child.data
234 node.nextSibling = child.nextSibling
235 child.unlink()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000236 elif data:
237 if L:
238 L[-1].nextSibling = child
239 child.previousSibling = L[-1]
240 else:
241 child.previousSibling = None
242 L.append(child)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000243 else:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000244 # empty text node; discard
245 child.unlink()
246 else:
247 if L:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000248 L[-1].nextSibling = child
249 child.previousSibling = L[-1]
Fred Drakef7cf40d2000-12-14 18:16:11 +0000250 else:
251 child.previousSibling = None
252 L.append(child)
253 if child.nodeType == Node.ELEMENT_NODE:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000254 child.normalize()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000255 self.childNodes[:] = L
Paul Prescod73678da2000-07-01 04:58:47 +0000256
Fred Drake1f549022000-09-24 05:21:58 +0000257 def cloneNode(self, deep):
Paul Prescod73678da2000-07-01 04:58:47 +0000258 import new
Fred Drake4ccf4a12000-11-21 22:02:22 +0000259 clone = new.instance(self.__class__, self.__dict__.copy())
260 if self._makeParentNodes:
261 clone.parentNode = None
Fred Drake3ac6a092001-09-28 04:33:06 +0000262 clone.childNodes = NodeList()
Fred Drake4ccf4a12000-11-21 22:02:22 +0000263 if deep:
264 for child in self.childNodes:
265 clone.appendChild(child.cloneNode(1))
Paul Prescod73678da2000-07-01 04:58:47 +0000266 return clone
Fred Drake55c38192000-06-29 19:39:57 +0000267
Fred Drake25239772001-02-02 19:40:19 +0000268 # DOM Level 3 (Working Draft 2001-Jan-26)
269
270 def isSameNode(self, other):
271 return self is other
272
273 # minidom-specific API:
274
Fred Drake1f549022000-09-24 05:21:58 +0000275 def unlink(self):
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000276 self.parentNode = self.ownerDocument = None
Fred Drake4ccf4a12000-11-21 22:02:22 +0000277 for child in self.childNodes:
278 child.unlink()
Fred Drake1f549022000-09-24 05:21:58 +0000279 self.childNodes = None
Paul Prescod4221ff02000-10-13 20:11:42 +0000280 self.previousSibling = None
281 self.nextSibling = None
Paul Prescod73678da2000-07-01 04:58:47 +0000282 if Node._debug:
Fred Drake1f549022000-09-24 05:21:58 +0000283 index = repr(id(self)) + repr(self.__class__)
284 self.debug.write("Deleting: %s\n" % index)
Paul Prescod73678da2000-07-01 04:58:47 +0000285 del Node.allnodes[index]
Fred Drake55c38192000-06-29 19:39:57 +0000286
Fred Drake1f549022000-09-24 05:21:58 +0000287def _write_data(writer, data):
Fred Drake55c38192000-06-29 19:39:57 +0000288 "Writes datachars to writer."
Neal Norwitzab199622002-05-31 20:46:39 +0000289 data = data.replace("&", "&amp;")
290 data = data.replace("<", "&lt;")
291 data = data.replace("\"", "&quot;")
292 data = data.replace(">", "&gt;")
Fred Drake55c38192000-06-29 19:39:57 +0000293 writer.write(data)
294
Fred Drake1f549022000-09-24 05:21:58 +0000295def _getElementsByTagNameHelper(parent, name, rc):
Fred Drake55c38192000-06-29 19:39:57 +0000296 for node in parent.childNodes:
Fred Drake1f549022000-09-24 05:21:58 +0000297 if node.nodeType == Node.ELEMENT_NODE and \
298 (name == "*" or node.tagName == name):
299 rc.append(node)
300 _getElementsByTagNameHelper(node, name, rc)
Fred Drake55c38192000-06-29 19:39:57 +0000301 return rc
302
Fred Drake1f549022000-09-24 05:21:58 +0000303def _getElementsByTagNameNSHelper(parent, nsURI, localName, rc):
Fred Drake55c38192000-06-29 19:39:57 +0000304 for node in parent.childNodes:
Fred Drake1f549022000-09-24 05:21:58 +0000305 if node.nodeType == Node.ELEMENT_NODE:
Martin v. Löwised525fb2001-06-03 14:06:42 +0000306 if ((localName == "*" or node.localName == localName) and
Fred Drake1f549022000-09-24 05:21:58 +0000307 (nsURI == "*" or node.namespaceURI == nsURI)):
308 rc.append(node)
Fred Drakef7cf40d2000-12-14 18:16:11 +0000309 _getElementsByTagNameNSHelper(node, nsURI, localName, rc)
310 return rc
Fred Drake55c38192000-06-29 19:39:57 +0000311
Martin v. Löwis126f2f62001-03-13 10:50:13 +0000312class DocumentFragment(Node):
313 nodeType = Node.DOCUMENT_FRAGMENT_NODE
314 nodeName = "#document-fragment"
315 nodeValue = None
316 attributes = None
317 parentNode = None
318 childNodeTypes = (Node.ELEMENT_NODE,
319 Node.TEXT_NODE,
320 Node.CDATA_SECTION_NODE,
321 Node.ENTITY_REFERENCE_NODE,
322 Node.PROCESSING_INSTRUCTION_NODE,
323 Node.COMMENT_NODE,
324 Node.NOTATION_NODE)
325
326
Fred Drake55c38192000-06-29 19:39:57 +0000327class Attr(Node):
Fred Drake1f549022000-09-24 05:21:58 +0000328 nodeType = Node.ATTRIBUTE_NODE
Fred Drake4ccf4a12000-11-21 22:02:22 +0000329 attributes = None
330 ownerElement = None
Andrew M. Kuchling291ed4f2000-12-31 03:50:23 +0000331 childNodeTypes = (Node.TEXT_NODE, Node.ENTITY_REFERENCE_NODE)
Martin v. Löwis52ce0d02001-01-27 08:47:37 +0000332
Fred Drake49a5d032001-11-30 22:21:58 +0000333 def __init__(self, qName, namespaceURI=EMPTY_NAMESPACE, localName=None, prefix=None):
Fred Drake55c38192000-06-29 19:39:57 +0000334 # skip setattr for performance
Fred Drake4ccf4a12000-11-21 22:02:22 +0000335 d = self.__dict__
336 d["localName"] = localName or qName
337 d["nodeName"] = d["name"] = qName
338 d["namespaceURI"] = namespaceURI
339 d["prefix"] = prefix
Fred Drake1f549022000-09-24 05:21:58 +0000340 Node.__init__(self)
Paul Prescod73678da2000-07-01 04:58:47 +0000341 # nodeValue and value are set elsewhere
Fred Drake55c38192000-06-29 19:39:57 +0000342
Fred Drake1f549022000-09-24 05:21:58 +0000343 def __setattr__(self, name, value):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000344 d = self.__dict__
Fred Drake1f549022000-09-24 05:21:58 +0000345 if name in ("value", "nodeValue"):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000346 d["value"] = d["nodeValue"] = value
347 elif name in ("name", "nodeName"):
348 d["name"] = d["nodeName"] = value
Fred Drake55c38192000-06-29 19:39:57 +0000349 else:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000350 d[name] = value
Fred Drake55c38192000-06-29 19:39:57 +0000351
Fred Drake4ccf4a12000-11-21 22:02:22 +0000352 def cloneNode(self, deep):
353 clone = Node.cloneNode(self, deep)
354 if clone.__dict__.has_key("ownerElement"):
355 del clone.ownerElement
356 return clone
357
Fred Drakef7cf40d2000-12-14 18:16:11 +0000358
359class NamedNodeMap:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000360 """The attribute list is a transient interface to the underlying
361 dictionaries. Mutations here will change the underlying element's
Fred Drakef7cf40d2000-12-14 18:16:11 +0000362 dictionary.
363
364 Ordering is imposed artificially and does not reflect the order of
365 attributes as found in an input document.
366 """
Fred Drake4ccf4a12000-11-21 22:02:22 +0000367
Fred Drake2998a552001-12-06 18:27:48 +0000368 def __init__(self, attrs, attrsNS, ownerElement):
Fred Drake1f549022000-09-24 05:21:58 +0000369 self._attrs = attrs
370 self._attrsNS = attrsNS
Fred Drake2998a552001-12-06 18:27:48 +0000371 self._ownerElement = ownerElement
Fred Drakef7cf40d2000-12-14 18:16:11 +0000372
Fred Draked1572372001-09-29 04:58:32 +0000373 try:
374 property
375 except NameError:
376 def __getattr__(self, name):
377 if name == "length":
378 return len(self._attrs)
379 raise AttributeError, name
380 else:
381 length = property(lambda self: len(self._attrs),
382 doc="Number of nodes in the NamedNodeMap.")
Fred Drake55c38192000-06-29 19:39:57 +0000383
Fred Drake1f549022000-09-24 05:21:58 +0000384 def item(self, index):
Fred Drake55c38192000-06-29 19:39:57 +0000385 try:
Fred Drakef7cf40d2000-12-14 18:16:11 +0000386 return self[self._attrs.keys()[index]]
Fred Drake55c38192000-06-29 19:39:57 +0000387 except IndexError:
388 return None
Fred Drake55c38192000-06-29 19:39:57 +0000389
Fred Drake1f549022000-09-24 05:21:58 +0000390 def items(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000391 L = []
392 for node in self._attrs.values():
Martin v. Löwisd5fb58f2001-01-27 08:38:34 +0000393 L.append((node.nodeName, node.value))
Fred Drake4ccf4a12000-11-21 22:02:22 +0000394 return L
Fred Drake1f549022000-09-24 05:21:58 +0000395
396 def itemsNS(self):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000397 L = []
398 for node in self._attrs.values():
Fred Drake49a5d032001-11-30 22:21:58 +0000399 L.append(((node.namespaceURI, node.localName), node.value))
Fred Drake4ccf4a12000-11-21 22:02:22 +0000400 return L
Fred Drake16f63292000-10-23 18:09:50 +0000401
Fred Drake1f549022000-09-24 05:21:58 +0000402 def keys(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000403 return self._attrs.keys()
Fred Drake55c38192000-06-29 19:39:57 +0000404
Fred Drake1f549022000-09-24 05:21:58 +0000405 def keysNS(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000406 return self._attrsNS.keys()
Fred Drake55c38192000-06-29 19:39:57 +0000407
Fred Drake1f549022000-09-24 05:21:58 +0000408 def values(self):
Paul Prescod73678da2000-07-01 04:58:47 +0000409 return self._attrs.values()
Fred Drake55c38192000-06-29 19:39:57 +0000410
Martin v. Löwisd5fb58f2001-01-27 08:38:34 +0000411 def get(self, name, value = None):
412 return self._attrs.get(name, value)
413
Fred Drake1f549022000-09-24 05:21:58 +0000414 def __len__(self):
Fred Drake55c38192000-06-29 19:39:57 +0000415 return self.length
416
Fred Drake1f549022000-09-24 05:21:58 +0000417 def __cmp__(self, other):
418 if self._attrs is getattr(other, "_attrs", None):
Fred Drake55c38192000-06-29 19:39:57 +0000419 return 0
Fred Drake16f63292000-10-23 18:09:50 +0000420 else:
Fred Drake1f549022000-09-24 05:21:58 +0000421 return cmp(id(self), id(other))
Fred Drake55c38192000-06-29 19:39:57 +0000422
423 #FIXME: is it appropriate to return .value?
Fred Drake1f549022000-09-24 05:21:58 +0000424 def __getitem__(self, attname_or_tuple):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000425 if type(attname_or_tuple) is _TupleType:
Paul Prescod73678da2000-07-01 04:58:47 +0000426 return self._attrsNS[attname_or_tuple]
Fred Drake55c38192000-06-29 19:39:57 +0000427 else:
Paul Prescod73678da2000-07-01 04:58:47 +0000428 return self._attrs[attname_or_tuple]
Fred Drake55c38192000-06-29 19:39:57 +0000429
Paul Prescod1e688272000-07-01 19:21:47 +0000430 # same as set
Fred Drake1f549022000-09-24 05:21:58 +0000431 def __setitem__(self, attname, value):
Fred Drake4ccf4a12000-11-21 22:02:22 +0000432 if type(value) in _StringTypes:
Fred Drake1f549022000-09-24 05:21:58 +0000433 node = Attr(attname)
Fred Drake4ccf4a12000-11-21 22:02:22 +0000434 node.value = value
Fred Drake2998a552001-12-06 18:27:48 +0000435 node.ownerDocument = self._ownerElement.ownerDocument
Paul Prescod1e688272000-07-01 19:21:47 +0000436 else:
Fred Drake4ccf4a12000-11-21 22:02:22 +0000437 if not isinstance(value, Attr):
438 raise TypeError, "value must be a string or Attr object"
Fred Drake1f549022000-09-24 05:21:58 +0000439 node = value
Fred Drakef7cf40d2000-12-14 18:16:11 +0000440 self.setNamedItem(node)
441
442 def setNamedItem(self, node):
Andrew M. Kuchlingbc8f72c2001-02-21 01:30:26 +0000443 if not isinstance(node, Attr):
444 raise HierarchyRequestErr, \
445 "%s cannot be child of %s" % (repr(node), repr(self))
Fred Drakef7cf40d2000-12-14 18:16:11 +0000446 old = self._attrs.get(node.name)
Paul Prescod1e688272000-07-01 19:21:47 +0000447 if old:
448 old.unlink()
Fred Drake1f549022000-09-24 05:21:58 +0000449 self._attrs[node.name] = node
450 self._attrsNS[(node.namespaceURI, node.localName)] = node
Fred Drake2998a552001-12-06 18:27:48 +0000451 node.ownerElement = self._ownerElement
Fred Drakef7cf40d2000-12-14 18:16:11 +0000452 return old
453
454 def setNamedItemNS(self, node):
455 return self.setNamedItem(node)
Paul Prescod73678da2000-07-01 04:58:47 +0000456
Fred Drake1f549022000-09-24 05:21:58 +0000457 def __delitem__(self, attname_or_tuple):
458 node = self[attname_or_tuple]
Paul Prescod73678da2000-07-01 04:58:47 +0000459 node.unlink()
460 del self._attrs[node.name]
461 del self._attrsNS[(node.namespaceURI, node.localName)]
Fred Drakef7cf40d2000-12-14 18:16:11 +0000462
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):
Neal Norwitzab199622002-05-31 20:46:39 +0000761 fields = qualifiedName.split(':', 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
Neal Norwitzab199622002-05-31 20:46:39 +0000790 feature = feature.lower()
Fred Drakef7cf40d2000-12-14 18:16:11 +0000791 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öwis7d650ca2002-06-30 15:05:00 +0000942 def writexml(self, writer, indent="", addindent="", newl="",
943 encoding = None):
944 if encoding is None:
945 writer.write('<?xml version="1.0" ?>\n')
946 else:
947 writer.write('<?xml version="1.0" encoding="%s"?>\n' % encoding)
Fred Drake55c38192000-06-29 19:39:57 +0000948 for node in self.childNodes:
Martin v. Löwis46fa39a2001-02-06 00:14:08 +0000949 node.writexml(writer, indent, addindent, newl)
Fred Drake55c38192000-06-29 19:39:57 +0000950
Fred Drake4ccf4a12000-11-21 22:02:22 +0000951def _get_StringIO():
Fred Drakef7cf40d2000-12-14 18:16:11 +0000952 # we can't use cStringIO since it doesn't support Unicode strings
953 from StringIO import StringIO
Fred Drake4ccf4a12000-11-21 22:02:22 +0000954 return StringIO()
955
Fred Drake1f549022000-09-24 05:21:58 +0000956def _doparse(func, args, kwargs):
957 events = apply(func, args, kwargs)
958 toktype, rootNode = events.getEvent()
959 events.expandNode(rootNode)
Martin v. Löwisb417be22001-02-06 01:16:06 +0000960 events.clear()
Fred Drake55c38192000-06-29 19:39:57 +0000961 return rootNode
962
Fred Drake1f549022000-09-24 05:21:58 +0000963def parse(*args, **kwargs):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000964 """Parse a file into a DOM by filename or file object."""
Fred Drake4ccf4a12000-11-21 22:02:22 +0000965 from xml.dom import pulldom
Fred Drake1f549022000-09-24 05:21:58 +0000966 return _doparse(pulldom.parse, args, kwargs)
Fred Drake55c38192000-06-29 19:39:57 +0000967
Fred Drake1f549022000-09-24 05:21:58 +0000968def parseString(*args, **kwargs):
Fred Drakef7cf40d2000-12-14 18:16:11 +0000969 """Parse a file into a DOM from a string."""
Fred Drake4ccf4a12000-11-21 22:02:22 +0000970 from xml.dom import pulldom
Fred Drake1f549022000-09-24 05:21:58 +0000971 return _doparse(pulldom.parseString, args, kwargs)
Martin v. Löwis7edbd4f2001-02-22 14:05:50 +0000972
973def getDOMImplementation():
974 return Document.implementation