OCA reporting engine fork for dev and update.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

126 lines
4.7 KiB

# -*- coding: utf-8 -*-
##############################################################################
#
# Author: Yannick Vaucher
# Copyright 2013 Camptocamp SA
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
import time
import base64
from PyPDF2 import PdfFileReader, PdfFileWriter
from StringIO import StringIO
from openerp.netsvc import ExportService
from openerp.report import report_sxw
from openerp import pooler
_POLLING_DELAY = 0.25
def assemble_pdf(pdf_list):
"""
Assemble a list of pdf
"""
# Even though we are using PyPDF2 we can't use PdfFileMerger
# as this issue still exists in mostly used wktohtml reports version
# http://code.google.com/p/wkhtmltopdf/issues/detail?id=635
#merger = PdfFileMerger()
#merger.append(fileobj=StringIO(invoice_pdf))
#merger.append(fileobj=StringIO(bvr_pdf))
#with tempfile.TemporaryFile() as merged_pdf:
#merger.write(merged_pdf)
#return merged_pdf.read(), 'pdf'
output = PdfFileWriter()
for pdf in pdf_list:
reader = PdfFileReader(StringIO(pdf))
for page in range(reader.getNumPages()):
output.addPage(reader.getPage(page))
s = StringIO()
output.write(s)
return s.getvalue()
class PDFReportAssembler(report_sxw.report_sxw):
""" PDFReportAssembler allows to put 2 pdf reports in one single pdf"""
def _generate_all_pdf(self, cr, uid, ids, data, report_ids, context=None):
"""
Return a list of pdf encoded in base64
"""
pool = pooler.get_pool(cr.dbname)
report_obj = pool.get('ir.actions.report.xml')
spool = ExportService._services['report']
pdf_reports = []
report_list = report_obj.browse(cr, uid, report_ids, context=context)
for report in report_list:
report_key = spool.exp_report(cr.dbname, uid, report.report_name,
ids, datas=data, context=context)
while 1:
res = spool.exp_report_get(cr.dbname, uid, report_key)
if res.get('state'):
break
time.sleep(_POLLING_DELAY)
pdf = base64.b64decode(res.get('result'))
pdf_reports.append(pdf)
return pdf_reports
def _get_report_ids(self, cr, uid, ids, context=None):
"""
Hook to define the list of report to print
"""
return []
def create_single_pdf(self, cr, uid, ids, data, report_xml, context=None):
"""Call both report to assemble both pdf"""
report_ids = self._get_report_ids(cr, uid, ids, context=context)
pdf_reports = self._generate_all_pdf(cr, uid, ids, data, report_ids, context=context)
pdf_assemblage = assemble_pdf(pdf_reports)
return pdf_assemblage, 'pdf'
def create(self, cr, uid, ids, data, context=None):
"""We override the create function in order to handle generator
Code taken from report openoffice. Thanks guys :) """
pool = pooler.get_pool(cr.dbname)
ir_obj = pool.get('ir.actions.report.xml')
report_xml_ids = ir_obj.search(cr, uid,
[('report_name', '=', self.name[7:])], context=context)
if report_xml_ids:
report_xml = ir_obj.browse(cr,
uid,
report_xml_ids[0],
context=context)
report_xml.report_rml = None
report_xml.report_rml_content = None
report_xml.report_sxw_content_data = None
report_xml.report_sxw_content = None
report_xml.report_sxw = None
else:
return super(PDFReportAssembler, self).create(cr, uid, ids, data, context)
if report_xml.report_type != 'assemblage':
return super(PDFReportAssembler, self).create(cr, uid, ids, data, context)
result = self.create_source_pdf(cr, uid, ids, data, report_xml, context)
if not result:
return (False, False)
return result