19.0 vanilla

odoo-bringout-oca-ocb-hr_expense/hr_expense/models/ir_actions_report.py

@@ -0,0 +1,54 @@
+import io
+from odoo import models, _
+from odoo.tools import pdf
+from odoo.tools.pdf import OdooPdfFileReader, OdooPdfFileWriter, PdfReadError, DependencyError
+
+
+class IrActionsReport(models.Model):
+    _inherit = 'ir.actions.report'
+
+    def _render_qweb_pdf_prepare_streams(self, report_ref, data, res_ids=None):
+        # OVERRIDE
+        res = super()._render_qweb_pdf_prepare_streams(report_ref, data, res_ids)
+        if not res_ids:
+            return res
+        report = self._get_report(report_ref)
+        if report.report_name == 'hr_expense.report_expense':
+            for expense in self.env['hr.expense'].browse(res_ids):
+                # Will contains the expense
+                stream_list = []
+                stream = res[expense.id]['stream']
+                stream_list.append(stream)
+                attachments = self.env['ir.attachment'].search([('res_id', 'in', expense.ids), ('res_model', '=', 'hr.expense')])
+                expense_report = OdooPdfFileReader(stream, strict=False)
+                output_pdf = OdooPdfFileWriter()
+                output_pdf.appendPagesFromReader(expense_report)
+                for attachment in self._prepare_local_attachments(attachments):
+                    if attachment.mimetype == 'application/pdf':
+                        attachment_stream = pdf.to_pdf_stream(attachment)
+                    else:
+                        # In case the attachment is not a pdf we will create a new PDF from the template "report_expense_img"
+                        # And then append to the stream. By doing so, the attachment is put on a new page with the name of the expense
+                        # associated to the attachment
+                        data['attachment'] = attachment
+                        attachment_prep_stream = self._render_qweb_pdf_prepare_streams('hr_expense.report_expense_img', data, res_ids=res_ids)
+                        attachment_stream = attachment_prep_stream[expense.id]['stream']
+                    attachment_reader = OdooPdfFileReader(attachment_stream, strict=False)
+                    try:
+                        output_pdf.appendPagesFromReader(attachment_reader)
+                    except (PdfReadError, DependencyError) as e:
+                        expense._message_log(body=_(
+                            "The attachment (%(attachment_name)s) has not been added to the report due to the following error: '%(error)s'",
+                            attachment_name=attachment.name,
+                            error=e
+                        ))
+                        continue
+                    stream_list.append(attachment_stream)
+
+                new_pdf_stream = io.BytesIO()
+                output_pdf.write(new_pdf_stream)
+                res[expense.id]['stream'] = new_pdf_stream
+
+                for stream in stream_list:
+                    stream.close()
+        return res