from odoo import api, models


class MailMessage(models.Model):
    _inherit = "mail.message"

    @api.multi
    def write(self, values):
        if values.get("needaction_partner_ids"):
            if not values.get("partner_ids"):
                values["partner_ids"] = []
            for triplet in values.get("needaction_partner_ids"):
                if triplet[0] == 6:
                    for id in triplet[2]:
                        values["partner_ids"].append((4, id, False))
        return super(MailMessage, self).write(values)


class MailComposer(models.TransientModel):

    _inherit = "mail.compose.message"

    @api.multi
    def send_mail(self, auto_commit=False):
        res = super(MailComposer, self).send_mail(auto_commit=auto_commit)
        notification = {}
        self.env["bus.bus"].sendone(
            (self._cr.dbname, "mail_base.mail_sent"), notification
        )

        return res