71 lines
3.1 KiB
Python
71 lines
3.1 KiB
Python
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
from odoo import api, fields, models, _
|
||
|
|
from odoo.exceptions import UserError
|
||
|
|
from odoo.tools.mimetypes import guess_mimetype
|
||
|
|
from odoo.tools.misc import format_date
|
||
|
|
|
||
|
|
|
||
|
|
class IrAttachment(models.Model):
|
||
|
|
_inherit = 'ir.attachment'
|
||
|
|
|
||
|
|
@api.ondelete(at_uninstall=True)
|
||
|
|
def _except_audit_trail(self):
|
||
|
|
audit_trail_attachments = self.filtered(lambda attachment:
|
||
|
|
attachment.res_model == 'account.move'
|
||
|
|
and attachment.res_id
|
||
|
|
and attachment.raw
|
||
|
|
and guess_mimetype(attachment.raw) in (
|
||
|
|
'application/pdf',
|
||
|
|
'application/xml',
|
||
|
|
)
|
||
|
|
)
|
||
|
|
moves = self.env['account.move'].browse(audit_trail_attachments.mapped('res_id')).exists()
|
||
|
|
id2move = {move.id: move for move in moves}
|
||
|
|
for attachment in audit_trail_attachments:
|
||
|
|
move = id2move.get(attachment.res_id)
|
||
|
|
if move and move.posted_before and move.country_code == 'DE':
|
||
|
|
ue = UserError(_("You cannot remove parts of the audit trail."))
|
||
|
|
ue._audit_trail = True
|
||
|
|
raise ue
|
||
|
|
|
||
|
|
def write(self, vals):
|
||
|
|
if vals.keys() & {'res_id', 'res_model', 'raw', 'datas', 'store_fname', 'db_datas', 'company_id'}:
|
||
|
|
try:
|
||
|
|
self._except_audit_trail()
|
||
|
|
except UserError as e:
|
||
|
|
if (
|
||
|
|
not hasattr(e, '_audit_trail')
|
||
|
|
or vals.get('res_model') != 'documents.document'
|
||
|
|
or vals.keys() & {'raw', 'datas', 'store_fname', 'db_datas'}
|
||
|
|
):
|
||
|
|
raise # do not raise if trying to version the attachment through a document
|
||
|
|
vals.pop('res_model', None)
|
||
|
|
vals.pop('res_id', None)
|
||
|
|
return super().write(vals)
|
||
|
|
|
||
|
|
def unlink(self):
|
||
|
|
invoice_pdf_attachments = self.filtered(lambda attachment:
|
||
|
|
attachment.res_model == 'account.move'
|
||
|
|
and attachment.res_id
|
||
|
|
and attachment.res_field in ('invoice_pdf_report_file', 'ubl_cii_xml_file')
|
||
|
|
and attachment.company_id.account_fiscal_country_id.code == 'DE'
|
||
|
|
)
|
||
|
|
if invoice_pdf_attachments:
|
||
|
|
# only detach the document from the field, but keep it in the database for the audit trail
|
||
|
|
# it shouldn't be an issue as there aren't any security group on the fields as it is the public report
|
||
|
|
invoice_pdf_attachments.res_field = False
|
||
|
|
today = format_date(self.env, fields.Date.context_today(self))
|
||
|
|
for attachment in invoice_pdf_attachments:
|
||
|
|
attachment_name, attachment_extension = os.path.splitext(attachment.name)
|
||
|
|
attachment.name = _(
|
||
|
|
'%(attachment_name)s (detached by %(user)s on %(date)s)%(attachment_extension)s',
|
||
|
|
attachment_name=attachment_name,
|
||
|
|
attachment_extension=attachment_extension,
|
||
|
|
user=self.env.user.name,
|
||
|
|
date=today,
|
||
|
|
)
|
||
|
|
return super(IrAttachment, self - invoice_pdf_attachments).unlink()
|