Enable iterables to be passed as new SP nodes.
[wrw.git] / wrw / sp / cons.py
1 import sys, collections.abc
2 import xml.dom.minidom
3
4 class node(object):
5     pass
6
7 class text(node, str):
8     def __todom__(self, doc):
9         return doc.createTextNode(self)
10
11 class raw(node, str):
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 = str(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.items():
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.items():
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[bytes] = lambda ob: text(ob, self.charset)
48         self.nodeconv[str] = text
49         self.nodeconv[int] = text
50         self.nodeconv[float] = text
51
52     def nodefrom(self, ob):
53         if isinstance(ob, node):
54             return ob
55         if hasattr(ob, "__tonode__"):
56             return ob.__tonode__()
57         if type(ob) in self.nodeconv:
58             return self.nodeconv[type(ob)](ob)
59         return None
60
61     def addchild(self, node, child):
62         if child is None:
63             return
64         new = self.nodefrom(child)
65         if new is not None:
66             node.children.append(self.nodefrom(child))
67         elif isinstance(child, collections.abc.Iterable):
68             for ch in child:
69                 self.addchild(node, ch)
70         else:
71             raise Exception("No node conversion known for %s objects" % str(type(ob)))
72
73     def addattr(self, node, k, v):
74         if v is not None:
75             node.attrs[str(k)] = str(v)
76
77 class constructor(object):
78     def __init__(self, ns, elcls=element, ctx=None):
79         self._ns = ns
80         self._elcls = elcls
81         if ctx is None: ctx = context()
82         self._ctx = ctx
83
84     def __getattr__(self, name):
85         return self._elcls(self._ns, name, self._ctx)
86
87 class doctype(node):
88     def __init__(self, rootname, pubid, dtdid):
89         self.rootname = rootname
90         self.pubid = pubid
91         self.dtdid = dtdid