X-Git-Url: https://source.tree.se/git?p=treecutter.git;a=blobdiff_plain;f=src%2Ftree-cutter.py;h=bc4ae7e0db92400f8f33019e7d0f09e33d0b1a19;hp=28b8669ed8efe115609d2979d11703952146b6d4;hb=9da51ba1b5901376d69406559002238877941fca;hpb=7cb8343499814446512fd4133bdfc206a6617614 diff --git a/src/tree-cutter.py b/src/tree-cutter.py index 28b8669..bc4ae7e 100755 --- a/src/tree-cutter.py +++ b/src/tree-cutter.py @@ -11,6 +11,8 @@ import argparse import shutil import pygraphviz as pgv import glob +import gettext +import shutil from amara import bindery from amara.xslt import transform from Cheetah.Template import Template @@ -23,9 +25,10 @@ parser.add_argument('--output', nargs='?', args = parser.parse_args() 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 @@ -72,7 +75,8 @@ class Directory(): class Page(): """Class representing a version of a webpage""" - def __init__(self,page): + def __init__(self,link,page): + self._link = link self._file = page[1] self._lang = page[0] self._doc = None @@ -81,12 +85,24 @@ class Page(): self._menu = None self._rendered_article = None + def language(self): + return self._lang + + def resources(self): + return set(self._resources) + + 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']") @@ -94,7 +110,7 @@ class Page(): for c in code: (p, ext) = os.path.splitext(c.href) if ext in valid_scripts: - exe = os.path.join(os.path.abspath(dirname+c.href)) + exe = os.path.join(os.path.abspath(dirname)+'/'+c.href) xml = subprocess.Popen([exe],stdout=subprocess.PIPE) xstr = bindery.parse(str(xml.stdout.read())) idp = c.xml_index_on_parent @@ -120,7 +136,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 +151,16 @@ class Page(): os.chdir(cwd) def template(self,sitemap): - htmlmenu = sitemap.gen_menu(self._lang,None) - levelmenu = sitemap.gen_menu(self._lang,self) - template = Template(file=style_tmpl, + 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=args.style+'index.'+self._lang+'.html.tmpl', searchList=[{'title':self._title}, {'menu':htmlmenu}, {'article':self._rendered_article}, {'levelmenu':levelmenu}, - {'levelname':levelname}]) - outfile = tmptarget+self._file+'.'+self._lang+'.html' + {'langmenu':langmenu}]) + outfile = tmptarget+'html'.join(self._file.rsplit('xml',1)) mkdir_p(os.path.dirname(outfile)) out = open(outfile, 'w') out.write(str(template)) @@ -161,7 +178,10 @@ class Link(): path = path+'index' lang = self._scan_languages(path) for l in lang: - self._pages.append(Page(l)) + self._pages.append(Page(self,l)) + + def add_page(self,l): + self._pages.append(Page(self,l)) def _scan_languages(self,path): lang = [] @@ -174,6 +194,37 @@ 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 + return None + + def resources(self): + res = set() + for page in self._pages: + res = res.union(page.resources()) + return res + + class Node(): def __init__(self,token,value): self._token = token @@ -199,8 +250,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 @@ -233,16 +284,48 @@ class Trie(): # G.draw('g.png') # print G.string() + def _menu(self, trie, lang, page, css): + html = "