Add mass allocation wizard

Add counter info on form view of hr.holidays
Fix an issue when checking counter

hr_holidays_usability/wizard/hr_holidays_mass_allocation.py

@@ -0,0 +1,97 @@
+# -*- encoding: utf-8 -*-
+##############################################################################
+#
+#    HR Holidays Usability module for Odoo
+#    Copyright (C) 2015 Akretion (http://www.akretion.com)
+#    @author Alexis de Lattre <alexis.delattre@akretion.com>
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as
+#    published by the Free Software Foundation, either version 3 of the
+#    License, or (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+
+from openerp.osv import orm, fields
+from openerp.tools.translate import _
+from openerp import netsvc
+
+
+class HrHolidaysMassAllocation(orm.TransientModel):
+    _name = 'hr.holidays.mass.allocation'
+    _description = 'Wizard for mass allocation of holidays'
+
+    def _get_all_employees(self, cr, uid, context=None):
+        return self.pool['hr.employee'].search(cr, uid, [], context=context)
+
+    _columns = {
+        'number_of_days': fields.float('Number of Days', required=True),
+        'holiday_status_id': fields.many2one(
+            'hr.holidays.status', 'Leave Type', required=True),
+        'employee_ids': fields.many2many(
+            'hr.employee', string='Employees'),
+        'auto_approve': fields.boolean('Automatic Approval'),
+        # size=64 because the name field of hr.holidays is size=64
+        'name': fields.char('Description', size=64),
+        }
+
+    _defaults = {
+        'number_of_days': 2.08,
+        'employee_ids': _get_all_employees,
+        'auto_approve': True,
+    }
+
+    _sql_constraints = [(
+        'number_of_days_positive',
+        'CHECK (number_of_days > 0)',
+        'The number of days must be positive',
+        )]
+
+    def run(self, cr, uid, ids, context=None):
+        assert len(ids) == 1, 'Only 1 ID'
+        wiz = self.browse(cr, uid, ids[0], context=context)
+        if not wiz.number_of_days:
+            raise orm.except_orm(
+                _('Error:'),
+                _('You must set a value for the number of days.'))
+        if not wiz.employee_ids:
+            raise orm.except_orm(
+                _('Error:'),
+                _('You must select at least one employee.'))
+        alloc_hol_ids = []
+        hho = self.pool['hr.holidays']
+        wf_service = netsvc.LocalService("workflow")
+        auto_approve = wiz.auto_approve
+        for employee in wiz.employee_ids:
+            hol_id = hho.create(cr, uid, {
+                'name': wiz.name,
+                'number_of_days_temp': wiz.number_of_days,
+                'employee_id': employee.id,
+                'type': 'add',
+                'holiday_type': 'employee',
+                'holiday_status_id': wiz.holiday_status_id.id,
+                }, context=context)
+            if auto_approve:
+                wf_service.trg_validate(
+                    uid, 'hr.holidays', hol_id, 'validate', cr)
+            alloc_hol_ids.append(hol_id)
+        act_model, act_id = self.pool['ir.model.data'].get_object_reference(
+            cr, uid, 'hr_holidays', 'open_allocation_holidays')
+        assert act_model == 'ir.actions.act_window', 'Wrong model'
+        action = self.pool[act_model].read(
+            cr, uid, act_id, context=context)
+        action.update({
+            'target': 'current',
+            'domain': [('id', 'in', alloc_hol_ids)],
+            'nodestroy': True,
+            'context': context,
+            })
+        return action