mirror of
https://github.com/bringout/oca-edi.git
synced 2026-04-18 07:52:05 +02:00
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
# Copyright 2022 Camptocamp SA
|
|
# @author: Simone Orsi <simahawk@gmail.com>
|
|
# Copyright 2023 Jacques-Etienne Baudoux (BCIM) <je@bcim.be>
|
|
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).
|
|
|
|
import io
|
|
import logging
|
|
|
|
from odoo import api, models
|
|
from odoo.tools.pdf import NameObject, OdooPdfFileReader, OdooPdfFileWriter
|
|
|
|
from ..utils import PDFParser
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class PDFHelper(models.AbstractModel):
|
|
_name = "pdf.helper"
|
|
_description = "PDF Helper"
|
|
|
|
_PDF_PARSER_KLASS = PDFParser
|
|
|
|
@api.model
|
|
def pdf_get_xml_files(self, pdf_file):
|
|
"""Extract XML attachments from pdf
|
|
|
|
:param pdf_file: binary PDF file content
|
|
:returns: a dict like {$filename: $parsed_xml_file_obj}.
|
|
"""
|
|
parser = self._PDF_PARSER_KLASS(pdf_file)
|
|
try:
|
|
return parser.get_xml_files()
|
|
except parser.get_xml_files_swallable_exceptions() as err:
|
|
_logger.error("PDF file parsing failed: %s", str(err))
|
|
return {}
|
|
|
|
@api.model
|
|
def pdf_embed_xml(self, pdf_content, xml_filename, xml_string):
|
|
"""Add an XML attachment in a pdf"""
|
|
with io.BytesIO(pdf_content) as reader_buffer, io.BytesIO() as new_pdf_stream:
|
|
reader = OdooPdfFileReader(reader_buffer, strict=False)
|
|
writer = OdooPdfFileWriter()
|
|
writer.cloneReaderDocumentRoot(reader)
|
|
writer.addAttachment(xml_filename, xml_string, subtype="text/xml")
|
|
# show attachments when opening PDF
|
|
writer._root_object.update(
|
|
{NameObject("/PageMode"): NameObject("/UseAttachments")}
|
|
)
|
|
writer.write(new_pdf_stream)
|
|
return new_pdf_stream.getvalue()
|