page: using lxml transform, simple docbook & image
[treecutter.git] / treecutter / page.py
index ef77bb7d861b1037f9b622dfaee418b41524776c..8843df975a45bf45204f5db7c1a3fe08b698341a 100644 (file)
@@ -2,11 +2,17 @@
 import os
 import subprocess
 import tempfile
-from amara import bindery
-from amara.xslt import transform
-from Cheetah.Template import Template
-from pkg_resources import resource_filename, resource_listdir
-import treecutter.const as const
+import re
+import getpass
+import gnupg
+import codecs
+from lxml import etree
+from lxml.builder import ElementMaker
+#from jinja2 import Template
+import jinja2
+from time import time
+import treecutter.constants as const
+from treecutter.docbook import Docbook
 from treecutter.tools import mkdir_p
 
 class Page():
@@ -34,90 +40,40 @@ class Page():
         self._rendered_article = art
 
     def prepare(self):
-        self._doc = bindery.parse(self._file, prefixes=const.PREFIXES)
-        if self._doc.xml_select(u'/db:article/db:info/db:title'):
-            self._title = unicode(self._doc.article.info.title)
-        if self._doc.xml_select(u'/db:article/db:info/db:titleabbrev'):
-            self._menu = unicode(self._doc.article.info.titleabbrev)
+        self._doc = Docbook(self._file)
+        (self._title, self._menu) = self._doc.title()
+#        self._doc.expand_imageobjects()
+        self._doc.parse_xincludes()
+        doc = self._doc.collect_links()
+        img = self._doc.collect_images()
+        form= self._doc.collect_forms()
+        self._resources = doc + img + form
 
-        dirname = os.path.dirname(self._file)
-        code  = self._doc.xml_select(u"//xi:include[@parse='text']")
-        if code:
-            for c in code:
-                (p, ext) = os.path.splitext(c.href)
-                if ext in const.valid_scripts:
-                    exe = []
-                    script = os.path.join(os.path.abspath(dirname)+'/'+c.href)
-                    if os.path.isfile(script):
-                        exe.append(script)
-                    else:
-                        if c.href in resource_listdir('xinclude', ''):
-                            script = resource_filename('xinclude', c.href)
-                            exe.append(script)
-                        else:
-                            print "Script "+c.href+" in "+self._file+" missing"
-                    if c.xml_select(u"//xi:include[@accept-language]"):
-                        alang = c.xml_attributes[None, "accept-language"]
-                        exe.append("lang="+alang)
-                    if c.xml_select(u"//xi:include[@xpointer]"):
-                        exe.append("xptr="+c.xpointer)
-                    xml = subprocess.Popen(exe,stdout=subprocess.PIPE)
-                    xmlblock = str(xml.stdout.read())
-                    xstr = bindery.parse(xmlblock)
-                    idp = c.xml_index_on_parent
-                    for x in xstr.xml_children:
-                        c.xml_parent.xml_insert(idp,x)
-                        c.xml_parent.xml_remove(c)
+    def render(self, transform):
+        self._rendered_article = self._doc.xslt(transform['xhtml5'])
 
-        for r in self._doc.xml_select(u"//db:link[@xl:href]"):
-            rf = os.path.join(dirname,r.href)
-            if os.path.isfile(rf):
-                self._resources.append(rf)
-        for i in self._doc.xml_select(u"//db:imagedata[@fileref]"):
-            im = os.path.join(dirname,i.fileref)
-            if os.path.isfile(im):
-                self._resources.append(im)
-        for i in self._doc.xml_select(u"//html:form[@action]"):
-            pyscript = re.split('\.py',i.action,1)[0]+'.py'
-            im = os.path.join(dirname,pyscript)
-            if os.path.isfile(im):
-                self._resources.append(im)
-
-    def render(self, style):
-        #  amara can not handle the docbook stylesheets
-        #  xmlarticle = transform(doc,style_xslt)
-        cwd = os.getcwd()
-        dirname = os.path.dirname(self._file)
-        os.chdir(dirname)
-        infile  = os.path.basename(tempfile.mktemp())
-        outfile = tempfile.mktemp()
-        tfi = open(infile,'w')
-        tfi.write(self._doc.xml_encode(omit_xml_declaration=True))
-        tfi.close()
-#  cmd = ["saxon-xslt-xinclude","-o",outfile,infile,style_xslt]
-        cmd = ["xsltproc","--xinclude","--output",outfile,style+"docbook.xsl",infile]
-        retcode = subprocess.call(cmd)
-        if retcode:
-            print 'Error: '+' '.join(cmd)+' Returncode ['+str(retcode)+']'
-        tfo = open(outfile,'r')
-        self._rendered_article = tfo.read()
-        tfo.close()
-        os.remove(infile)
-        os.remove(outfile)
-        os.chdir(cwd)
-
-    def template(self,sitemap,style,tdir):
+    def template(self,sitemap,style,tdir,subdir):
         htmlmenu =  sitemap.gen_menu(self._lang,None,"menu")
         levelmenu = sitemap.gen_menu(self._lang,self,"tree")
         langmenu = sitemap.lang_menu(self._lang,self._link)
-        template = Template(file=style+'index.'+self._lang+'.html.tmpl',
-                            searchList=[{'title':self._title},
-                                        {'menu':htmlmenu},
-                                        {'article':self._rendered_article},
-                                        {'levelmenu':levelmenu},
-                                        {'langmenu':langmenu}])
-        outfile = tdir+'html'.join(self._file.rsplit('xml',1))
+        article = self._rendered_article
+        templateLoader = jinja2.FileSystemLoader( searchpath="/" )
+        templateEnv = jinja2.Environment( loader=templateLoader )
+
+        templateVars = {'title':self._title,
+                        'menu':htmlmenu,
+                        'article':article,
+                        'levelmenu':levelmenu,
+                        'langmen':langmenu,
+                        'subdir':subdir}
+
+        s = unicode(style, "utf-8")
+        t = s+'index.'+self._lang+'.xhtml.tmpl'
+        template = templateEnv.get_template( t )
+        templateout = template.render( templateVars )
+
+        outfile = tdir+'xhtml'.join(self._file.rsplit('xml',1))
         mkdir_p(os.path.dirname(outfile))
-        out = open(outfile, 'w')
-        out.write(str(template))
+        out = codecs.open(outfile, 'w', 'utf-8')
+        out.write(templateout)
         out.close()