X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Fsp%2Fcons.py;h=bc9bfa2b23188831a2c3157de1683e5ff7ee8551;hb=fb5f9f27dcabe7fac75e3a69f820c42779539239;hp=5341b20f44dc6495c91662bb7238810ef1851d42;hpb=ff79cdbf7d4d95e6a84d6b002e28e6df86847954;p=wrw.git

diff --git a/wrw/sp/cons.py b/wrw/sp/cons.py
index 5341b20..bc9bfa2 100644
--- a/wrw/sp/cons.py
+++ b/wrw/sp/cons.py
@@ -1,14 +1,17 @@
+import sys
 import xml.dom.minidom
 
 class node(object):
-    def __str__(self):
-        doc = xml.dom.minidom.Document()
-        return self.__todom__(doc).toxml()
+    pass
 
 class text(node, unicode):
     def __todom__(self, doc):
         return doc.createTextNode(self)
 
+class raw(node, unicode):
+    def __todom__(self, doc):
+        raise Exception("Cannot convert raw code to DOM objects")
+
 class element(node):
     def __init__(self, ns, name, ctx):
         self.ns = ns
@@ -19,9 +22,9 @@ class element(node):
 
     def __call__(self, *children, **attrs):
         for child in children:
-            self.children.append(self.ctx.nodefrom(child))
+            self.ctx.addchild(self, child)
         for k, v in attrs.iteritems():
-            self.attrs[unicode(k)] = unicode(v)
+            self.ctx.addattr(self, k, v)
         return self
 
     def __todom__(self, doc):
@@ -32,10 +35,16 @@ class element(node):
             el.appendChild(child.__todom__(doc))
         return el
 
+    def __str__(self):
+        doc = xml.dom.minidom.Document()
+        return self.__todom__(doc).toxml()
+
 class context(object):
+    charset = (sys.getfilesystemencoding() or "ascii")
+
     def __init__(self):
         self.nodeconv = {}
-        self.nodeconv[str] = lambda ob: text(ob, "utf-8")
+        self.nodeconv[str] = lambda ob: text(ob, self.charset)
         self.nodeconv[unicode] = text
         self.nodeconv[int] = text
         self.nodeconv[long] = text
@@ -50,8 +59,17 @@ class context(object):
             return self.nodeconv[type(ob)](ob)
         raise Exception("No node conversion known for %s objects" % str(type(ob)))
 
+    def addchild(self, node, child):
+        if child is None:
+            return
+        node.children.append(self.nodefrom(child))
+
+    def addattr(self, node, k, v):
+        if v is not None:
+            node.attrs[unicode(k)] = unicode(v)
+
 class constructor(object):
-    def __init__(self, ns, elcls = element, ctx=None):
+    def __init__(self, ns, elcls=element, ctx=None):
         self._ns = ns
         self._elcls = elcls
         if ctx is None: ctx = context()
@@ -59,3 +77,9 @@ class constructor(object):
 
     def __getattr__(self, name):
         return self._elcls(self._ns, name, self._ctx)
+
+class doctype(node):
+    def __init__(self, rootname, pubid, dtdid):
+        self.rootname = rootname
+        self.pubid = pubid
+        self.dtdid = dtdid