1 # -*- coding: utf-8 -*-
2 ##################################################################################
4 # Copyright (c) 2005-2006 Axelor SARL. (http://www.axelor.com)
5 # and 2004-2010 Tiny SPRL (<http://tiny.be>).
7 # $Id: hr.py 4656 2006-11-24 09:58:42Z Cyp $
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.
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.
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/>.
22 ##############################################################################
27 from operator import attrgetter
29 from openerp import tools
30 from openerp.osv import fields, osv
31 from openerp.tools.translate import _
34 class hr_holidays_status(osv.osv):
35 _name = "hr.holidays.status"
36 _description = "Leave Type"
38 def get_days(self, cr, uid, ids, employee_id, context=None):
39 result = dict((id, dict(max_leaves=0, leaves_taken=0, remaining_leaves=0,
40 virtual_remaining_leaves=0)) for id in ids)
41 holiday_ids = self.pool['hr.holidays'].search(cr, uid, [('employee_id', '=', employee_id),
42 ('state', 'in', ['confirm', 'validate1', 'validate']),
43 ('holiday_status_id', 'in', ids)
45 for holiday in self.pool['hr.holidays'].browse(cr, uid, holiday_ids, context=context):
46 status_dict = result[holiday.holiday_status_id.id]
47 if holiday.type == 'add':
48 status_dict['virtual_remaining_leaves'] += holiday.number_of_days
49 if holiday.state == 'validate':
50 status_dict['max_leaves'] += holiday.number_of_days
51 status_dict['remaining_leaves'] += holiday.number_of_days
52 elif holiday.type == 'remove': # number of days is negative
53 status_dict['virtual_remaining_leaves'] += holiday.number_of_days
54 if holiday.state == 'validate':
55 status_dict['leaves_taken'] -= holiday.number_of_days
56 status_dict['remaining_leaves'] += holiday.number_of_days
59 def _user_left_days(self, cr, uid, ids, name, args, context=None):
61 if context and 'employee_id' in context:
62 employee_id = context['employee_id']
64 employee_ids = self.pool.get('hr.employee').search(cr, uid, [('user_id', '=', uid)], context=context)
66 employee_id = employee_ids[0]
68 res = self.get_days(cr, uid, ids, employee_id, context=context)
70 res = dict.fromkeys(ids, {'leaves_taken': 0, 'remaining_leaves': 0, 'max_leaves': 0})
74 'name': fields.char('Leave Type', size=64, required=True, translate=True),
75 'categ_id': fields.many2one('crm.meeting.type', 'Meeting Type',
76 help='Once a leave is validated, OpenERP will create a corresponding meeting of this type in the calendar.'),
77 '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.'),
78 'limit': fields.boolean('Allow to Override Limit', help='If you select this check box, the system allows the employees to take more leaves than the available ones for this type and will not take them into account for the "Remaining Legal Leaves" defined on the employee form.'),
79 '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."),
80 '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'),
81 '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'),
82 'remaining_leaves': fields.function(_user_left_days, string='Remaining Leaves', help='Maximum Leaves Allowed - Leaves Already Taken', multi='user_left_days'),
83 'virtual_remaining_leaves': fields.function(_user_left_days, string='Virtual Remaining Leaves', help='Maximum Leaves Allowed - Leaves Already Taken - Leaves Waiting Approval', multi='user_left_days'),
84 'double_validation': fields.boolean('Apply Double Validation', help="When selected, the Allocation/Leave Requests for this type require a second validation to be approved."),
91 def name_get(self, cr, uid, ids, context=None):
93 for record in self.browse(cr, uid, ids, context=context):
96 name = name + (' (%d/%d)' % (record.leaves_taken or 0.0, record.max_leaves or 0.0))
97 res.append((record.id, name))
101 class hr_holidays(osv.osv):
102 _name = "hr.holidays"
103 _description = "Leave"
104 _order = "type desc, date_from asc"
105 _inherit = ['mail.thread', 'ir.needaction_mixin']
108 'hr_holidays.mt_holidays_approved': lambda self, cr, uid, obj, ctx=None: obj.state == 'validate',
109 'hr_holidays.mt_holidays_refused': lambda self, cr, uid, obj, ctx=None: obj.state == 'refuse',
110 'hr_holidays.mt_holidays_confirmed': lambda self, cr, uid, obj, ctx=None: obj.state == 'confirm',
114 def _employee_get(self, cr, uid, context=None):
115 emp_id = context.get('default_employee_id', False)
118 ids = self.pool.get('hr.employee').search(cr, uid, [('user_id', '=', uid)], context=context)
123 def _compute_number_of_days(self, cr, uid, ids, name, args, context=None):
125 for hol in self.browse(cr, uid, ids, context=context):
126 if hol.type=='remove':
127 result[hol.id] = -hol.number_of_days_temp
129 result[hol.id] = hol.number_of_days_temp
132 def _check_date(self, cr, uid, ids):
133 for holiday in self.browse(cr, uid, ids):
134 holiday_ids = self.search(cr, uid, [('date_from', '<=', holiday.date_to), ('date_to', '>=', holiday.date_from), ('employee_id', '=', holiday.employee_id.id), ('id', '<>', holiday.id)])
140 'name': fields.char('Description', size=64),
141 'state': fields.selection([('draft', 'To Submit'), ('cancel', 'Cancelled'),('confirm', 'To Approve'), ('refuse', 'Refused'), ('validate1', 'Second Approval'), ('validate', 'Approved')],
142 'Status', readonly=True, track_visibility='onchange',
143 help='The status is set to \'To Submit\', when a holiday request is created.\
144 \nThe status is \'To Approve\', when holiday request is confirmed by user.\
145 \nThe status is \'Refused\', when holiday request is refused by manager.\
146 \nThe status is \'Approved\', when holiday request is approved by manager.'),
147 'user_id':fields.related('employee_id', 'user_id', type='many2one', relation='res.users', string='User', store=True),
148 'date_from': fields.datetime('Start Date', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}, select=True),
149 'date_to': fields.datetime('End Date', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
150 'holiday_status_id': fields.many2one("hr.holidays.status", "Leave Type", required=True,readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
151 'employee_id': fields.many2one('hr.employee', "Employee", select=True, invisible=False, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
152 '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'),
153 'notes': fields.text('Reasons',readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
154 'number_of_days_temp': fields.float('Allocation', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
155 'number_of_days': fields.function(_compute_number_of_days, string='Number of Days', store=True),
156 'meeting_id': fields.many2one('crm.meeting', 'Meeting'),
157 '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),
158 'parent_id': fields.many2one('hr.holidays', 'Parent'),
159 'linked_request_ids': fields.one2many('hr.holidays', 'parent_id', 'Linked Requests',),
160 'department_id':fields.related('employee_id', 'department_id', string='Department', type='many2one', relation='hr.department', readonly=True, store=True),
161 'category_id': fields.many2one('hr.employee.category', "Employee Tag", help='Category of Employee', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
162 'holiday_type': fields.selection([('employee','By Employee'),('category','By Employee Tag')], 'Allocation Mode', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}, help='By Employee: Allocation/Request for individual Employee, By Employee Tag: Allocation/Request for group of employees in category', required=True),
163 '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)'),
164 'double_validation': fields.related('holiday_status_id', 'double_validation', type='boolean', relation='hr.holidays.status', string='Apply Double Validation'),
167 'employee_id': _employee_get,
170 'user_id': lambda obj, cr, uid, context: uid,
171 'holiday_type': 'employee'
174 (_check_date, 'You can not have 2 leaves that overlaps on same day!', ['date_from','date_to']),
178 ('type_value', "CHECK( (holiday_type='employee' AND employee_id IS NOT NULL) or (holiday_type='category' AND category_id IS NOT NULL))",
179 "The employee or employee category of this request is missing. Please make sure that your user login is linked to an employee."),
180 ('date_check2', "CHECK ( (type='add') OR (date_from <= date_to))", "The start date must be anterior to the end date."),
181 ('date_check', "CHECK ( number_of_days_temp >= 0 )", "The number of days must be greater than 0."),
184 def copy(self, cr, uid, id, default=None, context=None):
189 default = default.copy()
190 default['date_from'] = False
191 default['date_to'] = False
192 return super(hr_holidays, self).copy(cr, uid, id, default, context=context)
194 def _create_resource_leave(self, cr, uid, leaves, context=None):
195 '''This method will create entry in resource calendar leave object at the time of holidays validated '''
196 obj_res_leave = self.pool.get('resource.calendar.leaves')
200 'date_from': leave.date_from,
201 'holiday_id': leave.id,
202 'date_to': leave.date_to,
203 'resource_id': leave.employee_id.resource_id.id,
204 'calendar_id': leave.employee_id.resource_id.calendar_id.id
206 obj_res_leave.create(cr, uid, vals, context=context)
209 def _remove_resource_leave(self, cr, uid, ids, context=None):
210 '''This method will create entry in resource calendar leave object at the time of holidays cancel/removed'''
211 obj_res_leave = self.pool.get('resource.calendar.leaves')
212 leave_ids = obj_res_leave.search(cr, uid, [('holiday_id', 'in', ids)], context=context)
213 return obj_res_leave.unlink(cr, uid, leave_ids, context=context)
215 def onchange_type(self, cr, uid, ids, holiday_type, employee_id=False, context=None):
217 if holiday_type == 'employee' and not employee_id:
218 ids_employee = self.pool.get('hr.employee').search(cr, uid, [('user_id','=', uid)])
221 'employee_id': ids_employee[0]
225 def onchange_employee(self, cr, uid, ids, employee_id):
226 result = {'value': {'department_id': False}}
228 employee = self.pool.get('hr.employee').browse(cr, uid, employee_id)
229 result['value'] = {'department_id': employee.department_id.id}
232 # TODO: can be improved using resource calendar method
233 def _get_number_of_days(self, date_from, date_to):
234 """Returns a float equals to the timedelta between two dates given as string."""
236 DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"
237 from_dt = datetime.datetime.strptime(date_from, DATETIME_FORMAT)
238 to_dt = datetime.datetime.strptime(date_to, DATETIME_FORMAT)
239 timedelta = to_dt - from_dt
240 diff_day = timedelta.days + float(timedelta.seconds) / 86400
243 def unlink(self, cr, uid, ids, context=None):
244 for rec in self.browse(cr, uid, ids, context=context):
245 if rec.state not in ['draft', 'cancel', 'confirm']:
246 raise osv.except_osv(_('Warning!'),_('You cannot delete a leave which is in %s state.')%(rec.state))
247 return super(hr_holidays, self).unlink(cr, uid, ids, context)
249 def onchange_date_from(self, cr, uid, ids, date_to, date_from):
251 If there are no date set for date_to, automatically set one 8 hours later than
253 Also update the number_of_days.
255 # date_to has to be greater than date_from
256 if (date_from and date_to) and (date_from > date_to):
257 raise osv.except_osv(_('Warning!'),_('The start date must be anterior to the end date.'))
259 result = {'value': {}}
261 # No date_to set so far: automatically compute one 8 hours later
262 if date_from and not date_to:
263 date_to_with_delta = datetime.datetime.strptime(date_from, tools.DEFAULT_SERVER_DATETIME_FORMAT) + datetime.timedelta(hours=8)
264 result['value']['date_to'] = str(date_to_with_delta)
266 # Compute and update the number of days
267 if (date_to and date_from) and (date_from <= date_to):
268 diff_day = self._get_number_of_days(date_from, date_to)
269 result['value']['number_of_days_temp'] = round(math.floor(diff_day))+1
271 result['value']['number_of_days_temp'] = 0
275 def onchange_date_to(self, cr, uid, ids, date_to, date_from):
277 Update the number_of_days.
280 # date_to has to be greater than date_from
281 if (date_from and date_to) and (date_from > date_to):
282 raise osv.except_osv(_('Warning!'),_('The start date must be anterior to the end date.'))
284 result = {'value': {}}
286 # Compute and update the number of days
287 if (date_to and date_from) and (date_from <= date_to):
288 diff_day = self._get_number_of_days(date_from, date_to)
289 result['value']['number_of_days_temp'] = round(math.floor(diff_day))+1
291 result['value']['number_of_days_temp'] = 0
295 def create(self, cr, uid, values, context=None):
296 """ Override to avoid automatic logging of creation """
299 context = dict(context, mail_create_nolog=True)
300 return super(hr_holidays, self).create(cr, uid, values, context=context)
302 def write(self, cr, uid, ids, vals, context=None):
303 check_fnct = self.pool.get('hr.holidays.status').check_access_rights
304 for holiday in self.browse(cr, uid, ids, context=context):
305 if holiday.state in ('validate','validate1') and not check_fnct(cr, uid, 'write', raise_exception=False):
306 raise osv.except_osv(_('Warning!'),_('You cannot modify a leave request that has been approved. Contact a human resource manager.'))
307 return super(hr_holidays, self).write(cr, uid, ids, vals, context=context)
309 def set_to_draft(self, cr, uid, ids, context=None):
310 self.write(cr, uid, ids, {
313 'manager_id2': False,
315 self.delete_workflow(cr, uid, ids)
316 self.create_workflow(cr, uid, ids)
318 for record in self.browse(cr, uid, ids, context=context):
319 for record2 in record.linked_request_ids:
320 self.set_to_draft(cr, uid, [record2.id], context=context)
321 to_unlink.append(record2.id)
323 self.unlink(cr, uid, to_unlink, context=context)
326 def holidays_first_validate(self, cr, uid, ids, context=None):
327 self.check_holidays(cr, uid, ids, context=context)
328 obj_emp = self.pool.get('hr.employee')
329 ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
330 manager = ids2 and ids2[0] or False
331 self.holidays_first_validate_notificate(cr, uid, ids, context=context)
332 return self.write(cr, uid, ids, {'state':'validate1', 'manager_id': manager})
334 def holidays_validate(self, cr, uid, ids, context=None):
335 self.check_holidays(cr, uid, ids, context=context)
336 obj_emp = self.pool.get('hr.employee')
337 ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
338 manager = ids2 and ids2[0] or False
339 self.write(cr, uid, ids, {'state':'validate'})
340 data_holiday = self.browse(cr, uid, ids)
341 for record in data_holiday:
342 if record.double_validation:
343 self.write(cr, uid, [record.id], {'manager_id2': manager})
345 self.write(cr, uid, [record.id], {'manager_id': manager})
346 if record.holiday_type == 'employee' and record.type == 'remove':
347 meeting_obj = self.pool.get('crm.meeting')
349 'name': record.name or _('Leave Request'),
350 'categ_ids': record.holiday_status_id.categ_id and [(6,0,[record.holiday_status_id.categ_id.id])] or [],
351 'duration': record.number_of_days_temp * 8,
352 'description': record.notes,
353 'user_id': record.user_id.id,
354 'date': record.date_from,
355 'end_date': record.date_to,
356 'date_deadline': record.date_to,
357 'state': 'open', # to block that meeting date in the calendar
359 meeting_id = meeting_obj.create(cr, uid, meeting_vals)
360 self._create_resource_leave(cr, uid, [record], context=context)
361 self.write(cr, uid, ids, {'meeting_id': meeting_id})
362 elif record.holiday_type == 'category':
363 emp_ids = obj_emp.search(cr, uid, [('category_ids', 'child_of', [record.category_id.id])])
365 for emp in obj_emp.browse(cr, uid, emp_ids):
369 'holiday_type': 'employee',
370 'holiday_status_id': record.holiday_status_id.id,
371 'date_from': record.date_from,
372 'date_to': record.date_to,
373 'notes': record.notes,
374 'number_of_days_temp': record.number_of_days_temp,
375 'parent_id': record.id,
376 'employee_id': emp.id
378 leave_ids.append(self.create(cr, uid, vals, context=None))
379 for leave_id in leave_ids:
380 # TODO is it necessary to interleave the calls?
381 self.signal_confirm(cr, uid, [leave_id])
382 self.signal_validate(cr, uid, [leave_id])
383 self.signal_second_validate(cr, uid, [leave_id])
386 def holidays_confirm(self, cr, uid, ids, context=None):
387 self.check_holidays(cr, uid, ids, context=context)
388 for record in self.browse(cr, uid, ids, context=context):
389 if record.employee_id and record.employee_id.parent_id and record.employee_id.parent_id.user_id:
390 self.message_subscribe_users(cr, uid, [record.id], user_ids=[record.employee_id.parent_id.user_id.id], context=context)
391 return self.write(cr, uid, ids, {'state': 'confirm'})
393 def holidays_refuse(self, cr, uid, ids, context=None):
394 obj_emp = self.pool.get('hr.employee')
395 ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
396 manager = ids2 and ids2[0] or False
397 for holiday in self.browse(cr, uid, ids, context=context):
398 if holiday.state == 'validate1':
399 self.write(cr, uid, [holiday.id], {'state': 'refuse', 'manager_id': manager})
401 self.write(cr, uid, [holiday.id], {'state': 'refuse', 'manager_id2': manager})
402 self.holidays_cancel(cr, uid, ids, context=context)
405 def holidays_cancel(self, cr, uid, ids, context=None):
406 meeting_obj = self.pool.get('crm.meeting')
407 for record in self.browse(cr, uid, ids):
409 if record.meeting_id:
410 meeting_obj.unlink(cr, uid, [record.meeting_id.id])
412 # If a category that created several holidays, cancel all related
413 self.signal_refuse(cr, uid, map(attrgetter('id'), record.linked_request_ids or []))
415 self._remove_resource_leave(cr, uid, ids, context=context)
418 def check_holidays(self, cr, uid, ids, context=None):
419 for record in self.browse(cr, uid, ids, context=context):
420 if record.holiday_type != 'employee' or record.type != 'remove' or not record.employee_id or record.holiday_status_id.limit:
422 leave_days = self.pool.get('hr.holidays.status').get_days(cr, uid, [record.holiday_status_id.id], record.employee_id.id, context=context)[record.holiday_status_id.id]
423 if leave_days['remaining_leaves'] < record.number_of_days_temp:
424 raise osv.except_osv(_('Warning!'),
425 _('There are not enough remaining days available in %s for employee %s.') % (record.holiday_status_id.name, record.employee_id.name))
426 if leave_days['virtual_remaining_leaves'] < record.number_of_days_temp:
427 raise osv.except_osv(_('Warning!'),
428 _('Other pending requests already book too much days in %s for employee %s.') % (record.holiday_status_id.name, record.employee_id.name))
431 # -----------------------------
432 # OpenChatter and notifications
433 # -----------------------------
435 def _needaction_domain_get(self, cr, uid, context=None):
436 emp_obj = self.pool.get('hr.employee')
437 empids = emp_obj.search(cr, uid, [('parent_id.user_id', '=', uid)], context=context)
438 dom = ['&', ('state', '=', 'confirm'), ('employee_id', 'in', empids)]
439 # if this user is a hr.manager, he should do second validations
440 if self.pool.get('res.users').has_group(cr, uid, 'base.group_hr_manager'):
441 dom = ['|'] + dom + [('state', '=', 'validate1')]
444 def holidays_first_validate_notificate(self, cr, uid, ids, context=None):
445 for obj in self.browse(cr, uid, ids, context=context):
446 self.message_post(cr, uid, [obj.id],
447 _("Request approved, waiting second validation."), context=context)
449 class resource_calendar_leaves(osv.osv):
450 _inherit = "resource.calendar.leaves"
451 _description = "Leave Detail"
453 'holiday_id': fields.many2one("hr.holidays", "Leave Request"),
458 class hr_employee(osv.osv):
459 _inherit="hr.employee"
461 def create(self, cr, uid, vals, context=None):
462 # don't pass the value of remaining leave if it's 0 at the creation time, otherwise it will trigger the inverse
463 # function _set_remaining_days and the system may not be configured for. Note that we don't have this problem on
464 # the write because the clients only send the fields that have been modified.
465 if 'remaining_leaves' in vals and not vals['remaining_leaves']:
466 del(vals['remaining_leaves'])
467 return super(hr_employee, self).create(cr, uid, vals, context=context)
469 def _set_remaining_days(self, cr, uid, empl_id, name, value, arg, context=None):
470 employee = self.browse(cr, uid, empl_id, context=context)
471 diff = value - employee.remaining_leaves
472 type_obj = self.pool.get('hr.holidays.status')
473 holiday_obj = self.pool.get('hr.holidays')
474 # Find for holidays status
475 status_ids = type_obj.search(cr, uid, [('limit', '=', False)], context=context)
476 if len(status_ids) != 1 :
477 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)))
478 status_id = status_ids and status_ids[0] or False
482 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)
484 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)
487 holiday_obj.signal_confirm(cr, uid, [leave_id])
488 holiday_obj.signal_validate(cr, uid, [leave_id])
489 holiday_obj.signal_second_validate(cr, uid, [leave_id])
492 def _get_remaining_days(self, cr, uid, ids, name, args, context=None):
494 sum(h.number_of_days) as days,
498 join hr_holidays_status s on (s.id=h.holiday_status_id)
500 h.state='validate' and
502 h.employee_id in (%s)
503 group by h.employee_id"""% (','.join(map(str,ids)),) )
504 res = cr.dictfetchall()
507 remaining[r['employee_id']] = r['days']
508 for employee_id in ids:
509 if not remaining.get(employee_id):
510 remaining[employee_id] = 0.0
513 def _get_leave_status(self, cr, uid, ids, name, args, context=None):
514 holidays_obj = self.pool.get('hr.holidays')
515 holidays_id = holidays_obj.search(cr, uid,
516 [('employee_id', 'in', ids), ('date_from','<=',time.strftime('%Y-%m-%d %H:%M:%S')),
517 ('date_to','>=',time.strftime('%Y-%m-%d 23:59:59')),('type','=','remove'),('state','not in',('cancel','refuse'))],
522 'current_leave_state': False,
523 'current_leave_id': False,
524 'leave_date_from':False,
525 'leave_date_to':False,
527 for holiday in self.pool.get('hr.holidays').browse(cr, uid, holidays_id, context=context):
528 result[holiday.employee_id.id]['leave_date_from'] = holiday.date_from
529 result[holiday.employee_id.id]['leave_date_to'] = holiday.date_to
530 result[holiday.employee_id.id]['current_leave_state'] = holiday.state
531 result[holiday.employee_id.id]['current_leave_id'] = holiday.holiday_status_id.id
535 '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 request. Total based on all the leave types without overriding limit.'),
536 'current_leave_state': fields.function(_get_leave_status, multi="leave_status", string="Current Leave Status", type="selection",
537 selection=[('draft', 'New'), ('confirm', 'Waiting Approval'), ('refuse', 'Refused'),
538 ('validate1', 'Waiting Second Approval'), ('validate', 'Approved'), ('cancel', 'Cancelled')]),
539 'current_leave_id': fields.function(_get_leave_status, multi="leave_status", string="Current Leave Type",type='many2one', relation='hr.holidays.status'),
540 'leave_date_from': fields.function(_get_leave_status, multi='leave_status', type='date', string='From Date'),
541 'leave_date_to': fields.function(_get_leave_status, multi='leave_status', type='date', string='To Date'),
545 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: