addons/hr_holidays/hr_holidays.py

   1 # -*- coding: utf-8 -*-
   2 ##################################################################################
   3 #
   4 # Copyright (c) 2005-2006 Axelor SARL. (http://www.axelor.com)
   5 # and 2004-2010 Tiny SPRL (<http://tiny.be>).
   6 #
   7 # $Id: hr.py 4656 2006-11-24 09:58:42Z Cyp $
   8 #
   9 #     This program is free software: you can redistribute it and/or modify
  10 #     it under the terms of the GNU Affero General Public License as
  11 #     published by the Free Software Foundation, either version 3 of the
  12 #     License, or (at your option) any later version.
  13 #
  14 #     This program is distributed in the hope that it will be useful,
  15 #     but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 #     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17 #     GNU Affero General Public License for more details.
  18 #
  19 #     You should have received a copy of the GNU Affero General Public License
  20 #     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 #
  22 ##############################################################################
  23
  24 import datetime, time
  25 from itertools import groupby
  26 from operator import itemgetter
  27
  28 import math
  29 import netsvc
  30 from osv import fields, osv
  31 from tools.translate import _
  32
  33
  34 class hr_holidays_status(osv.osv):
  35     _name = "hr.holidays.status"
  36     _description = "Leave Type"
  37
  38     def get_days(self, cr, uid, ids, employee_id, return_false, context=None):
  39         cr.execute("""SELECT id, type, number_of_days, holiday_status_id FROM hr_holidays WHERE employee_id = %s AND state='validate' AND holiday_status_id in %s""",
  40             [employee_id, tuple(ids)])
  41         result = sorted(cr.dictfetchall(), key=lambda x: x['holiday_status_id'])
  42         grouped_lines = dict((k, [v for v in itr]) for k, itr in groupby(result, itemgetter('holiday_status_id')))
  43         res = {}
  44         for record in self.browse(cr, uid, ids, context=context):
  45             res[record.id] = {}
  46             max_leaves = leaves_taken = 0
  47             if not return_false:
  48                 if record.id in grouped_lines:
  49                     leaves_taken = -sum([item['number_of_days'] for item in grouped_lines[record.id] if item['type'] == 'remove'])
  50                     max_leaves = sum([item['number_of_days'] for item in grouped_lines[record.id] if item['type'] == 'add'])
  51             res[record.id]['max_leaves'] = max_leaves
  52             res[record.id]['leaves_taken'] = leaves_taken
  53             res[record.id]['remaining_leaves'] = max_leaves - leaves_taken
  54         return res
  55
  56     def _user_left_days(self, cr, uid, ids, name, args, context=None):
  57         return_false = False
  58         employee_id = False
  59         res = {}
  60         if context and context.has_key('employee_id'):
  61             if not context['employee_id']:
  62                 return_false = True
  63             employee_id = context['employee_id']
  64         else:
  65             employee_ids = self.pool.get('hr.employee').search(cr, uid, [('user_id','=',uid)], context=context)
  66             if employee_ids:
  67                 employee_id = employee_ids[0]
  68             else:
  69                 return_false = True
  70         if employee_id:
  71             res = self.get_days(cr, uid, ids, employee_id, return_false, context=context)
  72         else:
  73             res = dict.fromkeys(ids, {'leaves_taken': 0, 'remaining_leaves': 0, 'max_leaves': 0})
  74         return res
  75
  76     _columns = {
  77         'name': fields.char('Leave Type', size=64, required=True, translate=True),
  78         'categ_id': fields.many2one('crm.meeting.type', 'Meeting Type',
  79             help='Once a leave is validated, OpenERP will create a corresponding meeting of this type in the calendar.'),
  80         'color_name': fields.selection([('red', 'Red'),('blue','Blue'), ('lightgreen', 'Light Green'), ('lightblue','Light Blue'), ('lightyellow', 'Light Yellow'), ('magenta', 'Magenta'),('lightcyan', 'Light Cyan'),('black', 'Black'),('lightpink', 'Light Pink'),('brown', 'Brown'),('violet', 'Violet'),('lightcoral', 'Light Coral'),('lightsalmon', 'Light Salmon'),('lavender', 'Lavender'),('wheat', 'Wheat'),('ivory', 'Ivory')],'Color in Report', required=True, help='This color will be used in the leaves summary located in Reporting\Leaves by Department.'),
  81         'limit': fields.boolean('Allow to Override Limit', help='If you select this checkbox, the system allows the employees to take more leaves than the available ones for this type.'),
  82         'active': fields.boolean('Active', help="If the active field is set to false, it will allow you to hide the leave type without removing it."),
  83         'max_leaves': fields.function(_user_left_days, string='Maximum Allowed', help='This value is given by the sum of all holidays requests with a positive value.', multi='user_left_days'),
  84         'leaves_taken': fields.function(_user_left_days, string='Leaves Already Taken', help='This value is given by the sum of all holidays requests with a negative value.', multi='user_left_days'),
  85         'remaining_leaves': fields.function(_user_left_days, string='Remaining Leaves', help='Maximum Leaves Allowed - Leaves Already Taken', multi='user_left_days'),
  86         'double_validation': fields.boolean('Apply Double Validation', help="When selected, the Allocation/Leave Requests for this type require a second validation to be approved."),
  87     }
  88     _defaults = {
  89         'color_name': 'red',
  90         'active': True,
  91     }
  92 hr_holidays_status()
  93
  94 class hr_holidays(osv.osv):
  95     _name = "hr.holidays"
  96     _description = "Leave"
  97     _order = "type desc, date_from asc"
  98     _inherit = [ 'mail.thread','ir.needaction_mixin']
  99
 100     def _employee_get(self, cr, uid, context=None):
 101         ids = self.pool.get('hr.employee').search(cr, uid, [('user_id', '=', uid)], context=context)
 102         if ids:
 103             return ids[0]
 104         return False
 105
 106     def _compute_number_of_days(self, cr, uid, ids, name, args, context=None):
 107         result = {}
 108         for hol in self.browse(cr, uid, ids, context=context):
 109             if hol.type=='remove':
 110                 result[hol.id] = -hol.number_of_days_temp
 111             else:
 112                 result[hol.id] = hol.number_of_days_temp
 113         return result
 114
 115     _columns = {
 116         'name': fields.char('Description', size=64),
 117         'state': fields.selection([('draft', 'To Submit'), ('cancel', 'Cancelled'),('confirm', 'To Approve'), ('refuse', 'Refused'), ('validate1', 'Second Approval'), ('validate', 'Approved')],
 118             'Status', readonly=True, help='The status is set to \'To Submit\', when a holiday request is created.\
 119             \nThe status is \'To Approve\', when holiday request is confirmed by user.\
 120             \nThe status is \'Refused\', when holiday request is refused by manager.\
 121             \nThe status is \'Approved\', when holiday request is approved by manager.'),
 122         'user_id':fields.related('employee_id', 'user_id', type='many2one', relation='res.users', string='User', store=True),
 123         'date_from': fields.datetime('Start Date', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}, select=True),
 124         'date_to': fields.datetime('End Date', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
 125         'holiday_status_id': fields.many2one("hr.holidays.status", "Leave Type", required=True,readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
 126         'employee_id': fields.many2one('hr.employee', "Employee", select=True, invisible=False, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}, help='Leave Manager can let this field empty if this leave request/allocation is for every employee'),
 127         'manager_id': fields.many2one('hr.employee', 'First Approval', invisible=False, readonly=True, help='This area is automatically filled by the user who validate the leave'),
 128         'notes': fields.text('Reasons',readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
 129         'number_of_days_temp': fields.float('Number of Days', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
 130         'number_of_days': fields.function(_compute_number_of_days, string='Number of Days', store=True),
 131         'meeting_id': fields.many2one('crm.meeting', 'Meeting'),
 132         'type': fields.selection([('remove','Leave Request'),('add','Allocation Request')], 'Request Type', required=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}, help="Choose 'Leave Request' if someone wants to take an off-day. \nChoose 'Allocation Request' if you want to increase the number of leaves available for someone", select=True),
 133         'parent_id': fields.many2one('hr.holidays', 'Parent'),
 134         'linked_request_ids': fields.one2many('hr.holidays', 'parent_id', 'Linked Requests',),
 135         'department_id':fields.related('employee_id', 'department_id', string='Department', type='many2one', relation='hr.department', readonly=True, store=True),
 136         'category_id': fields.many2one('hr.employee.category', "Category", help='Category of Employee', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
 137         'holiday_type': fields.selection([('employee','By Employee'),('category','By Employee Category')], 'Allocation Mode', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}, help='By Employee: Allocation/Request for individual Employee, By Employee Category: Allocation/Request for group of employees in category', required=True),
 138         'manager_id2': fields.many2one('hr.employee', 'Second Approval', readonly=True, help='This area is automaticly filled by the user who validate the leave with second level (If Leave type need second validation)'),
 139         'double_validation': fields.related('holiday_status_id', 'double_validation', type='boolean', relation='hr.holidays.status', string='Apply Double Validation'),
 140     }
 141     _defaults = {
 142         'employee_id': _employee_get,
 143         'state': 'draft',
 144         'type': 'remove',
 145         'user_id': lambda obj, cr, uid, context: uid,
 146         'holiday_type': 'employee'
 147     }
 148     _sql_constraints = [
 149         ('type_value', "CHECK( (holiday_type='employee' AND employee_id IS NOT NULL) or (holiday_type='category' AND category_id IS NOT NULL))", "The employee or employee category of this request is missing."),
 150         ('date_check2', "CHECK ( (type='add') OR (date_from <= date_to))", "The start date must be before the end date !"),
 151         ('date_check', "CHECK ( number_of_days_temp >= 0 )", "The number of days must be greater than 0 !"),
 152     ]
 153
 154     def _create_resource_leave(self, cr, uid, leaves, context=None):
 155         '''This method will create entry in resource calendar leave object at the time of holidays validated '''
 156         obj_res_leave = self.pool.get('resource.calendar.leaves')
 157         for leave in leaves:
 158             vals = {
 159                 'name': leave.name,
 160                 'date_from': leave.date_from,
 161                 'holiday_id': leave.id,
 162                 'date_to': leave.date_to,
 163                 'resource_id': leave.employee_id.resource_id.id,
 164                 'calendar_id': leave.employee_id.resource_id.calendar_id.id
 165             }
 166             obj_res_leave.create(cr, uid, vals, context=context)
 167         return True
 168
 169     def _remove_resource_leave(self, cr, uid, ids, context=None):
 170         '''This method will create entry in resource calendar leave object at the time of holidays cancel/removed'''
 171         obj_res_leave = self.pool.get('resource.calendar.leaves')
 172         leave_ids = obj_res_leave.search(cr, uid, [('holiday_id', 'in', ids)], context=context)
 173         return obj_res_leave.unlink(cr, uid, leave_ids, context=context)
 174
 175     def onchange_type(self, cr, uid, ids, holiday_type):
 176         result = {'value': {'employee_id': False}}
 177         if holiday_type == 'employee':
 178             ids_employee = self.pool.get('hr.employee').search(cr, uid, [('user_id','=', uid)])
 179             if ids_employee:
 180                 result['value'] = {
 181                     'employee_id': ids_employee[0]
 182                 }
 183         return result
 184
 185     def onchange_employee(self, cr, uid, ids, employee_id):
 186         result = {'value': {'department_id': False}}
 187         if employee_id:
 188             employee = self.pool.get('hr.employee').browse(cr, uid, employee_id)
 189             result['value'] = {'department_id': employee.department_id.id}
 190         return result
 191
 192     # TODO: can be improved using resource calendar method
 193     def _get_number_of_days(self, date_from, date_to):
 194         """Returns a float equals to the timedelta between two dates given as string."""
 195
 196         DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"
 197         from_dt = datetime.datetime.strptime(date_from, DATETIME_FORMAT)
 198         to_dt = datetime.datetime.strptime(date_to, DATETIME_FORMAT)
 199         timedelta = to_dt - from_dt
 200         diff_day = timedelta.days + float(timedelta.seconds) / 86400
 201         return diff_day
 202
 203     def unlink(self, cr, uid, ids, context=None):
 204         for rec in self.browse(cr, uid, ids, context=context):
 205             if rec.state not in ['draft', 'cancel', 'confirm']:
 206                 raise osv.except_osv(_('Warning!'),_('You cannot delete a leave which is in %s state!')%(rec.state))
 207         return super(hr_holidays, self).unlink(cr, uid, ids, context)
 208
 209     def onchange_date_from(self, cr, uid, ids, date_from):
 210         DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"
 211         date_to = False
 212         if date_from:
 213             date_to = datetime.datetime.strptime(date_from, DATETIME_FORMAT) + datetime.timedelta(hours=8)
 214         return {'value':{'date_to': str(date_to)}}
 215
 216     def onchange_date_to(self, cr, uid, ids, date_to, date_from):
 217         result = {}
 218         if date_to and date_from:
 219             diff_day = self._get_number_of_days(date_from, date_to)
 220             result['value'] = {
 221                 'number_of_days_temp': round(math.floor(diff_day))+1
 222             }
 223             return result
 224         result['value'] = {
 225             'number_of_days_temp': 0,
 226         }
 227         return result
 228
 229     def set_to_draft(self, cr, uid, ids, context=None):
 230         self.write(cr, uid, ids, {
 231             'state': 'draft',
 232             'manager_id': False,
 233             'manager_id2': False,
 234         })
 235         wf_service = netsvc.LocalService("workflow")
 236         for id in ids:
 237             wf_service.trg_delete(uid, 'hr.holidays', id, cr)
 238             wf_service.trg_create(uid, 'hr.holidays', id, cr)
 239         to_unlink = []
 240         for record in self.browse(cr, uid, ids, context=context):
 241             for record2 in record.linked_request_ids:
 242                 self.set_to_draft(cr, uid, [record2.id], context=context)
 243                 to_unlink.append(record2.id)
 244         if to_unlink:
 245             self.unlink(cr, uid, to_unlink, context=context)
 246         return True
 247
 248     def holidays_first_validate(self, cr, uid, ids, context=None):
 249         self.check_holidays(cr, uid, ids, context=context)
 250         obj_emp = self.pool.get('hr.employee')
 251         ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
 252         manager = ids2 and ids2[0] or False
 253         self.holidays_first_validate_notificate(cr, uid, ids, context=context)
 254         return self.write(cr, uid, ids, {'state':'validate1', 'manager_id': manager})
 255
 256     def holidays_validate(self, cr, uid, ids, context=None):
 257         self.check_holidays(cr, uid, ids, context=context)
 258         obj_emp = self.pool.get('hr.employee')
 259         ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
 260         manager = ids2 and ids2[0] or False
 261         self.write(cr, uid, ids, {'state':'validate'})
 262         data_holiday = self.browse(cr, uid, ids)
 263         for record in data_holiday:
 264             if record.double_validation:
 265                 self.write(cr, uid, [record.id], {'manager_id2': manager})
 266             else:
 267                 self.write(cr, uid, [record.id], {'manager_id': manager})
 268             if record.holiday_type == 'employee' and record.type == 'remove':
 269                 meeting_obj = self.pool.get('crm.meeting')
 270                 meeting_vals = {
 271                     'name': record.name or _('Leave Request'),
 272                     'categ_ids': record.holiday_status_id.categ_id and [(6,0,[record.holiday_status_id.categ_id.id])] or [],
 273                     'duration': record.number_of_days_temp * 8,
 274                     'description': record.notes,
 275                     'user_id': record.user_id.id,
 276                     'date': record.date_from,
 277                     'end_date': record.date_to,
 278                     'date_deadline': record.date_to,
 279                     'state': 'open',            # to block that meeting date in the calendar
 280                 }
 281                 meeting_id = meeting_obj.create(cr, uid, meeting_vals)
 282                 self._create_resource_leave(cr, uid, [record], context=context)
 283                 self.write(cr, uid, ids, {'meeting_id': meeting_id})
 284             elif record.holiday_type == 'category':
 285                 emp_ids = obj_emp.search(cr, uid, [('category_ids', 'child_of', [record.category_id.id])])
 286                 leave_ids = []
 287                 for emp in obj_emp.browse(cr, uid, emp_ids):
 288                     vals = {
 289                         'name': record.name,
 290                         'type': record.type,
 291                         'holiday_type': 'employee',
 292                         'holiday_status_id': record.holiday_status_id.id,
 293                         'date_from': record.date_from,
 294                         'date_to': record.date_to,
 295                         'notes': record.notes,
 296                         'number_of_days_temp': record.number_of_days_temp,
 297                         'parent_id': record.id,
 298                         'employee_id': emp.id
 299                     }
 300                     leave_ids.append(self.create(cr, uid, vals, context=None))
 301                 wf_service = netsvc.LocalService("workflow")
 302                 for leave_id in leave_ids:
 303                     wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'confirm', cr)
 304                     wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'validate', cr)
 305                     wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'second_validate', cr)
 306         self.holidays_validate_notificate(cr, uid, ids, context=context)
 307         return True
 308
 309     def holidays_confirm(self, cr, uid, ids, context=None):
 310         self.check_holidays(cr, uid, ids, context=context)
 311         for record in self.browse(cr, uid, ids, context=context):
 312             if record.employee_id and record.employee_id.parent_id and record.employee_id.parent_id.user_id:
 313                 self.message_subscribe_users(cr, uid, [record.id], user_ids=[record.employee_id.parent_id.user_id.id], context=context)
 314         self.holidays_confirm_notificate(cr, uid, ids, context=context)
 315         return self.write(cr, uid, ids, {'state':'confirm'})
 316
 317     def holidays_refuse(self, cr, uid, ids, context=None):
 318         obj_emp = self.pool.get('hr.employee')
 319         ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
 320         manager = ids2 and ids2[0] or False
 321         for holiday in self.browse(cr, uid, ids, context=context):
 322             if holiday.state == 'validate1':
 323                 self.write(cr, uid, [holiday.id], {'state': 'refuse', 'manager_id': manager})
 324             else:
 325                 self.write(cr, uid, [holiday.id], {'state': 'refuse', 'manager_id2': manager})
 326         self.holidays_refuse_notificate(cr, uid, ids, context=context)
 327         self.holidays_cancel(cr, uid, ids, context=context)
 328         return True
 329
 330     def holidays_cancel(self, cr, uid, ids, context=None):
 331         meeting_obj = self.pool.get('crm.meeting')
 332         for record in self.browse(cr, uid, ids):
 333             # Delete the meeting
 334             if record.meeting_id:
 335                 meeting_obj.unlink(cr, uid, [record.meeting_id.id])
 336
 337             # If a category that created several holidays, cancel all related
 338             wf_service = netsvc.LocalService("workflow")
 339             for request in record.linked_request_ids or []:
 340                 wf_service.trg_validate(uid, 'hr.holidays', request.id, 'refuse', cr)
 341
 342         self._remove_resource_leave(cr, uid, ids, context=context)
 343         return True
 344
 345     def check_holidays(self, cr, uid, ids, context=None):
 346         holi_status_obj = self.pool.get('hr.holidays.status')
 347         for record in self.browse(cr, uid, ids):
 348             if record.holiday_type == 'employee' and record.type == 'remove':
 349                 if record.employee_id and not record.holiday_status_id.limit:
 350                     leaves_rest = holi_status_obj.get_days( cr, uid, [record.holiday_status_id.id], record.employee_id.id, False)[record.holiday_status_id.id]['remaining_leaves']
 351                     if leaves_rest < record.number_of_days_temp:
 352                         raise osv.except_osv(_('Warning!'), _('There are not enough %s allocated for employee %s; please create an allocation request for this leave type.') % (record.holiday_status_id.name, record.employee_id.name))
 353         return True
 354
 355     # -----------------------------
 356     # OpenChatter and notifications
 357     # -----------------------------
 358
 359     def needaction_domain_get(self, cr, uid, ids, context=None):
 360         emp_obj = self.pool.get('hr.employee')
 361         empids = emp_obj.search(cr, uid, [('parent_id.user_id', '=', uid)], context=context)
 362         dom = ['&', ('state', '=', 'confirm'), ('employee_id', 'in', empids)]
 363         # if this user is a hr.manager, he should do second validations
 364         if self.pool.get('res.users').has_group(cr, uid, 'base.group_hr_manager'):
 365             dom = ['|'] + dom + [('state', '=', 'validate1')]
 366         return dom
 367
 368     def create_notificate(self, cr, uid, ids, context=None):
 369         for obj in self.browse(cr, uid, ids, context=context):
 370             self.message_post(cr, uid, ids,
 371                 _("Request <b>created</b>, waiting confirmation."), context=context)
 372         return True
 373
 374     def holidays_confirm_notificate(self, cr, uid, ids, context=None):
 375         for obj in self.browse(cr, uid, ids):
 376             self.message_post(cr, uid, [obj.id],
 377                 _("Request <b>submitted</b>, waiting for validation by the manager."), context=context)
 378
 379     def holidays_first_validate_notificate(self, cr, uid, ids, context=None):
 380         for obj in self.browse(cr, uid, ids, context=context):
 381             self.message_post(cr, uid, [obj.id],
 382                 _("Request <b>approved</b>, waiting second validation."), context=context)
 383
 384     def holidays_validate_notificate(self, cr, uid, ids, context=None):
 385         for obj in self.browse(cr, uid, ids):
 386             if obj.double_validation:
 387                 self.message_post(cr, uid, [obj.id],
 388                     _("Request <b>validated</b>."), context=context)
 389             else:
 390                 self.message_post(cr, uid, [obj.id],
 391                     _("The request has been <b>approved</b>."), context=context)
 392
 393     def holidays_refuse_notificate(self, cr, uid, ids, context=None):
 394         for obj in self.browse(cr, uid, ids):
 395             self.message_post(cr, uid, [obj.id],
 396                 _("Request <b>refused</b>"), context=context)
 397
 398
 399 class resource_calendar_leaves(osv.osv):
 400     _inherit = "resource.calendar.leaves"
 401     _description = "Leave Detail"
 402     _columns = {
 403         'holiday_id': fields.many2one("hr.holidays", "Leave Request"),
 404     }
 405
 406 resource_calendar_leaves()
 407
 408
 409 class hr_employee(osv.osv):
 410     _inherit="hr.employee"
 411
 412     def create(self, cr, uid, vals, context=None):
 413         # don't pass the value of remaining leave if it's 0 at the creation time, otherwise it will trigger the inverse
 414         # function _set_remaining_days and the system may not be configured for. Note that we don't have this problem on
 415         # the write because the clients only send the fields that have been modified.
 416         if 'remaining_leaves' in vals and not vals['remaining_leaves']:
 417             del(vals['remaining_leaves'])
 418         return super(hr_employee, self).create(cr, uid, vals, context=context)
 419
 420     def _set_remaining_days(self, cr, uid, empl_id, name, value, arg, context=None):
 421         employee = self.browse(cr, uid, empl_id, context=context)
 422         diff = value - employee.remaining_leaves
 423         type_obj = self.pool.get('hr.holidays.status')
 424         holiday_obj = self.pool.get('hr.holidays')
 425         # Find for holidays status
 426         status_ids = type_obj.search(cr, uid, [('limit', '=', False)], context=context)
 427         if len(status_ids) != 1 :
 428             raise osv.except_osv(_('Warning!'),_("The feature behind the field 'Remaining Legal Leaves' can only be used when there is only one leave type with the option 'Allow to Override Limit' unchecked. (%s Found). Otherwise, the update is ambiguous as we cannot decide on which leave type the update has to be done. \nYou may prefer to use the classic menus 'Leave Requests' and 'Allocation Requests' located in 'Human Resources \ Leaves' to manage the leave days of the employees if the configuration does not allow to use this field.") % (len(status_ids)))
 429         status_id = status_ids and status_ids[0] or False
 430         if not status_id:
 431             return False
 432         if diff > 0:
 433             leave_id = holiday_obj.create(cr, uid, {'name': _('Allocation for %s') % employee.name, 'employee_id': employee.id, 'holiday_status_id': status_id, 'type': 'add', 'holiday_type': 'employee', 'number_of_days_temp': diff}, context=context)
 434         elif diff < 0:
 435             leave_id = holiday_obj.create(cr, uid, {'name': _('Leave Request for %s') % employee.name, 'employee_id': employee.id, 'holiday_status_id': status_id, 'type': 'remove', 'holiday_type': 'employee', 'number_of_days_temp': abs(diff)}, context=context)
 436         else:
 437             return False
 438         wf_service = netsvc.LocalService("workflow")
 439         wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'confirm', cr)
 440         wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'validate', cr)
 441         wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'second_validate', cr)
 442         return True
 443
 444     def _get_remaining_days(self, cr, uid, ids, name, args, context=None):
 445         cr.execute("""SELECT
 446                 sum(h.number_of_days) as days,
 447                 h.employee_id
 448             from
 449                 hr_holidays h
 450                 join hr_holidays_status s on (s.id=h.holiday_status_id)
 451             where
 452                 h.state='validate' and
 453                 s.limit=False and
 454                 h.employee_id in (%s)
 455             group by h.employee_id"""% (','.join(map(str,ids)),) )
 456         res = cr.dictfetchall()
 457         remaining = {}
 458         for r in res:
 459             remaining[r['employee_id']] = r['days']
 460         for employee_id in ids:
 461             if not remaining.get(employee_id):
 462                 remaining[employee_id] = 0.0
 463         return remaining
 464
 465     def _get_leave_status(self, cr, uid, ids, name, args, context=None):
 466         holidays_obj = self.pool.get('hr.holidays')
 467         holidays_id = holidays_obj.search(cr, uid,
 468            [('employee_id', 'in', ids), ('date_from','<=',time.strftime('%Y-%m-%d %H:%M:%S')),
 469            ('date_to','>=',time.strftime('%Y-%m-%d 23:59:59')),('type','=','remove'),('state','not in',('cancel','refuse'))],
 470            context=context)
 471         result = {}
 472         for id in ids:
 473             result[id] = {
 474                 'current_leave_state': False,
 475                 'current_leave_id': False,
 476                 'leave_date_from':False,
 477                 'leave_date_to':False,
 478             }
 479         for holiday in self.pool.get('hr.holidays').browse(cr, uid, holidays_id, context=context):
 480             result[holiday.employee_id.id]['leave_date_from'] = holiday.date_from
 481             result[holiday.employee_id.id]['leave_date_to'] = holiday.date_to
 482             result[holiday.employee_id.id]['current_leave_state'] = holiday.state
 483             result[holiday.employee_id.id]['current_leave_id'] = holiday.holiday_status_id.id
 484         return result
 485
 486     _columns = {
 487         'remaining_leaves': fields.function(_get_remaining_days, string='Remaining Legal Leaves', fnct_inv=_set_remaining_days, type="float", help='Total number of legal leaves allocated to this employee, change this value to create allocation/leave requests.'),
 488         'current_leave_state': fields.function(_get_leave_status, multi="leave_status", string="Current Leave Status", type="selection",
 489             selection=[('draft', 'New'), ('confirm', 'Waiting Approval'), ('refuse', 'Refused'),
 490             ('validate1', 'Waiting Second Approval'), ('validate', 'Approved'), ('cancel', 'Cancelled')]),
 491         'current_leave_id': fields.function(_get_leave_status, multi="leave_status", string="Current Leave Type",type='many2one', relation='hr.holidays.status'),
 492         'leave_date_from': fields.function(_get_leave_status, multi='leave_status', type='date', string='From Date'),
 493         'leave_date_to': fields.function(_get_leave_status, multi='leave_status', type='date', string='To Date'),
 494     }
 495
 496 hr_employee()
 497
 498 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: