Reimplemented the SP output formatters to hopefully work better and faster.
[wrw.git] / wrw / sp / cons.py
1 import sys
2 import xml.dom.minidom
3
4 class node(object):
5     pass
6
7 class text(node, unicode):
8     def __todom__(self, doc):
9         return doc.createTextNode(self)
10
11 class raw(node, unicode):
12     def __todom__(self, doc):
13         raise Exception("Cannot convert raw code to DOM objects")
14
15 class element(node):
16     def __init__(self, ns, name, ctx):
17         self.ns = ns
18         self.name = unicode(name)
19         self.ctx = ctx
20         self.attrs = {}
21         self.children = []
22
23     def __call__(self, *children, **attrs):
24         for child in children:
25             self.ctx.addchild(self, child)
26         for k, v in attrs.iteritems():
27             self.ctx.addattr(self, k, v)
28         return self
29
30     def __todom__(self, doc):
31         el = doc.createElementNS(self.ns, self.name)
32         for k, v in self.attrs.iteritems():
33             el.setAttribute(k, v)
34         for child in self.children:
35             el.appendChild(child.__todom__(doc))
36         return el
37
38     def __str__(self):
39         doc = xml.dom.minidom.Document()
40         return self.__todom__(doc).toxml()
41
42 class context(object):
43     charset = (sys.getfilesystemencoding() or "ascii")
44
45     def __init__(self):
46         self.nodeconv = {}
47         self.nodeconv[str] = lambda ob: text(ob, self.charset)
48         self.nodeconv[unicode] = text
49         self.nodeconv[int] = text
50         self.nodeconv[long] = text
51         self.nodeconv[float] = text
52
53     def nodefrom(self, ob):
54         if isinstance(ob, node):
55             return ob
56         if hasattr(ob, "__tonode__"):
57             return ob.__tonode__()
58         if type(ob) in self.nodeconv:
59             return self.nodeconv[type(ob)](ob)
60         raise Exception("No node conversion known for %s objects" % str(type(ob)))
61
62     def addchild(self, node, child):
63         if child is None:
64             return
65         node.children.append(self.nodefrom(child))
66
67     def addattr(self, node, k, v):
68         if v is not None:
69             node.attrs[unicode(k)] = unicode(v)
70
71 class constructor(object):
72     def __init__(self, ns, elcls=element, ctx=None):
73         self._ns = ns
74         self._elcls = elcls
75         if ctx is None: ctx = context()
76         self._ctx = ctx
77
78     def __getattr__(self, name):
79         return self._elcls(self._ns, name, self._ctx)
80
81 class doctype(node):
82     def __init__(self, rootname, pubid, dtdid):
83         self.rootname = rootname
84         self.pubid = pubid
85         self.dtdid = dtdid