X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Fsp%2Fcons.py;h=1cc93ea5e89b45b315c80f076af9e0eef34bf168;hb=e216962e64bbfd0377511f456edc40570f8efb46;hp=0c36458f8af3b99d84a5dc70470a4d05db7484ea;hpb=8a7ba0dcff7734cce7328e487c1ab38712b84168;p=wrw.git

diff --git a/wrw/sp/cons.py b/wrw/sp/cons.py
index 0c36458..1cc93ea 100644
--- a/wrw/sp/cons.py
+++ b/wrw/sp/cons.py
@@ -1,21 +1,21 @@
-import sys
+import sys, collections
 import xml.dom.minidom
 
 class node(object):
     pass
 
-class text(node, str):
+class text(node, unicode):
     def __todom__(self, doc):
         return doc.createTextNode(self)
 
-class raw(node, str):
+class raw(node, unicode):
     def __todom__(self, doc):
         raise Exception("Cannot convert raw code to DOM objects")
 
 class element(node):
     def __init__(self, ns, name, ctx):
         self.ns = ns
-        self.name = str(name)
+        self.name = unicode(name)
         self.ctx = ctx
         self.attrs = {}
         self.children = []
@@ -23,13 +23,13 @@ class element(node):
     def __call__(self, *children, **attrs):
         for child in children:
             self.ctx.addchild(self, child)
-        for k, v in attrs.items():
+        for k, v in attrs.iteritems():
             self.ctx.addattr(self, k, v)
         return self
 
     def __todom__(self, doc):
         el = doc.createElementNS(self.ns, self.name)
-        for k, v in self.attrs.items():
+        for k, v in self.attrs.iteritems():
             el.setAttribute(k, v)
         for child in self.children:
             el.appendChild(child.__todom__(doc))
@@ -44,9 +44,10 @@ class context(object):
 
     def __init__(self):
         self.nodeconv = {}
-        self.nodeconv[bytes] = lambda ob: text(ob, self.charset)
-        self.nodeconv[str] = text
+        self.nodeconv[str] = lambda ob: text(ob, self.charset)
+        self.nodeconv[unicode] = text
         self.nodeconv[int] = text
+        self.nodeconv[long] = text
         self.nodeconv[float] = text
 
     def nodefrom(self, ob):
@@ -56,16 +57,23 @@ class context(object):
             return ob.__tonode__()
         if type(ob) in self.nodeconv:
             return self.nodeconv[type(ob)](ob)
-        raise Exception("No node conversion known for %s objects" % str(type(ob)))
+        return None
 
     def addchild(self, node, child):
         if child is None:
             return
-        node.children.append(self.nodefrom(child))
+        new = self.nodefrom(child)
+        if new is not None:
+            node.children.append(new)
+        elif isinstance(child, collections.Iterable):
+            for ch in child:
+                self.addchild(node, ch)
+        else:
+            raise Exception("No node conversion known for %s objects" % str(type(child)))
 
     def addattr(self, node, k, v):
         if v is not None:
-            node.attrs[str(k)] = str(v)
+            node.attrs[unicode(k)] = unicode(v)
 
 class constructor(object):
     def __init__(self, ns, elcls=element, ctx=None):