social/mail_attach_existing_attach.../wizard/mail_compose_message.py

58 lines
2.0 KiB
Python

# Copyright 2015 ACSONE SA/NV
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
from odoo import api, fields, models
class MailComposeMessage(models.TransientModel):
_inherit = "mail.compose.message"
@api.model
def default_get(self, fields_list):
res = super().default_get(fields_list)
if (
"can_attach_attachment" not in res
and res.get("model")
and res.get("res_ids")
and res.get("composition_mode", "") != "mass_mail"
):
res["can_attach_attachment"] = True # pragma: no cover
return res
can_attach_attachment = fields.Boolean()
object_attachment_ids = fields.Many2many(
comodel_name="ir.attachment",
relation="mail_compose_message_ir_attachments_object_rel",
column1="wizard_id",
column2="attachment_id",
string="Object Attachments",
)
display_object_attachment_ids = fields.One2many(
comodel_name="ir.attachment",
compute="_compute_display_object_attachment_ids",
)
@api.depends("res_ids", "model")
def _compute_display_object_attachment_ids(self):
for composer in self:
res_ids = self._evaluate_res_ids()
model = self.model
if model and res_ids:
attachments = self.env["ir.attachment"].search(
[
("res_model", "=", model),
("res_id", "in", res_ids),
]
)
composer.display_object_attachment_ids = attachments
else:
composer.display_object_attachment_ids = False
def _prepare_mail_values(self, res_ids):
res = super()._prepare_mail_values(res_ids)
if self.object_attachment_ids.ids and self.model and len(res_ids) == 1:
res[res_ids[0]].setdefault("attachment_ids", []).extend(
self.object_attachment_ids.ids
)
return res