X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Fsp%2Fcons.py;h=1e3d678c0cec11e87599fb3677ee9111fb8445d9;hb=50be6d543e7a6067b631930a025bfdaf53fedbc4;hp=9df4725ede9220ffbf0695c5be887747ce466008;hpb=f3464a4a95d776ac14e32f8203340a9ae08795e5;p=wrw.git diff --git a/wrw/sp/cons.py b/wrw/sp/cons.py index 9df4725..1e3d678 100644 --- a/wrw/sp/cons.py +++ b/wrw/sp/cons.py @@ -1,9 +1,8 @@ +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, unicode): def __todom__(self, doc): @@ -23,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.iteritems(): - self.attrs[unicode(k)] = unicode(v) + self.ctx.addattr(self, k, v) return self def __todom__(self, doc): @@ -36,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[str] = lambda ob: text(ob, "utf-8") + self.nodeconv[str] = lambda ob: text(ob, self.charset) self.nodeconv[unicode] = text self.nodeconv[int] = text self.nodeconv[long] = text @@ -54,8 +59,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[unicode(k)] = unicode(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()