from odoo import _, fields, models


class TierMoveToMailboxWizard(models.TransientModel):
    _name = 'tier.move.to.mailbox.wizard'
    _description = 'Переместить сообщение в ящик'

    message_id = fields.Many2one('mail.message', string='Сообщение', required=True)
    mailbox_ids = fields.Many2many(
        'tier.mailbox',
        string='Ящики',
        help='Выберите ящики, в которые нужно поместить это сообщение.',
    )

    def action_confirm(self):
        """Apply mailbox assignment to the message."""
        self.ensure_one()
        message = self.message_id
        # Remove existing mailbox tags
        message.mailbox_tag_ids.unlink()
        # Create new ones
        for mailbox in self.mailbox_ids:
            self.env['tier.mailbox.tag'].create({
                'mailbox_id': mailbox.id,
                'message_id': message.id,
            })
        return {'type': 'ir.actions.act_window_close'}
