from openerp 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