# Copyright 2019 Artem Rafailov <https://it-projects.info/team/Ommo73/>
# Copyright 2019 Eugene Molotov <https://it-projects.info/team/em230418>
# License LGPL-3.0 (https://www.gnu.org/licenses/lgpl.html).
from odoo import models, api


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

    # взято с mail_base
    @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 i in triplet[2]:
                        values['partner_ids'].append((4, i, False))
        return super(MailMessage, self).write(values)

    @api.multi
    def message_format(self):
        messages_values = super(MailMessage, self).message_format()
        for i in messages_values:
            if i['channel_ids']:
                i['channel_names'] = self.env['mail.channel'].browse(i['channel_ids']).mapped(
                    lambda r: [r.id, '#' + r.display_name])

        return messages_values