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

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

diff --git a/wrw/sp/xhtml.py b/wrw/sp/xhtml.py
new file mode 100644 (file)
index 0000000..abf4b96
--- /dev/null
+++ b/wrw/sp/xhtml.py
@@ -0,0 +1,53 @@
+import xml.dom.minidom, StringIO
+import cons as _cons
+import util
+dom = xml.dom.minidom.getDOMImplementation()
+
+ns = u"http://www.w3.org/1999/xhtml"
+doctype = u"-//W3C//DTD XHTML 1.1//EN"
+dtd = u"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd"
+
+class htmlelement(_cons.element):
+    def __todoc__(self):
+        doc = dom.createDocument(None, None, None)
+        doc.appendChild(dom.createDocumentType("html", doctype, dtd))
+        doc.appendChild(self.__todom__(doc))
+        return doc
+
+def cons(ctx=None):
+    return _cons.constructor(ns, htmlelement, ctx)
+
+def head(title=None, css=None):
+    h = cons()
+    head = h.head
+    if title:
+        head(h.title(title))
+    if isinstance(css, str) or isinstance(css, unicode):
+        head(h.link(rel="stylesheet", type="text/css", href=css))
+    elif css:
+        for ss in css:
+            head(h.link(rel="stylesheet", type="text/css", href=ss))
+    return head
+
+class htmlformatter(util.formatter):
+    allowshort = set([u"br", u"hr", u"img", u"input"])
+    def element(self, el, **extra):
+        if el.name in self.allowshort:
+            super(htmlformatter, self).element(el, **extra)
+        else:
+            self.longtag(el, **extra)
+
+class htmlindenter(util.indenter, htmlformatter):
+    pass
+
+def forreq(req, tree):
+    # XXX: Use proper Content-Type for clients accepting it.
+    req.ohead["Content-Type"] = "text/html; charset=utf-8"
+    buf = StringIO.StringIO()
+    htmlindenter.output(buf, tree, doctype=(doctype, dtd), charset="utf-8")
+    return [buf.getvalue()]
+
+def xhtmlresp(callable):
+    def wrapper(req):
+        return forreq(req, callable(req))
+    return wrapper