10 #from jinja2 import Template
12 from pkg_resources import resource_filename, resource_listdir
14 import treecutter.constants as const
15 from treecutter.tools import mkdir_p
18 """Class representing a version of a webpage"""
19 def __init__(self,link,page):
27 self._rendered_article = None
33 return set(self._resources)
38 def set_article(self,art):
39 self._rendered_article = art
42 self._doc = etree.parse(self._file)
43 t = self._doc.xpath(u'/db:article/db:info/db:title',namespaces=const.XPATH)
45 self._title = unicode(t[0].text)
46 ta = self._doc.xpath(u'/db:article/db:info/db:titleabbrev',namespaces=const.XPATH)
48 self._menu = unicode(ta[0].text)
50 dirname = os.path.dirname(self._file)
52 code = self._doc.xpath(u"//xi:include[@parse='text']",namespaces=const.XPATH)
56 alang = c.get('accept-language')
57 xpointer = c.get('xpointer')
58 (p, ext) = os.path.splitext(href)
59 if ext in const.valid_scripts:
61 script = os.path.join(os.path.abspath(dirname)+'/'+href)
62 if os.path.isfile(script):
65 if href in resource_listdir('xinclude', ''):
66 script = resource_filename('xinclude', href)
69 print "Script "+href+" in "+self._file+" missing"
71 exe.append("lang="+alang)
73 exe.append("xptr="+xpointer)
74 print " executing %15s" % (href),
77 xml = subprocess.Popen(exe,stdout=subprocess.PIPE,
78 stderr=subprocess.PIPE)
79 (stdout, stderr) = xml.communicate()
81 print " ".join(exe)+" ERROR : [ "+stderr+" ]"
84 print " [%5.2f s] (%s)" % (round(te-ts,2),xpointer)
85 xstr = etree.fromstring(stdout)
86 # inserting the generated code and remove the xinclude reference
88 idp.insert(idp.index(c)+1,xstr)
91 for r in self._doc.xpath(u"//db:link[@xlink:href]",namespaces=const.XPATH):
92 rf = os.path.join(dirname,r.get(const.XLINK+'href'))
93 if os.path.isfile(rf):
94 if r.get('security')=='encrypt':
95 with open(rf, 'rb') as f:
97 status = gpg.encrypt_file(
98 f, None, passphrase=getpass.getpass(rf+' password:'), symmetric=True,
100 r.set(const.XLINK+'href', r.get(const.XLINK+'href')+'.gpg')
102 self._resources.append(rf)
103 for i in self._doc.xpath(u"//db:imagedata[@fileref]",namespaces=const.XPATH):
104 im = os.path.join(dirname,i.get('fileref'))
105 if os.path.isfile(im):
106 self._resources.append(im)
108 print "WARNING: File "+im+" is missing!"
109 for i in self._doc.xpath(u"//html:form[@action]",namespaces=const.XPATH):
110 pyscript = re.split('\.py',i.get('action'),1)[0]+'.py'
111 im = os.path.join(dirname,pyscript)
112 if os.path.isfile(im):
113 self._resources.append(im)
115 def render(self, style):
117 # xslt_root = etree.XML(open(style+"docbook.xsl", 'r').read())
118 # transform = etree.XSLT(xslt_root)
119 # result = etree.tostring(transform(xml_root))
122 dirname = os.path.dirname(self._file)
124 infile = os.path.basename(tempfile.mktemp())
125 outfile = tempfile.mktemp()
126 tfi = open(infile,'w')
127 tfi.write(etree.tostring(self._doc,encoding='UTF-8',pretty_print=False))
129 # cmd = ["saxon-xslt-xinclude","-o",outfile,infile,style_xslt]
130 cmd = ["xsltproc","--xinclude","--output",outfile,style+"docbook.xsl",infile]
131 retcode = subprocess.call(cmd)
133 print 'Error: '+' '.join(cmd)+' Returncode ['+str(retcode)+']'
134 tfo = codecs.open(outfile, "r", "utf-8")
135 self._rendered_article = tfo.read()
141 def template(self,sitemap,style,tdir):
142 htmlmenu = sitemap.gen_menu(self._lang,None,"menu")
143 levelmenu = sitemap.gen_menu(self._lang,self,"tree")
144 langmenu = sitemap.lang_menu(self._lang,self._link)
145 article = self._rendered_article
146 templateLoader = jinja2.FileSystemLoader( searchpath="/" )
147 templateEnv = jinja2.Environment( loader=templateLoader )
149 templateVars = {'title':self._title,
152 'levelmenu':levelmenu,
155 s = unicode(style, "utf-8")
156 t = s+'index.'+self._lang+'.xhtml.tmpl'
157 template = templateEnv.get_template( t )
158 templateout = template.render( templateVars )
160 outfile = tdir+'xhtml'.join(self._file.rsplit('xml',1))
161 mkdir_p(os.path.dirname(outfile))
162 out = codecs.open(outfile, 'w', 'utf-8')
163 out.write(templateout)