Untested, unfinished first cut at generating an elementtree. Will not work yet.

jgraham · jgraham · commit f34b5003659f · 2006-12-28T22:05:10.000Z
--HG--
extra : convert_revision : svn%3Aacbfec75-9323-0410-a652-858a13e371e0/trunk%40369
diff --git a/src/treebuilders/elementtree.py b/src/treebuilders/elementtree.py
@@ -0,0 +1,94 @@
+try:
+    from xml.etree import ElementTree
+except NameError:
+    from elementtree import ElementTree
+
+import base
+
+class Element(object):
+    def __init__(self, name):
+        self._element = ElementTree.Element()
+        self.name = name
+        self.parent = None
+
+    def _setName(self, name):
+        self._element.tag = name
+    
+    def _getName(self):
+        return self._element.tag
+
+    name = property(_getName, _setName)
+
+    def _getAttributes(self):
+        #Just return the _element which implements a dict-like interface to the attributes
+        return self._element
+
+    def _setAttributes(self, attributes):
+        #Delete existing attributes first
+        #XXX - there may be a better way to do this...
+        for key in self._element.attrib.keys():
+            del self._element.attrib[key]
+        for key, value in attributes.iteritems():
+            self._elements.set(key) = value
+
+    attributes = property(_getAttributes, _setAttributes)
+
+    def appendChild(self, node):
+        self._element.append(node._element)
+        node.parent = self
+
+    def insertBefore(self, node, refNode):
+        index = self._element.getChildren().index(refNode._element)
+        self._element.insert(index, node._element)
+        node.parent = self
+
+    def removeChild(self, node):
+        self._element.remove(node._element)
+        node.parent=None
+
+    def insertText(self, text, insertBefore):
+        if not(len(self._element)):
+            self._element.text += data
+        elif insertBefore is None:
+            #Insert the text as the tail of the last child element
+            self._element[-1].tail += data
+        else:
+            #Insert the text before the specified node
+            children = self._element.getChildren()
+            index = children.index(insertBefore._element)
+            if index > 0:
+                self._element[index-1].tail += data
+            else:
+                self.text += data
+
+    def cloneNode(self):
+        element = Element(self.name)
+        element.attributes = self.attributes
+        return element
+
+class Comment(Element):
+
+    def __init__(self, data):
+        self._element = ElementTree.Comment(data)
+        self.name = None
+        self.parent = None
+
+    def _getData(self):
+        return self._element.text
+
+    def _setData(self, value):
+        self._element.text = value
+
+    data = property(_getData, _setData)
+
+class DocumentType(Element):
+    pass
+
+class TreeBuilder(base.TreeBuilder):
+    documentClass = Element
+    doctypeClass = DocumentType
+    elementClass = Element
+    commentClass = Comment
+
+    def getDocument(self):
+        return self.document._element
