Added support for mapping around reserved keywords in SP attribute assignment.
[wrw.git] / wrw / sp / cons.py
1 import xml.dom.minidom
2
3 class node(object):
4     def __str__(self):
5         doc = xml.dom.minidom.Document()
6         return self.__todom__(doc).toxml()
7
8 class text(node, unicode):
9     def __todom__(self, doc):
10         return doc.createTextNode(self)
11
12 class raw(node, unicode):
13     def __todom__(self, doc):
14         raise Exception("Cannot convert raw code to DOM objects")
15
16 class element(node):
17     def __init__(self, ns, name, ctx):
18         self.ns = ns
19         self.name = unicode(name)
20         self.ctx = ctx
21         self.attrs = {}
22         self.children = []
23
24     def __call__(self, *children, **attrs):
25         for child in children:
26             self.ctx.addchild(self, child)
27         for k, v in attrs.iteritems():
28             self.ctx.addattr(self, k, v)
29         return self
30
31     def __todom__(self, doc):
32         el = doc.createElementNS(self.ns, self.name)
33         for k, v in self.attrs.iteritems():
34             el.setAttribute(k, v)
35         for child in self.children:
36             el.appendChild(child.__todom__(doc))
37         return el
38
39 class context(object):
40     def __init__(self):
41         self.nodeconv = {}
42         self.nodeconv[str] = lambda ob: text(ob, "utf-8")
43         self.nodeconv[unicode] = text
44         self.nodeconv[int] = text
45         self.nodeconv[long] = text
46         self.nodeconv[float] = text
47
48     def nodefrom(self, ob):
49         if isinstance(ob, node):
50             return ob
51         if hasattr(ob, "__tonode__"):
52             return ob.__tonode__()
53         if type(ob) in self.nodeconv:
54             return self.nodeconv[type(ob)](ob)
55         raise Exception("No node conversion known for %s objects" % str(type(ob)))
56
57     def addchild(self, node, child):
58         node.children.append(self.nodefrom(child))
59
60     def addattr(self, node, k, v):
61         node.attrs[unicode(k)] = unicode(v)
62
63 class constructor(object):
64     def __init__(self, ns, elcls = element, ctx=None):
65         self._ns = ns
66         self._elcls = elcls
67         if ctx is None: ctx = context()
68         self._ctx = ctx
69
70     def __getattr__(self, name):
71         return self._elcls(self._ns, name, self._ctx)