X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Fsp%2Fcons.py;h=99656a3b14ea5170a358dfcb0dee142980f3c1d7;hb=dc590bea957bbffa808e6cf3f516d4ae9beda98a;hp=9e81677295a108458f9c9bc0fc8122c93abfe3d5;hpb=625517699b4576f727b9cd81eb7053a0bc31e9b9;p=wrw.git diff --git a/wrw/sp/cons.py b/wrw/sp/cons.py index 9e81677..99656a3 100644 --- a/wrw/sp/cons.py +++ b/wrw/sp/cons.py @@ -1,14 +1,17 @@ +import sys import xml.dom.minidom class node(object): - def __str__(self): - doc = xml.dom.minidom.Document() - return self.__todom__(doc).toxml() + pass 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 @@ -19,9 +22,9 @@ class element(node): def __call__(self, *children, **attrs): for child in children: - self.children.append(self.ctx.nodefrom(child)) + self.ctx.addchild(self, child) for k, v in attrs.items(): - self.attrs[str(k)] = str(v) + self.ctx.addattr(self, k, v) return self def __todom__(self, doc): @@ -32,10 +35,16 @@ class element(node): el.appendChild(child.__todom__(doc)) return el + def __str__(self): + doc = xml.dom.minidom.Document() + return self.__todom__(doc).toxml() + class context(object): + charset = (sys.getfilesystemencoding() or "ascii") + def __init__(self): self.nodeconv = {} - self.nodeconv[bytes] = lambda ob: text(ob, "utf-8") + self.nodeconv[bytes] = lambda ob: text(ob, self.charset) self.nodeconv[str] = text self.nodeconv[int] = text self.nodeconv[float] = text @@ -49,8 +58,15 @@ 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): + if v is not None: + node.attrs[str(k)] = str(v) + class constructor(object): - def __init__(self, ns, elcls = element, ctx=None): + def __init__(self, ns, elcls=element, ctx=None): self._ns = ns self._elcls = elcls if ctx is None: ctx = context()