9 from Cheetah.Template import Template
10 from pkg_resources import resource_filename, resource_listdir
12 import treecutter.constants as const
13 from treecutter.tools import mkdir_p
16 """Class representing a version of a webpage"""
17 def __init__(self,link,page):
25 self._rendered_article = None
31 return set(self._resources)
36 def set_article(self,art):
37 self._rendered_article = art
40 self._doc = etree.parse(self._file)
41 t = self._doc.xpath(u'/db:article/db:info/db:title',namespaces=const.XPATH)
43 self._title = unicode(t[0].text)
44 ta = self._doc.xpath(u'/db:article/db:info/db:titleabbrev',namespaces=const.XPATH)
46 self._menu = unicode(ta[0].text)
48 dirname = os.path.dirname(self._file)
50 code = self._doc.xpath(u"//xi:include[@parse='text']",namespaces=const.XPATH)
54 alang = c.get('accept-language')
55 xpointer = c.get('xpointer')
56 (p, ext) = os.path.splitext(href)
57 if ext in const.valid_scripts:
59 script = os.path.join(os.path.abspath(dirname)+'/'+href)
60 if os.path.isfile(script):
63 if href in resource_listdir('xinclude', ''):
64 script = resource_filename('xinclude', href)
67 print "Script "+href+" in "+self._file+" missing"
69 exe.append("lang="+alang)
71 exe.append("xptr="+xpointer)
72 print " executing %15s" % (href),
75 xml = subprocess.Popen(exe,stdout=subprocess.PIPE,
76 stderr=subprocess.PIPE)
77 (stdout, stderr) = xml.communicate()
79 print " ".join(exe)+" ERROR : [ "+stderr+" ]"
82 print " [%5.2f s] (%s)" % (round(te-ts,2),xpointer)
83 xstr = etree.fromstring(stdout)
84 # inserting the generated code and remove the xinclude reference
86 idp.insert(idp.index(c)+1,xstr)
89 for r in self._doc.xpath(u"//db:link[@xlink:href]",namespaces=const.XPATH):
90 rf = os.path.join(dirname,r.get(const.XLINK+'href'))
91 if os.path.isfile(rf):
92 if r.get('security')=='encrypt':
93 with open(rf, 'rb') as f:
95 status = gpg.encrypt_file(
96 f, None, passphrase=getpass.getpass(rf+' password:'), symmetric=True,
98 r.set(const.XLINK+'href', r.get(const.XLINK+'href')+'.gpg')
100 self._resources.append(rf)
101 for i in self._doc.xpath(u"//db:imagedata[@fileref]",namespaces=const.XPATH):
102 im = os.path.join(dirname,i.get('fileref'))
103 if os.path.isfile(im):
104 self._resources.append(im)
106 print "WARNING: File "+im+" is missing!"
107 for i in self._doc.xpath(u"//html:form[@action]",namespaces=const.XPATH):
108 pyscript = re.split('\.py',i.get('action'),1)[0]+'.py'
109 im = os.path.join(dirname,pyscript)
110 if os.path.isfile(im):
111 self._resources.append(im)
113 def render(self, style):
115 # xslt_root = etree.XML(open(style+"docbook.xsl", 'r').read())
116 # transform = etree.XSLT(xslt_root)
117 # result = etree.tostring(transform(xml_root))
120 dirname = os.path.dirname(self._file)
122 infile = os.path.basename(tempfile.mktemp())
123 outfile = tempfile.mktemp()
124 tfi = open(infile,'w')
125 tfi.write(etree.tostring(self._doc,encoding='UTF-8',pretty_print=False))
127 # cmd = ["saxon-xslt-xinclude","-o",outfile,infile,style_xslt]
128 cmd = ["xsltproc","--xinclude","--output",outfile,style+"docbook.xsl",infile]
129 retcode = subprocess.call(cmd)
131 print 'Error: '+' '.join(cmd)+' Returncode ['+str(retcode)+']'
132 tfo = open(outfile,'r')
133 self._rendered_article = tfo.read()
139 def template(self,sitemap,style,tdir):
140 htmlmenu = sitemap.gen_menu(self._lang,None,"menu")
141 levelmenu = sitemap.gen_menu(self._lang,self,"tree")
142 langmenu = sitemap.lang_menu(self._lang,self._link)
143 template = Template(file=style+'index.'+self._lang+'.html.tmpl',
144 searchList=[{'title':self._title},
146 {'article':self._rendered_article},
147 {'levelmenu':levelmenu},
148 {'langmenu':langmenu}])
149 outfile = tdir+'html'.join(self._file.rsplit('xml',1))
150 mkdir_p(os.path.dirname(outfile))
151 out = open(outfile, 'w')
152 out.write(str(template))