Techsystech
/
account-financial-reporting
mirror of https://github.com/OCA/account-financial-reporting.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Adapt account_financial_report_webkit to fixes in addons 

https://github.com/odoo/odoo/pull/55
https://github.com/odoo/odoo/pull/56

Plus few improvement following pep8 to remove useless variable and import
pull/7/merge
Yannick Vaucher 2014-06-17 16:01:19 +02:00 committed by Guewen Baconnier
parent dd6cd86fcd 8be3b386b5
commit 5496a3fa0f
1 changed files with 52 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
account_financial_report_webkit/report/webkit_parser_header_fix.py
View File

@ -30,8 +30,8 @@
import os import os
import subprocess import subprocess
import tempfile import tempfile
import time
import logging import logging
from functools import partial
from mako import exceptions from mako import exceptions
@ -48,11 +48,12 @@ _logger = logging.getLogger('financial.reports.webkit')
# Class used only as a workaround to bug: # Class used only as a workaround to bug:
# http://code.google.com/p/wkhtmltopdf/issues/detail?id=656 # http://code.google.com/p/wkhtmltopdf/issues/detail?id=656
# html headers and footers do not work on big files (hundreds of pages) so we replace them by # html headers and footers do not work on big files (hundreds of pages) so we
# text headers and footers passed as arguments to wkhtmltopdf # replace them by text headers and footers passed as arguments to wkhtmltopdf
# this class has to be removed once the bug is fixed # this class has to be removed once the bug is fixed
# in your report class, to print headers and footers as text, you have to add them in the localcontext with a key 'additional_args' # in your report class, to print headers and footers as text, you have to add
# them in the localcontext with a key 'additional_args'
# for instance: # for instance:
# header_report_name = _('PARTNER LEDGER') # header_report_name = _('PARTNER LEDGER')
# footer_date_time = self.formatLang(str(datetime.today()), date_time=True) # footer_date_time = self.formatLang(str(datetime.today()), date_time=True)
@ -75,6 +76,7 @@ _logger = logging.getLogger('financial.reports.webkit')
from mako.template import Template from mako.template import Template
from mako.lookup import TemplateLookup from mako.lookup import TemplateLookup
def mako_template(text): def mako_template(text):
"""Build a Mako template. """Build a Mako template.
@ -86,12 +88,13 @@ def mako_template(text):
class HeaderFooterTextWebKitParser(webkit_report.WebKitParser): class HeaderFooterTextWebKitParser(webkit_report.WebKitParser):
def generate_pdf(self, comm_path, report_xml, header, footer, html_list, webkit_header=False): def generate_pdf(self, comm_path, report_xml, header, footer, html_list,
webkit_header=False, parser_instance=False):
"""Call webkit in order to generate pdf""" """Call webkit in order to generate pdf"""
if not webkit_header: if not webkit_header:
webkit_header = report_xml.webkit_header webkit_header = report_xml.webkit_header
tmp_dir = tempfile.gettempdir() fd, out_filename = tempfile.mkstemp(suffix=".pdf",
out_filename = tempfile.mktemp(suffix=".pdf", prefix="webkit.tmp.") prefix="webkit.tmp.")
file_to_del = [out_filename] file_to_del = [out_filename]
if comm_path: if comm_path:
command = [comm_path] command = [comm_path]
@ -115,16 +118,16 @@ class HeaderFooterTextWebKitParser(webkit_report.WebKitParser):
if webkit_header.format: if webkit_header.format:
command.extend(['--page-size', str(webkit_header.format).replace(',', '.')]) command.extend(['--page-size', str(webkit_header.format).replace(',', '.')])
if self.parser_instance.localcontext.get('additional_args', False): if parser_instance.localcontext.get('additional_args', False):
for arg in self.parser_instance.localcontext['additional_args']: for arg in parser_instance.localcontext['additional_args']:
command.extend(arg) command.extend(arg)
count = 0 count = 0
for html in html_list: for html in html_list:
html_file = file(os.path.join(tmp_dir, str(time.time()) + str(count) +'.body.html'), 'w') with tempfile.NamedTemporaryFile(suffix="%d.body.html" % count,
delete=False) as html_file:
count += 1 count += 1
html_file.write(html) html_file.write(self._sanitize_html(html))
html_file.close()
file_to_del.append(html_file.name) file_to_del.append(html_file.name)
command.append(html_file.name) command.append(html_file.name)
command.append(out_filename) command.append(out_filename)
@ -144,9 +147,9 @@ class HeaderFooterTextWebKitParser(webkit_report.WebKitParser):
if status: if status:
raise except_osv(_('Webkit error'), raise except_osv(_('Webkit error'),
_("The command 'wkhtmltopdf' failed with error code = %s. Message: %s") % (status, error_message)) _("The command 'wkhtmltopdf' failed with error code = %s. Message: %s") % (status, error_message))
pdf_file = open(out_filename, 'rb') with open(out_filename, 'rb') as pdf_file:
pdf = pdf_file.read() pdf = pdf_file.read()
pdf_file.close() os.close(fd)
finally: finally:
if stderr_fd is not None: if stderr_fd is not None:
os.close(stderr_fd) os.close(stderr_fd)
@ -160,20 +163,23 @@ class HeaderFooterTextWebKitParser(webkit_report.WebKitParser):
# override needed to keep the attachments' storing procedure # override needed to keep the attachments' storing procedure
def create_single_pdf(self, cursor, uid, ids, data, report_xml, context=None): def create_single_pdf(self, cursor, uid, ids, data, report_xml, context=None):
"""generate the PDF""" """generate the PDF"""
if context is None: if context is None:
context = {} context = {}
htmls = [] htmls = []
if report_xml.report_type != 'webkit': if report_xml.report_type != 'webkit':
return super(HeaderFooterTextWebKitParser,self).create_single_pdf(cursor, uid, ids, data, report_xml, context=context) return super(HeaderFooterTextWebKitParser, self
).create_single_pdf(cursor, uid, ids, data,
report_xml, context=context)
self.parser_instance = self.parser(cursor, parser_instance = self.parser(cursor,
uid, uid,
self.name2, self.name2,
context=context) context=context)
self.pool = pooler.get_pool(cursor.dbname) self.pool = pooler.get_pool(cursor.dbname)
objs = self.getObjects(cursor, uid, ids, context) objs = self.getObjects(cursor, uid, ids, context)
self.parser_instance.set_context(objs, data, ids, report_xml.report_type) parser_instance.set_context(objs, data, ids, report_xml.report_type)
template = False template = False
@ -186,31 +192,31 @@ class HeaderFooterTextWebKitParser(webkit_report.WebKitParser):
if not template: if not template:
raise except_osv(_('Error!'), _('Webkit Report template not found !')) raise except_osv(_('Error!'), _('Webkit Report template not found !'))
header = report_xml.webkit_header.html header = report_xml.webkit_header.html
footer = report_xml.webkit_header.footer_html
if not header and report_xml.header: if not header and report_xml.header:
raise except_osv( raise except_osv(
_('No header defined for this Webkit report!'), _('No header defined for this Webkit report!'),
_('Please set a header in company settings') _('Please set a header in company settings.')
) )
css = report_xml.webkit_header.css css = report_xml.webkit_header.css
if not css: if not css:
css = '' css = ''
user = self.pool.get('res.users').browse(cursor, uid, uid)
translate_call = partial(self.translate_call, parser_instance)
#default_filters=['unicode', 'entity'] can be used to set global filter #default_filters=['unicode', 'entity'] can be used to set global filter
body_mako_tpl = mako_template(template) body_mako_tpl = mako_template(template)
helper = WebKitHelper(cursor, uid, report_xml.id, context) helper = WebKitHelper(cursor, uid, report_xml.id, context)
if report_xml.precise_mode: if report_xml.precise_mode:
for obj in objs: for obj in objs:
self.parser_instance.localcontext['objects'] = [obj] parser_instance.localcontext['objects'] = [obj]
try: try:
html = body_mako_tpl.render(helper=helper, html = body_mako_tpl.render(helper=helper,
css=css, css=css,
_=self.translate_call, _=translate_call,
**self.parser_instance.localcontext) **parser_instance.localcontext)
htmls.append(html) htmls.append(html)
except Exception, e: except Exception:
msg = exceptions.text_error_template().render() msg = exceptions.text_error_template().render()
_logger.error(msg) _logger.error(msg)
raise except_osv(_('Webkit render'), msg) raise except_osv(_('Webkit render'), msg)
@ -218,15 +224,16 @@ class HeaderFooterTextWebKitParser(webkit_report.WebKitParser):
try: try:
html = body_mako_tpl.render(helper=helper, html = body_mako_tpl.render(helper=helper,
css=css, css=css,
_=self.translate_call, _=translate_call,
**self.parser_instance.localcontext) **parser_instance.localcontext)
htmls.append(html) htmls.append(html)
except Exception, e: except Exception:
msg = exceptions.text_error_template().render() msg = exceptions.text_error_template().render()
_logger.error(msg) _logger.error(msg)
raise except_osv(_('Webkit render'), msg) raise except_osv(_('Webkit render'), msg)
# NO html footer and header because we write them as text with wkhtmltopdf # NO html footer and header because we write them as text with
# wkhtmltopdf
head = foot = False head = foot = False
if report_xml.webkit_debug: if report_xml.webkit_debug:
@ -234,13 +241,14 @@ class HeaderFooterTextWebKitParser(webkit_report.WebKitParser):
deb = body_mako_tpl.render(helper=helper, deb = body_mako_tpl.render(helper=helper,
css=css, css=css,
_debug=tools.ustr("\n".join(htmls)), _debug=tools.ustr("\n".join(htmls)),
_=self.translate_call, _=translate_call,
**self.parser_instance.localcontext) **parser_instance.localcontext)
except Exception, e: except Exception:
msg = exceptions.text_error_template().render() msg = exceptions.text_error_template().render()
_logger.error(msg) _logger.error(msg)
raise except_osv(_('Webkit render'), msg) raise except_osv(_('Webkit render'), msg)
return (deb, 'html') return (deb, 'html')
bin = self.get_lib(cursor, uid) bin = self.get_lib(cursor, uid)
pdf = self.generate_pdf(bin, report_xml, head, foot, htmls) pdf = self.generate_pdf(bin, report_xml, head, foot, htmls,
parser_instance=parser_instance)
return (pdf, 'pdf') return (pdf, 'pdf')