X-Git-Url: https://source.tree.se/git?p=treecutter.git;a=blobdiff_plain;f=src%2Ftree-cutter.py;h=37b893a0961529fcdb66b230f5bb9d68f0e83e31;hp=97ac6a2a65f15fdb7bf00c3fa09ba7c12b3f7ff2;hb=9efe89fdf3454731169d3f51ce357cf16a04bd76;hpb=b73ab8db3f73bbdcd88746151f5e0d4a3d8960a1 diff --git a/src/tree-cutter.py b/src/tree-cutter.py index 97ac6a2..37b893a 100755 --- a/src/tree-cutter.py +++ b/src/tree-cutter.py @@ -10,6 +10,7 @@ import time import argparse import shutil import pygraphviz as pgv +import glob from amara import bindery from amara.xslt import transform from Cheetah.Template import Template @@ -25,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 @@ -70,15 +73,141 @@ class Directory(): return set(self._tree) class Page(): + """Class representing a version of a webpage""" + def __init__(self,page): + self._file = page[1] + self._lang = page[0] + self._doc = None + self._resources = [] + self._title = None + 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(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) + + 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 valid_scripts: + 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 + for x in xstr.xml_children: + c.xml_parent.xml_insert(idp,x) + c.xml_parent.xml_remove(c) + + 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) + + def render(self): + # 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()) + tfi.close() +# cmd = ["saxon-xslt-xinclude","-o",outfile,infile,style_xslt] + cmd = ["xsltproc","--xinclude","--output",outfile,style_xslt,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): + 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':'Menu'}]) + outfile = tmptarget+'html'.join(self._file.rsplit('xml',1)) + mkdir_p(os.path.dirname(outfile)) + out = open(outfile, 'w') + out.write(str(template)) + out.close() + + +class Link(): """Class representing a webpage on the site""" def __init__(self,link): self._link = link - self._resources = [] - self._script = 0 + # find the representations of the link. + self._pages = [] + path = link + if self._link[-1] == '/': + path = path+'index' + lang = self._scan_languages(path) + for l in lang: + self._pages.append(Page(l)) + + def _scan_languages(self,path): + lang = [] + for l in glob.glob('.'+path+'*'): + ls = l.split('.') + if len(ls) > 3 and ls[3] == 'xml': + lang.append((ls[2],l)) + return lang 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 @@ -98,6 +227,15 @@ class Trie(): def __init__(self): self._root = [] + def __iter__(self): + return self.inorder(self._root) + + def inorder(self,t): + for l in t: + yield l.value() + for x in self.inorder(l.children()): + yield x + def _add(self,trie, key, content): # is the key a leaf k = key.pop(0) @@ -127,20 +265,40 @@ 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 = "