from odoo import _, api, fields, models
from odoo.exceptions import ValidationError


class TemporaryExemption(models.TransientModel):
    _name = "beesdoo.shift.temporary_exemption"
    _description = "beesdoo.shift.temporary_exemption"
    _inherit = "beesdoo.shift.action_mixin"

    temporary_exempt_reason_id = fields.Many2one(
        "cooperative.exempt.reason", "Exempt Reason", required=True
    )
    temporary_exempt_start_date = fields.Date(
        default=fields.Date.today, required=True
    )
    temporary_exempt_end_date = fields.Date(required=True)

    @api.multi
    def exempt(self):
        self = self._check()  # maybe a different group
        status_id = self.env["cooperative.status"].search(
            [("cooperator_id", "=", self.cooperator_id.id)]
        )
        if (
            status_id.temporary_exempt_end_date
            and status_id.temporary_exempt_end_date >= status_id.today
        ):
            raise ValidationError(
                _(
                    "You cannot encode new temporary exemption since the "
                    "previous one are not over yet "
                )
            )
        reason = self.temporary_exempt_reason_id
        status_id.sudo().write(
            {
                "temporary_exempt_start_date": self.temporary_exempt_start_date,
                "temporary_exempt_end_date": self.temporary_exempt_end_date,
                "temporary_exempt_reason_id": reason.id,
            }
        )
        self.env["beesdoo.shift.shift"].sudo().unsubscribe_from_today(
            [self.cooperator_id.id],
            today=self.temporary_exempt_start_date,
            end_date=self.temporary_exempt_end_date,
        )