X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=python%2Fxmltemplate.py;h=182b8e4a3712953d48a572d4183791ef4c3946ea;hb=c00c787ebcd86ec94285b708db96a20286e57ab0;hp=8f2b9222eab4340ee0866138d3df8ce83f42133c;hpb=975f70845bb6f795508bd1a75411122748c76a27;p=xml-template

diff --git a/python/xmltemplate.py b/python/xmltemplate.py
index 8f2b922..182b8e4 100644
--- 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
@@ -46,7 +63,7 @@ def process(node, obj, clean = True):
 
 			if not processed:
 				process(child, obj, clean)
-	elif isinstance(obj, list):			# repeat
+	elif hasattr(obj, '__iter__'):			# repeat
 		doc = _get_document_element(node)
 		frag = doc.createElement("temporary-fragment")     # ugh
 
@@ -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,10 +98,27 @@ 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):
-	pass
+	if node.nodeType == xml.dom.Node.ELEMENT_NODE and node.namespaceURI == "http://template.sesse.net/":
+		# as this is a dummy node, we want to remove it and move everything further up
+		# after we've done any required replacements
+		doc = _get_document_element(node)
+		parent = node.parentNode
+
+		while not node.firstChild is None:
+			child = node.firstChild
+			node.removeChild(child)
+			parent.insertBefore(child, node)
+
+		parent.removeChild(node)
 
 # ugh
 def _get_document_element(node):