X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Fsp%2Fcons.py;h=74a6ea6e30830ad324be9a51ee15beda2bc0fec8;hb=c4126c8a6fff40b684cb3332728510725b1a3738;hp=5341b20f44dc6495c91662bb7238810ef1851d42;hpb=ff79cdbf7d4d95e6a84d6b002e28e6df86847954;p=wrw.git diff --git a/wrw/sp/cons.py b/wrw/sp/cons.py index 5341b20..74a6ea6 100644 --- a/wrw/sp/cons.py +++ b/wrw/sp/cons.py @@ -1,44 +1,49 @@ import xml.dom.minidom class node(object): - def __str__(self): - doc = xml.dom.minidom.Document() - return self.__todom__(doc).toxml() + pass -class text(node, unicode): +class text(node, str): def __todom__(self, doc): return doc.createTextNode(self) +class raw(node, str): + 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 = unicode(name) + self.name = str(name) self.ctx = ctx self.attrs = {} self.children = [] def __call__(self, *children, **attrs): for child in children: - self.children.append(self.ctx.nodefrom(child)) - for k, v in attrs.iteritems(): - self.attrs[unicode(k)] = unicode(v) + self.ctx.addchild(self, child) + for k, v in attrs.items(): + 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.iteritems(): + for k, v in self.attrs.items(): el.setAttribute(k, v) for child in self.children: el.appendChild(child.__todom__(doc)) return el + def __str__(self): + doc = xml.dom.minidom.Document() + return self.__todom__(doc).toxml() + class context(object): def __init__(self): self.nodeconv = {} - self.nodeconv[str] = lambda ob: text(ob, "utf-8") - self.nodeconv[unicode] = text + self.nodeconv[bytes] = lambda ob: text(ob, "utf-8") + self.nodeconv[str] = text self.nodeconv[int] = text - self.nodeconv[long] = text self.nodeconv[float] = text def nodefrom(self, ob): @@ -50,6 +55,12 @@ class context(object): return self.nodeconv[type(ob)](ob) raise Exception("No node conversion known for %s objects" % str(type(ob))) + def addchild(self, node, child): + node.children.append(self.nodefrom(child)) + + def addattr(self, node, k, v): + node.attrs[str(k)] = str(v) + class constructor(object): def __init__(self, ns, elcls = element, ctx=None): self._ns = ns