Added a HTML-generation engine that might hopefully be useful.

[wrw.git] / wrw / sp / cons.py

diff --git a/wrw/sp/cons.py b/wrw/sp/cons.py
new file mode 100644 (file)
index 0000000..5341b20
--- /dev/null
+++ b/wrw/sp/cons.py
@@ -0,0 +1,61 @@
+import xml.dom.minidom
+
+class node(object):
+    def __str__(self):
+        doc = xml.dom.minidom.Document()
+        return self.__todom__(doc).toxml()
+
+class text(node, unicode):
+    def __todom__(self, doc):
+        return doc.createTextNode(self)
+
+class element(node):
+    def __init__(self, ns, name, ctx):
+        self.ns = ns
+        self.name = unicode(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)
+        return self
+
+    def __todom__(self, doc):
+        el = doc.createElementNS(self.ns, self.name)
+        for k, v in self.attrs.iteritems():
+            el.setAttribute(k, v)
+        for child in self.children:
+            el.appendChild(child.__todom__(doc))
+        return el
+
+class context(object):
+    def __init__(self):
+        self.nodeconv = {}
+        self.nodeconv[str] = lambda ob: text(ob, "utf-8")
+        self.nodeconv[unicode] = text
+        self.nodeconv[int] = text
+        self.nodeconv[long] = text
+        self.nodeconv[float] = text
+
+    def nodefrom(self, ob):
+        if isinstance(ob, node):
+            return ob
+        if hasattr(ob, "__tonode__"):
+            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)))
+
+class constructor(object):
+    def __init__(self, ns, elcls = element, ctx=None):
+        self._ns = ns
+        self._elcls = elcls
+        if ctx is None: ctx = context()
+        self._ctx = ctx
+
+    def __getattr__(self, name):
+        return self._elcls(self._ns, name, self._ctx)