Fix a potential bug in the C++0x cleaning.

[xml-template] / python / xmltemplate.py

diff --git a/python/xmltemplate.py b/python/xmltemplate.py
index b134ae9ffba21c3c114f5f93cb53d80907043778..b24d7642ccda30a0173ccd345fc413a313e19dbd 100644 (file)
--- a/python/xmltemplate.py
+++ b/python/xmltemplate.py
@@ -8,13 +8,22 @@ def process_file(filename, obj, clean = True):
        return doc
 
 def process(node, obj, clean = True):
-       if isinstance(obj, str):                        # overwrite
-               for child in node.childNodes:
-                       node.removeChild(child)
+       if isinstance(obj, basestring):                 # overwrite
+               while not node.firstChild is None:
+                       node.removeChild(node.firstChild)
                doc = _get_document_element(node)
                node.appendChild(doc.createTextNode(obj))
        elif isinstance(obj, xml.dom.Node):             # overwrite
-               pass
+               while not node.firstChild is None:
+                       node.removeChild(node.firstChild)
+
+               if isinstance(obj, xml.dom.minidom.Document):
+                       obj = obj.documentElement
+
+               newobj = obj.cloneNode(True)
+               node.appendChild(newobj)
+
+               process(newobj, {}, clean)
        elif isinstance(obj, dict):                     # substitute
                for child in node.childNodes:
                        processed = False
@@ -23,13 +32,21 @@ def process(node, obj, clean = True):
                                id = None
 
                                attrs = child.attributes
+                               attrs_to_remove = []
                                if not attrs is None:
                                        for i in range(attrs.length):
-                                               attr = attrs.item(0)
-                                               if attr.namespaceURI == "http://template.sesse.net/" or attr.name == "id":
+                                               attr = attrs.item(i)
+                                               if attr.namespaceURI == "http://template.sesse.net/" and attr.localName == "id":
                                                        id = attr.value
                                                        if clean:
-                                                               child.removeAttribute(attr.name)
+                                                               attrs_to_remove.append(attr.name)
+                                               if attr.name.startswith("xmlns:") and attr.value == "http://template.sesse.net/" and clean:
+                                                       attrs_to_remove.append(attr.name)
+
+                                       for a in attrs_to_remove:
+                                               if child.hasAttribute(a):
+                                                       child.removeAttribute(a)
+
 
                                # check all substitutions to see if we found anything
                                # appropriate
@@ -56,11 +73,12 @@ def process(node, obj, clean = True):
                        frag.appendChild(child)
 
                for instance in obj:
-                       newnode = frag.cloneNode(True)
-                       node.appendChild(newnode)
-                       process(newnode, instance, clean)
-                       if clean:
-                               _clean(newnode)
+                       if instance is not None:
+                               newnode = frag.cloneNode(True)
+                               node.appendChild(newnode)
+                               process(newnode, instance, clean)
+                               if clean:
+                                       _clean(newnode)
 
                # remove all the <fragment> tags
 
@@ -80,6 +98,12 @@ def process(node, obj, clean = True):
                _clean(node)
 
 def alternate(tag, array, *elems):
+       i = 0
+       for ref in array:
+               if ref is not None:
+                       ref[tag] = elems[i % len(elems)]
+                       i = i + 1
+               
        return array
 
 def _clean(node):