# -*- coding: utf-8 -*-
# © 2015-2017 Akretion (http://www.akretion.com)
# @author Alexis de Lattre <alexis.delattre@akretion.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

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


class HrHolidaysMassAllocation(models.TransientModel):
    _name = 'hr.holidays.mass.allocation'
    _description = 'Wizard for mass allocation of holidays'

    @api.model
    def _default_employees(self):
        return self.env['hr.employee'].search([
            ('holiday_exclude_mass_allocation', '=', False),
            ('company_id', '=', self.env.user.company_id.id),
            ])

    @api.model
    def _get_default_holiday_status(self):
        res = self.env.user.company_id.\
            mass_allocation_default_holiday_status_id or False
        return res

    number_of_days = fields.Float(
        string='Number of Days', required=True, default=2.08)
    holiday_status_id = fields.Many2one(
        'hr.holidays.status', string='Leave Type', required=True,
        default=_get_default_holiday_status)
    employee_ids = fields.Many2many(
        'hr.employee', string='Employees', default=_default_employees)
    auto_approve = fields.Boolean(
        string='Automatic Approval', default=True)
    # size=64 because the name field of hr.holidays is size=64
    name = fields.Char('Description', size=64)

    _sql_constraints = [(
        'number_of_days_positive',
        'CHECK (number_of_days > 0)',
        'The number of days must be positive',
        )]

    @api.multi
    def run(self):
        self.ensure_one()
        if not self.number_of_days:
            raise UserError(_(
                'You must set a value for the number of days.'))
        if not self.employee_ids:
            raise UserError(_(
                'You must select at least one employee.'))
        alloc_hol_ids = []
        hho = self.env['hr.holidays']
        auto_approve = self.auto_approve
        for employee in self.employee_ids:
            hol = hho.create({
                'name': self.name,
                'number_of_days_temp': self.number_of_days,
                'employee_id': employee.id,
                'type': 'add',
                'holiday_type': 'employee',
                'holiday_status_id': self.holiday_status_id.id,
                'state': 'confirm',
                })
            if auto_approve:
                hol.with_context(no_email_notification=True).action_validate()
            alloc_hol_ids.append(hol.id)
        action = self.env['ir.actions.act_window'].for_xml_id(
            'hr_holidays', 'open_allocation_holidays')
        action.update({
            'target': 'current',
            'domain': [('id', 'in', alloc_hol_ids)],
            'nodestroy': True,
            'context': "{'default_type':'add'}",
            })
        return action