Initial commit: Accounting packages

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

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+from collections import OrderedDict
+from zlib import error as zlib_error
+try:
+    from PyPDF2.errors import PdfReadError
+except ImportError:
+    from PyPDF2.utils import PdfReadError
+
+try:
+    from PyPDF2.errors import DependencyError
+except ImportError:
+    DependencyError = NotImplementedError
+
+from odoo import api, models, _
+from odoo.exceptions import UserError
+from odoo.tools import pdf
+
+
+class IrActionsReport(models.Model):
+    _inherit = 'ir.actions.report'
+
+    def _render_qweb_pdf_prepare_streams(self, report_ref, data, res_ids=None):
+        # Custom behavior for 'account.report_original_vendor_bill'.
+        if self._get_report(report_ref).report_name != 'account.report_original_vendor_bill':
+            return super()._render_qweb_pdf_prepare_streams(report_ref, data, res_ids=res_ids)
+
+        invoices = self.env['account.move'].browse(res_ids)
+        original_attachments = invoices.message_main_attachment_id
+        if not original_attachments:
+            raise UserError(_("No original purchase document could be found for any of the selected purchase documents."))
+
+        collected_streams = OrderedDict()
+        for invoice in invoices:
+            attachment = invoice.message_main_attachment_id
+            if attachment:
+                stream = pdf.to_pdf_stream(attachment)
+                if stream:
+                    record = self.env[attachment.res_model].browse(attachment.res_id)
+                    try:
+                        stream = pdf.add_banner(stream, record.name, logo=True)
+                    except (ValueError, PdfReadError, TypeError, zlib_error, NotImplementedError, DependencyError, ArithmeticError):
+                        record._message_log(body=_(
+                            "There was an error when trying to add the banner to the original PDF.\n"
+                            "Please make sure the source file is valid."
+                        ))
+                collected_streams[invoice.id] = {
+                    'stream': stream,
+                    'attachment': attachment,
+                }
+        return collected_streams
+
+    def _is_invoice_report(self, report_ref):
+        return self._get_report(report_ref).report_name in ('account.report_invoice_with_payments', 'account.report_invoice')
+
+    def _render_qweb_pdf(self, report_ref, res_ids=None, data=None):
+        # Check for reports only available for invoices.
+        # + append context data with the display_name_in_footer parameter
+        if self._is_invoice_report(report_ref):
+            invoices = self.env['account.move'].browse(res_ids)
+            if self.env['ir.config_parameter'].sudo().get_param('account.display_name_in_footer'):
+                data = data and dict(data) or {}
+                data.update({'display_name_in_footer': True})
+            if any(x.move_type == 'entry' for x in invoices):
+                raise UserError(_("Only invoices could be printed."))
+
+        return super()._render_qweb_pdf(report_ref, res_ids=res_ids, data=data)
+
+    @api.ondelete(at_uninstall=False)
+    def _unlink_except_master_tags(self):
+        master_xmlids = [
+            "account_invoices",
+            "action_account_original_vendor_bill"
+            "account_invoices_without_payment",
+            "action_report_journal",
+            "action_report_payment_receipt",
+            "action_report_account_statement",
+            "action_report_account_hash_integrity",
+        ]
+        for master_xmlid in master_xmlids:
+            master_report = self.env.ref(f"account.{master_xmlid}", raise_if_not_found=False)
+            if master_report and master_report in self:
+                raise UserError(_("You cannot delete this report (%s), it is used by the accounting PDF generation engine.", master_report.name))