reporting-engine/report_py3o_fusion_server/models/py3o_report.py

# -*- coding: utf-8 -*-
# © 2013 XCG Consulting <http://odoo.consulting>
# © 2016 ACSONE SA/NV
# © 2017 Therp BV <http://therp.nl>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
import json
import logging
import os
import requests
import tempfile
from contextlib import closing
from openerp import _, api, models
from openerp.exceptions import UserError
from StringIO import StringIO

logger = logging.getLogger(__name__)

try:
    from py3o.template import Template
    from py3o.template.helpers import Py3oConvertor
except ImportError:
    logger.debug('Cannot import py3o.template')


class Py3oReport(models.TransientModel):
    _inherit = 'py3o.report'

    @api.multi
    def _create_single_report(self, model_instance, data, save_in_attachment):
        """ This function to generate our py3o report
        """
        self.ensure_one()
        report_xml = self.ir_actions_report_xml_id
        filetype = report_xml.py3o_filetype
        if report_xml.py3o_is_local_fusion:
            result_path = super(Py3oReport, self)._create_single_report(
                model_instance, data, save_in_attachment,
            )
            with closing(open(result_path, 'r')) as out_stream:
                tmpl_data = out_stream.read()
            datadict = {}
        else:
            result_fd, result_path = tempfile.mkstemp(
                suffix='.' + filetype, prefix='p3o.report.tmp.')
            tmpl_data = self.get_template(model_instance)

            in_stream = StringIO(tmpl_data)
            with closing(os.fdopen(result_fd, 'w+')) as out_stream:
                template = Template(in_stream, out_stream, escape_false=True)
                localcontext = self._get_parser_context(model_instance, data)
                expressions = template.get_all_user_python_expression()
                py_expression = template.convert_py3o_to_python_ast(
                    expressions)
                convertor = Py3oConvertor()
                data_struct = convertor(py_expression)
                datadict = data_struct.render(localcontext)

        # Call py3o.server to render the template in the desired format
        files = {
            'tmpl_file': tmpl_data,
        }
        fields = {
            "targetformat": filetype,
            "datadict": json.dumps(datadict),
            "image_mapping": "{}",
            "escape_false": "on",
        }
        if report_xml.py3o_is_local_fusion:
            fields['skipfusion'] = '1'
        r = requests.post(
            report_xml.py3o_server_id.url, data=fields, files=files)
        if r.status_code != 200:
            # server says we have an issue... let's tell that to enduser
            raise UserError(
                _('Fusion server error %s') % r.text,
            )

        chunk_size = 1024
        with open(result_path, 'w+') as fd:
            for chunk in r.iter_content(chunk_size):
                fd.write(chunk)
        if len(model_instance) == 1:
            self._postprocess_report(
                result_path, model_instance.id, save_in_attachment)
        return result_path