mirror of
https://github.com/OCA/report-print-send.git
synced 2025-02-16 07:11:31 +02:00
0d8676547a
Standard migration, incl. changes proposed in https://github.com/OCA/report-print-send/pull/290/ Add unit test.
90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
# Author: Guewen Baconnier
|
|
# Copyright 2012-2023 Camptocamp SA
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
|
|
|
|
import base64
|
|
|
|
import requests
|
|
|
|
from odoo import _, api, fields, models
|
|
from odoo.exceptions import UserError
|
|
|
|
|
|
class IrAttachment(models.Model):
|
|
|
|
_inherit = "ir.attachment"
|
|
|
|
send_to_pingen = fields.Boolean("Send to Pingen.com")
|
|
pingen_document_ids = fields.One2many(
|
|
"pingen.document", "attachment_id", string="Pingen Document", readonly=True
|
|
)
|
|
|
|
def _prepare_pingen_document_vals(self):
|
|
return {
|
|
"attachment_id": self.id,
|
|
}
|
|
|
|
def _handle_pingen_document(self):
|
|
"""Reponsible of the related ``pingen.document``
|
|
when the ``send_to_pingen``
|
|
field is modified.
|
|
Only one pingen document can be created per attachment.
|
|
When ``send_to_pingen`` is activated:
|
|
* Create a ``pingen.document`` if it does not already exist
|
|
* Put the related ``pingen.document`` to ``pending``
|
|
if it already exist
|
|
When it is deactivated:
|
|
* Do nothing if no related ``pingen.document`` exists
|
|
* Or cancel it
|
|
* If it has already been pushed to pingen.com, raises
|
|
an `osv.except_osv` exception
|
|
"""
|
|
pingen_document_obj = self.env["pingen.document"]
|
|
document = self.pingen_document_ids[0] if self.pingen_document_ids else None
|
|
if self.send_to_pingen:
|
|
if document:
|
|
document.write({"state": "pending"})
|
|
else:
|
|
pingen_document_obj.create(self._prepare_pingen_document_vals())
|
|
else:
|
|
if document:
|
|
if document.state == "pushed":
|
|
raise UserError(
|
|
_(
|
|
"Error. The attachment %s is "
|
|
"already pushed to pingen.com."
|
|
)
|
|
% self.name
|
|
)
|
|
document.write({"state": "canceled"})
|
|
return
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
attachment = super(IrAttachment, self).create(vals)
|
|
if "send_to_pingen" in vals:
|
|
attachment._handle_pingen_document()
|
|
return attachment
|
|
|
|
def write(self, vals):
|
|
res = super(IrAttachment, self).write(vals)
|
|
if "send_to_pingen" in vals:
|
|
for attachment in self:
|
|
attachment._handle_pingen_document()
|
|
return res
|
|
|
|
def _decoded_content(self):
|
|
"""Returns the decoded content of an attachment (stored or url)
|
|
Returns None if the type is 'url' and the url is not reachable.
|
|
"""
|
|
decoded_document = None
|
|
if self.type == "binary":
|
|
decoded_document = base64.b64decode(self.datas)
|
|
elif self.type == "url":
|
|
response = requests.get(self.url, timeout=30)
|
|
if response.ok:
|
|
decoded_document = requests.content
|
|
else:
|
|
raise UserError(_("The type of attachment %s is not handled") % self.type)
|
|
return decoded_document
|