Implementing generation of a menu, currently only the full menu with and without...

[treecutter.git] / src / tree-cutter.py

diff --git a/src/tree-cutter.py b/src/tree-cutter.py
index 0b72e6b5a0d7e36b0f5295b704e49c2b917cc92f..b19953480b3e3339fb835e2d9bf932382bd6bbac 100755 (executable)
--- a/src/tree-cutter.py
+++ b/src/tree-cutter.py
@@ -26,6 +26,8 @@ style_xslt = args.style+"docbook.xsl"
 style_tmpl = args.style+"index.en.html.tmpl"
 outputdir = args.output
 
+tmptarget = tempfile.mkdtemp()+'/'
+
 valid_scripts = ['.py','.pl']
 MAXLEVEL = 10000
 
@@ -81,12 +83,21 @@ class Page():
         self._menu = None
         self._rendered_article = None
 
+    def language(self):
+        return self._lang
+
+    def menu(self):
+        return self._menu
+
+    def set_article(self,art):
+        self._rendered_article = art
+
     def prepare(self):
         self._doc = bindery.parse(self._file, prefixes=PREFIXES)
         if self._doc.xml_select(u'/db:article/db:info/db:title'):
-            self._title = unicode(doc.article.info.title)
+            self._title = unicode(self._doc.article.info.title)
         if self._doc.xml_select(u'/db:article/db:info/db:titleabbrev'):
-            self._menu = unicode(doc.article.info.titleabbrev)
+            self._menu = unicode(self._doc.article.info.titleabbrev)
 
         dirname = os.path.dirname(self._file)
         code  = self._doc.xml_select(u"//xi:include[@parse='text']")
@@ -120,7 +131,7 @@ class Page():
         infile  = os.path.basename(tempfile.mktemp())
         outfile = tempfile.mktemp()
         tfi = open(infile,'w')
-        tfi.write(doc.xml_encode())
+        tfi.write(self._doc.xml_encode())
         tfi.close()
 #  cmd = ["saxon-xslt-xinclude","-o",outfile,infile,style_xslt]
         cmd = ["xsltproc","--xinclude","--output",outfile,style_xslt,infile]
@@ -135,15 +146,15 @@ class Page():
         os.chdir(cwd)
 
     def template(self,sitemap):
-        htmlmenu =  sitemap.gen_menu(self._lang,None)
-        levelmenu = sitemap.gen_menu(self._lang,self)
+        htmlmenu =  sitemap.gen_menu(self._lang,None,None)
+        levelmenu = sitemap.gen_menu(self._lang,self,"tree")
         template = Template(file=style_tmpl,
                             searchList=[{'title':self._title},
                                         {'menu':htmlmenu},
                                         {'article':self._rendered_article},
                                         {'levelmenu':levelmenu},
-                                        {'levelname':levelname}])
-        outfile = tmptarget+self._file+'.'+self._lang+'.html'
+                                        {'levelname':'Menu'}])
+        outfile = tmptarget+'html'.join(self._file.rsplit('xml',1))
         mkdir_p(os.path.dirname(outfile))
         out = open(outfile, 'w')
         out.write(str(template))
@@ -174,6 +185,29 @@ class Link():
     def link(self):
         return self._link
 
+    def prepare(self):
+        for page in self._pages:
+            page.prepare()
+
+    def languages(self):
+        p = []
+        for page in self._pages:
+            p.append(page.language())
+        return p
+
+    def render(self):
+        for page in self._pages:
+            page.render()
+
+    def template(self,sitemap):
+        for page in self._pages:
+            page.template(sitemap)
+
+    def page(self,lang):
+        for page in self._pages:
+            if page.language()==lang:
+                return page
+
 class Node():
     def __init__(self,token,value):
         self._token = token
@@ -199,8 +233,8 @@ class Trie():
     def inorder(self,t):
         for l in t:
             yield l.value()
-            for ch in l.children():
-                self.inorder(ch)
+            for x in self.inorder(l.children()):
+                yield x
 
     def _add(self,trie, key, content):
         # is the key a leaf
@@ -231,7 +265,25 @@ class Trie():
         self._graph(self._root, G)
 #        G.layout('dot')
 #        G.draw('g.png')
-        print G.string()
+#        print G.string()
+
+    def _menu(self, trie, lang, page, css):
+        html = "<ul%s>\n" % css
+        for l in trie:
+            sel = ''
+            if l.value().page(lang) == page:
+                sel = ' class="selected"'
+            html += '<li%s><a href="%s">%s</a>\n' \
+            % (sel,l.value().link(),l.value().page(lang).menu())
+            html += self._menu(l.children(), lang, page, "")
+        html += "</ul>\n"
+        return html
+
+    def menu(self,lang,page,cssclass):
+        css = ''
+        if cssclass:
+            css = ' class="'+cssclass+'"'
+        return self._menu(self._root, lang, page, css)
 
 class Sitemap():
     """Class keeping the internal site structure"""