Martin Trigaux
ffc525419a
The render API was confusing as mixing the access to the report and the rendering env. The ambiguity was present for code such as `report.sudo()._render(record_ids)` where it was not clear if the `sudo()` is needed to access to `report` or to `record_ids`. For low priviledge users (such as portal or public), it was common to use `report.with_user(SUPERUSER_ID)._render(record_ids)`. This PR changes the render methods signature to be `api.model`. The `report_ref` can be: - ir.actions.report external id - ir.actions.report id - ir.actions.report recod - `report_name` value This will allow to call the report methods with any user and no longer need to use `with_user(1)` to render reports as public user. Task-id 2670865 closes odoo/odoo#91341 Related: odoo/upgrade#3650 Related: odoo/enterprise#27323 Signed-off-by: Martin Trigaux (mat) <mat@odoo.com>
55 lines
2.5 KiB
Python
55 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
import io
|
|
import textwrap
|
|
from collections import OrderedDict
|
|
|
|
from odoo import 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)
|
|
if any(x.move_type not in ('in_invoice', 'in_receipt') for x in invoices):
|
|
raise UserError(_("You can only print the original document for vendor bills."))
|
|
|
|
original_attachments = invoices.message_main_attachment_id
|
|
if not original_attachments:
|
|
raise UserError(_("No original vendor bills could be found for any of the selected vendor bills."))
|
|
|
|
collected_streams = OrderedDict()
|
|
for invoice in invoices:
|
|
attachment = invoice.message_main_attachment_id
|
|
if attachment:
|
|
stream = io.BytesIO(attachment.raw)
|
|
if attachment.mimetype == 'application/pdf':
|
|
record = self.env[attachment.res_model].browse(attachment.res_id)
|
|
try:
|
|
stream = pdf.add_banner(stream, record.name, logo=True)
|
|
except ValueError:
|
|
raise UserError(_(
|
|
"Error when reading the original PDF for: %r.\nPlease make sure the file is valid.",
|
|
textwrap.shorten(record.name, width=100)
|
|
))
|
|
collected_streams[invoice.id] = {
|
|
'stream': stream,
|
|
'attachment': attachment,
|
|
}
|
|
return collected_streams
|
|
|
|
def _render_qweb_pdf(self, report_ref, res_ids=None, data=None):
|
|
# Check for reports only available for invoices.
|
|
if self._get_report(report_ref).report_name in ('account.report_invoice_with_payments', 'account.report_invoice'):
|
|
invoices = self.env['account.move'].browse(res_ids)
|
|
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)
|