X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Fsp%2Fcons.py;h=1cc93ea5e89b45b315c80f076af9e0eef34bf168;hb=e216962e64bbfd0377511f456edc40570f8efb46;hp=4ae21b84e03adfe39d25c678e724743f1a119262;hpb=1273171c643351d41f6d4fbf90f236432d005fc6;p=wrw.git diff --git a/wrw/sp/cons.py b/wrw/sp/cons.py index 4ae21b8..1cc93ea 100644 --- a/wrw/sp/cons.py +++ b/wrw/sp/cons.py @@ -4,18 +4,18 @@ 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): @@ -63,16 +64,16 @@ class context(object): return new = self.nodefrom(child) if new is not None: - node.children.append(self.nodefrom(child)) + 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(ob))) + 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):