File: ir_actions_report.py

package info (click to toggle)
odoo 18.0.0%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 878,716 kB
  • sloc: javascript: 927,937; python: 685,670; xml: 388,524; sh: 1,033; sql: 415; makefile: 26
file content (53 lines) | stat: -rw-r--r-- 1,975 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import io

from odoo import models
from odoo.tools.pdf import OdooPdfFileReader, OdooPdfFileWriter


class IrActionsReport(models.Model):
    _inherit = 'ir.actions.report'

    def _render_qweb_pdf_prepare_streams(self, report_ref, data, res_ids=None):
        # EXTENDS base
        collected_streams = super()._render_qweb_pdf_prepare_streams(report_ref, data, res_ids=res_ids)

        if collected_streams \
                and res_ids \
                and len(res_ids) == 1 \
                and self._is_purchase_order_report(report_ref):
            purchase_order = self.env['purchase.order'].browse(res_ids)
            builders = purchase_order._get_edi_builders()

            if len(builders) == 0:
                return collected_streams

            # Read pdf content.
            pdf_stream = collected_streams[purchase_order.id]['stream']
            pdf_content = pdf_stream.getvalue()
            reader_buffer = io.BytesIO(pdf_content)
            reader = OdooPdfFileReader(reader_buffer, strict=False)
            writer = OdooPdfFileWriter()
            writer.cloneReaderDocumentRoot(reader)
            for builder in builders:
                xml_content = builder._export_order(purchase_order)

                # Post-process and embed the edi document.
                writer.addAttachment(
                    builder._export_purchase_order_filename(purchase_order),
                    xml_content,
                    subtype='text/xml'
                )

            # Replace the current content.
            pdf_stream.close()
            new_pdf_stream = io.BytesIO()
            writer.write(new_pdf_stream)
            collected_streams[purchase_order.id]['stream'] = new_pdf_stream

        return collected_streams

    def _is_purchase_order_report(self, report_ref):
        return self._get_report(report_ref).report_name in (
            'purchase.report_purchasequotation',
            'purchase.report_purchaseorder'
        )