Odoo18-Base/addons/account/models/ir_actions_report.py

81 lines
3.8 KiB
Python
Raw Permalink Normal View History

2025-03-10 10:52:11 +07:00
# -*- coding: utf-8 -*-
from collections import OrderedDict
from zlib import error as zlib_error
from odoo import api, models, _
from odoo.exceptions import UserError
from odoo.tools import pdf
from odoo.tools.pdf import PdfReadError, PdfStreamError
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, PdfStreamError, PdfReadError, TypeError, zlib_error, NotImplementedError):
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))
def _get_rendering_context(self, report, docids, data):
data = super()._get_rendering_context(report, docids, data)
if self.env.context.get('proforma_invoice'):
data['proforma'] = True
return data