mirror of
https://github.com/OCA/report-print-send.git
synced 2025-02-16 07:11:31 +02:00
0b9574f3cc
on pingen ir.attachment. When multiple module redefine create, write it can lead to TypeError
100 lines
3.6 KiB
Python
100 lines
3.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Author: Guewen Baconnier
|
|
# Copyright 2012-2017 Camptocamp SA
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
|
|
|
|
import requests
|
|
import base64
|
|
|
|
from odoo import models, fields, _, api
|
|
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)
|
|
pingen_send = fields.Boolean(
|
|
'Send', help="Defines if a document is merely uploaded or also sent",
|
|
default=True)
|
|
pingen_speed = fields.Selection(
|
|
[('1', 'Priority'), ('2', 'Economy')],
|
|
'Speed', default='2',
|
|
help='Defines the sending speed if the document is automatically sent')
|
|
pingen_color = fields.Selection([('0', 'B/W'), ('1', 'Color')],
|
|
'Type of print',
|
|
default='0')
|
|
|
|
def _prepare_pingen_document_vals(self):
|
|
return {'attachment_id': self.id,
|
|
'config': 'created from attachment'}
|
|
|
|
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
|
|
|
|
@api.multi
|
|
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)
|
|
if response.ok:
|
|
decoded_document = requests.content
|
|
else:
|
|
raise UserError(
|
|
_('The type of attachment %s is not handled')
|
|
% self.type)
|
|
return decoded_document
|