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 datetime import datetime
  26 from itertools import groupby
  27 from operator import itemgetter
  28
  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.case.categ', 'Meeting', domain="[('object_id.model', '=', 'crm.meeting')]", help='If you set a meeting type, OpenERP will create a meeting in the calendar once a leave is validated.'),
  79         '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 Departement'),
  80         'limit': fields.boolean('Allow to Override Limit', help='If you tick this checkbox, the system will allow, for this section, the employees to take more leaves than the available ones.'),
  81         '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."),
  82         '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'),
  83         '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'),
  84         'remaining_leaves': fields.function(_user_left_days, string='Remaining Leaves', help='Maximum Leaves Allowed - Leaves Already Taken', multi='user_left_days'),
  85         'double_validation': fields.boolean('Apply Double Validation', help="If its True then its Allocation/Request have to be validated by second validator")
  86     }
  87     _defaults = {
  88         'color_name': 'red',
  89         'active': True,
  90     }
  91 hr_holidays_status()
  92
  93 class hr_holidays(osv.osv):
  94     _name = "hr.holidays"
  95     _description = "Leave"
  96     _order = "type desc, date_from asc"
  97     _inherit = ['ir.needaction_mixin', 'mail.thread']
  98
  99     def _employee_get(self, cr, uid, context=None):
 100         ids = self.pool.get('hr.employee').search(cr, uid, [('user_id', '=', uid)], context=context)
 101         if ids:
 102             return ids[0]
 103         return False
 104
 105     def _compute_number_of_days(self, cr, uid, ids, name, args, context=None):
 106         result = {}
 107         for hol in self.browse(cr, uid, ids, context=context):
 108             if hol.type=='remove':
 109                 result[hol.id] = -hol.number_of_days_temp
 110             else:
 111                 result[hol.id] = hol.number_of_days_temp
 112         return result
 113
 114     _columns = {
 115         'name': fields.char('Description', required=True, size=64),
 116         'state': fields.selection([('draft', 'New'), ('confirm', 'Waiting Approval'), ('refuse', 'Refused'),
 117             ('validate1', 'Waiting Second Approval'), ('validate', 'Approved'), ('cancel', 'Cancelled')],
 118             'State', readonly=True, help='The state is set to \'Draft\', when a holiday request is created.\
 119             \nThe state is \'Waiting Approval\', when holiday request is confirmed by user.\
 120             \nThe state is \'Refused\', when holiday request is refused by manager.\
 121             \nThe state 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)]}, select=True),
 124         'date_to': fields.datetime('End Date', readonly=True, states={'draft':[('readonly',False)]}),
 125         'holiday_status_id': fields.many2one("hr.holidays.status", "Leave Type", required=True,readonly=True, states={'draft':[('readonly',False)]}),
 126         'employee_id': fields.many2one('hr.employee', "Employee", select=True, invisible=False, readonly=True, states={'draft':[('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', 'Leave Manager', invisible=False, readonly=True, help='This area is automatically filled by the user who validate the leave'),
 128         #'notes': fields.text('Notes',readonly=True, states={'draft':[('readonly',False)]}),
 129         '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'),
 130         'notes': fields.text('Reasons',readonly=True, states={'draft':[('readonly',False)]}),
 131         'number_of_days_temp': fields.float('Number of Days', readonly=True, states={'draft':[('readonly',False)]}),
 132         'number_of_days': fields.function(_compute_number_of_days, string='Number of Days', store=True),
 133         'case_id': fields.many2one('crm.meeting', 'Meeting'),
 134         'type': fields.selection([('remove','Leave Request'),('add','Allocation Request')], 'Request Type', required=True, readonly=True, states={'draft':[('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),
 135         'parent_id': fields.many2one('hr.holidays', 'Parent'),
 136         'linked_request_ids': fields.one2many('hr.holidays', 'parent_id', 'Linked Requests',),
 137         'department_id':fields.related('employee_id', 'department_id', string='Department', type='many2one', relation='hr.department', readonly=True, store=True),
 138         'category_id': fields.many2one('hr.employee.category', "Category", help='Category of Employee'),
 139         'holiday_type': fields.selection([('employee','By Employee'),('category','By Employee Category')], 'Allocation Type', help='By Employee: Allocation/Request for individual Employee, By Employee Category: Allocation/Request for group of employees in category', required=True),
 140         '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)'),
 141         'double_validation': fields.related('holiday_status_id', 'double_validation', type='boolean', relation='hr.holidays.status', string='Apply Double Validation'),
 142     }
 143     _defaults = {
 144         'employee_id': _employee_get,
 145         'state': 'draft',
 146         'type': 'remove',
 147         'user_id': lambda obj, cr, uid, context: uid,
 148         'holiday_type': 'employee'
 149     }
 150     _sql_constraints = [
 151         ('type_value', "CHECK( (holiday_type='employee' AND employee_id IS NOT NULL) or (holiday_type='category' AND category_id IS NOT NULL))", "You have to select an employee or a category"),
 152         ('date_check2', "CHECK ( (type='add') OR (date_from <= date_to))", "The start date must be before the end date !"),
 153         ('date_check', "CHECK ( number_of_days_temp >= 0 )", "The number of days must be greater than 0 !"),
 154     ]
 155
 156     def create(self, cr, uid, vals, context=None):
 157         obj_id = super(hr_holidays, self).create(cr, uid, vals, context=context)
 158         self.create_notificate(cr, uid, [obj_id], context=context)
 159         return obj_id
 160
 161     def _create_resource_leave(self, cr, uid, leaves, context=None):
 162         '''This method will create entry in resource calendar leave object at the time of holidays validated '''
 163         obj_res_leave = self.pool.get('resource.calendar.leaves')
 164         for leave in leaves:
 165             vals = {
 166                 'name': leave.name,
 167                 'date_from': leave.date_from,
 168                 'holiday_id': leave.id,
 169                 'date_to': leave.date_to,
 170                 'resource_id': leave.employee_id.resource_id.id,
 171                 'calendar_id': leave.employee_id.resource_id.calendar_id.id
 172             }
 173             obj_res_leave.create(cr, uid, vals, context=context)
 174         return True
 175
 176     def _remove_resource_leave(self, cr, uid, ids, context=None):
 177         '''This method will create entry in resource calendar leave object at the time of holidays cancel/removed'''
 178         obj_res_leave = self.pool.get('resource.calendar.leaves')
 179         leave_ids = obj_res_leave.search(cr, uid, [('holiday_id', 'in', ids)], context=context)
 180         return obj_res_leave.unlink(cr, uid, leave_ids, context=context)
 181
 182     def onchange_type(self, cr, uid, ids, holiday_type):
 183         result = {'value': {'employee_id': False}}
 184         if holiday_type == 'employee':
 185             ids_employee = self.pool.get('hr.employee').search(cr, uid, [('user_id','=', uid)])
 186             if ids_employee:
 187                 result['value'] = {
 188                     'employee_id': ids_employee[0]
 189                 }
 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<>'draft':
 206                 raise osv.except_osv(_('Warning!'),_('You cannot delete a leave which is not in draft state !'))
 207         return super(hr_holidays, self).unlink(cr, uid, ids, context)
 208
 209     def onchange_date_from(self, cr, uid, ids, date_to, date_from):
 210         result = {}
 211         if date_to and date_from:
 212             diff_day = self._get_number_of_days(date_from, date_to)
 213             result['value'] = {
 214                 'number_of_days_temp': round(diff_day)+1
 215             }
 216             return result
 217         result['value'] = {
 218             'number_of_days_temp': 0,
 219         }
 220         return result
 221
 222     def onchange_sec_id(self, cr, uid, ids, status, context=None):
 223         warning = {}
 224         double_validation = False
 225         obj_holiday_status = self.pool.get('hr.holidays.status')
 226         if status:
 227             holiday_status = obj_holiday_status.browse(cr, uid, status, context=context)
 228             double_validation = holiday_status.double_validation
 229             if holiday_status.categ_id and holiday_status.categ_id.section_id and not holiday_status.categ_id.section_id.allow_unlink:
 230                 warning = {
 231                     'title': "Warning for ",
 232                     'message': "You won\'t be able to cancel this leave request because the CRM Sales Team of the leave type disallows."
 233                 }
 234         return {'warning': warning, 'value': {'double_validation': double_validation}}
 235
 236     def set_to_draft(self, cr, uid, ids, context=None):
 237         self.write(cr, uid, ids, {
 238             'state': 'draft',
 239             'manager_id': False,
 240             'manager_id2': False,
 241         })
 242         wf_service = netsvc.LocalService("workflow")
 243         for id in ids:
 244             wf_service.trg_delete(uid, 'hr.holidays', id, cr)
 245             wf_service.trg_create(uid, 'hr.holidays', id, cr)
 246         to_unlink = []
 247         for record in self.browse(cr, uid, ids, context=context):
 248             for record2 in record.linked_request_ids:
 249                 self.set_to_draft(cr, uid, [record2.id], context=context)
 250                 to_unlink.append(record2.id)
 251         if to_unlink:
 252             self.unlink(cr, uid, to_unlink, context=context)
 253         return True
 254
 255     def holidays_validate(self, cr, uid, ids, context=None):
 256         self.check_holidays(cr, uid, ids, context=context)
 257         obj_emp = self.pool.get('hr.employee')
 258         ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
 259         manager = ids2 and ids2[0] or False
 260         self.holidays_validate_notificate(cr, uid, ids, context=context)
 261         return self.write(cr, uid, ids, {'state':'validate1', 'manager_id': manager})
 262
 263     def holidays_validate2(self, cr, uid, ids, context=None):
 264         self.check_holidays(cr, uid, ids, context=context)
 265         obj_emp = self.pool.get('hr.employee')
 266         ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
 267         manager = ids2 and ids2[0] or False
 268         self.write(cr, uid, ids, {'state':'validate'})
 269         data_holiday = self.browse(cr, uid, ids)
 270         holiday_ids = []
 271         for record in data_holiday:
 272             if record.holiday_status_id.double_validation:
 273                 holiday_ids.append(record.id)
 274             if record.holiday_type == 'employee' and record.type == 'remove':
 275                 meeting_obj = self.pool.get('crm.meeting')
 276                 vals = {
 277                     'name': record.name,
 278                     'categ_id': record.holiday_status_id.categ_id.id,
 279                     'duration': record.number_of_days_temp * 8,
 280                     'description': record.notes,
 281                     'user_id': record.user_id.id,
 282                     'date': record.date_from,
 283                     'end_date': record.date_to,
 284                     'date_deadline': record.date_to,
 285                 }
 286                 case_id = meeting_obj.create(cr, uid, vals)
 287                 self._create_resource_leave(cr, uid, [record], context=context)
 288                 self.write(cr, uid, ids, {'case_id': case_id})
 289             elif record.holiday_type == 'category':
 290                 emp_ids = obj_emp.search(cr, uid, [('category_ids', 'child_of', [record.category_id.id])])
 291                 leave_ids = []
 292                 for emp in obj_emp.browse(cr, uid, emp_ids):
 293                     vals = {
 294                         'name': record.name,
 295                         'type': record.type,
 296                         'holiday_type': 'employee',
 297                         'holiday_status_id': record.holiday_status_id.id,
 298                         'date_from': record.date_from,
 299                         'date_to': record.date_to,
 300                         'notes': record.notes,
 301                         'number_of_days_temp': record.number_of_days_temp,
 302                         'parent_id': record.id,
 303                         'employee_id': emp.id
 304                     }
 305                     leave_ids.append(self.create(cr, uid, vals, context=None))
 306                 wf_service = netsvc.LocalService("workflow")
 307                 for leave_id in leave_ids:
 308                     wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'confirm', cr)
 309                     wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'validate', cr)
 310                     wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'second_validate', cr)
 311         if holiday_ids:
 312             self.holidays_valid2_notificate(self, cr, uid, [holiday_ids], context=context)
 313             self.write(cr, uid, holiday_ids, {'manager_id2': manager})
 314         return True
 315
 316     def holidays_confirm(self, cr, uid, ids, context=None):
 317         self.check_holidays(cr, uid, ids, context=context)
 318         self.holidays_confirm_notificate(cr, uid, ids, context=context)
 319         return self.write(cr, uid, ids, {'state':'confirm'})
 320
 321     def holidays_refuse(self, cr, uid, ids, approval, context=None):
 322         obj_emp = self.pool.get('hr.employee')
 323         ids2 = obj_emp.search(cr, uid, [('user_id', '=', uid)])
 324         manager = ids2 and ids2[0] or False
 325         if approval == 'first_approval':
 326             self.write(cr, uid, ids, {'state': 'refuse', 'manager_id': manager})
 327         else:
 328             self.write(cr, uid, ids, {'state': 'refuse', 'manager_id2': manager})
 329         self.holidays_refuse_notificate(cr, uid, ids, approval, context=context)
 330         self.holidays_cancel(cr, uid, ids, context=context)
 331         return True
 332
 333     def holidays_cancel(self, cr, uid, ids, context=None):
 334         obj_crm_meeting = self.pool.get('crm.meeting')
 335         for record in self.browse(cr, uid, ids):
 336             # Delete the meeting
 337             if record.case_id:
 338                 obj_crm_meeting.unlink(cr, uid, [record.case_id.id])
 339
 340             # If a category that created several holidays, cancel all related
 341             wf_service = netsvc.LocalService("workflow")
 342             for request in record.linked_request_ids or []:
 343                 wf_service.trg_validate(uid, 'hr.holidays', request.id, 'cancel', cr)
 344
 345         self._remove_resource_leave(cr, uid, ids, context=context)
 346         return True
 347
 348     def check_holidays(self, cr, uid, ids, context=None):
 349         holi_status_obj = self.pool.get('hr.holidays.status')
 350         for record in self.browse(cr, uid, ids):
 351             if record.holiday_type == 'employee' and record.type == 'remove':
 352                 if record.employee_id and not record.holiday_status_id.limit:
 353                     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']
 354                     if leaves_rest < record.number_of_days_temp:
 355                         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))
 356         return True
 357
 358     # -----------------------------
 359     # OpenChatter and notifications
 360     # -----------------------------
 361
 362     def get_needaction_user_ids(self, cr, uid, ids, context=None):
 363         result = dict.fromkeys(ids, [])
 364         for obj in self.browse(cr, uid, ids, context=context):
 365             if obj.state == 'confirm' and obj.employee_id.parent_id:
 366                 result[obj.id] = [obj.employee_id.parent_id.user_id.id]
 367             elif obj.state == 'validate1':
 368                 # get group_hr_manager: everyone will be warned of second validation
 369                 res = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'base', 'group_hr_manager') or False
 370                 obj_id = res and res[1] or False
 371                 if obj_id:
 372                     hr_manager_group = self.pool.get('res.groups').read(cr, uid, [obj_id], ['users'], context=context)[0]
 373                     result[obj.id] = hr_manager_group['users']
 374         return result
 375
 376     def message_get_subscribers(self, cr, uid, ids, context=None):
 377         sub_ids = self.message_get_subscribers_ids(cr, uid, ids, context=context);
 378         # add the employee and its manager if specified to the subscribed users
 379         for obj in self.browse(cr, uid, ids, context=context):
 380             if obj.employee_id.parent_id:
 381                 sub_ids.append(obj.employee_id.parent_id.user_id.id)
 382         return self.pool.get('res.users').read(cr, uid, sub_ids, context=context)
 383
 384     def create_notificate(self, cr, uid, ids, context=None):
 385         for obj in self.browse(cr, uid, ids, context=context):
 386             self.message_append_note(cr, uid, ids, _('System notification'),
 387                         _("The %s request has been <b>created</b> and is waiting confirmation")
 388                         % ('leave' if obj.type == 'remove' else 'allocation',), type='notification', context=context)
 389         return True
 390
 391     def holidays_confirm_notificate(self, cr, uid, ids, context=None):
 392         for obj in self.browse(cr, uid, ids):
 393             self.message_append_note(cr, uid, [obj.id], _('System notification'),
 394                     _("The %s request has been <b>confirmed</b> and is waiting for validation by the manager.")
 395                     % ('leave' if obj.type == 'remove' else 'allocation',), type='notification')
 396
 397     def holidays_validate_notificate(self, cr, uid, ids, context=None):
 398         for obj in self.browse(cr, uid, ids):
 399             if obj.holiday_status_id.double_validation:
 400                 self.message_append_note(cr, uid, [obj.id], _('System notification'),
 401                     _("The %s request has been <b>validated</b>. A second validation is necessary and is now pending.")
 402                     % ('leave' if obj.type == 'remove' else 'allocation',), type='notification', context=context)
 403             else:
 404                 self.message_append_note(cr, uid, [obj.id], _('System notification'),
 405                     _("The %s request has been <b>validated</b>. The validation process is now over.")
 406                     % ('leave' if obj.type == 'remove' else 'allocation',), type='notification', context=context)
 407
 408     def holidays_valid2_notificate(self, cr, uid, ids, context=None):
 409         for obj in self.browse(cr, uid, ids):
 410             self.message_append_note(cr, uid, [obj.id], _('System notification'),
 411                     _("The %s request has been <b>double validated</b>. The validation process is now over.")
 412                     % ('leave' if obj.type == 'remove' else 'allocation',), type='notification', context=context)
 413
 414     def holidays_refuse_notificate(self, cr, uid, ids, approval, context=None):
 415         for obj in self.browse(cr, uid, ids):
 416             self.message_append_note(cr, uid, [obj.id], _('System notification'),
 417                     _("The %s request has been <b>refused</b>. The validation process is now over.")
 418                     % ('leave' if obj.type == 'remove' else 'allocation',), type='notification', context=context)
 419
 420 hr_holidays()
 421
 422 class resource_calendar_leaves(osv.osv):
 423     _inherit = "resource.calendar.leaves"
 424     _description = "Leave Detail"
 425     _columns = {
 426         'holiday_id': fields.many2one("hr.holidays", "Holiday"),
 427     }
 428
 429 resource_calendar_leaves()
 430
 431
 432 class hr_employee(osv.osv):
 433     _inherit="hr.employee"
 434
 435     def create(self, cr, uid, vals, context=None):
 436         # don't pass the value of remaining leave if it's 0 at the creation time, otherwise it will trigger the inverse
 437         # function _set_remaining_days and the system may not be configured for. Note that we don't have this problem on
 438         # the write because the clients only send the fields that have been modified.
 439         if 'remaining_leaves' in vals and not vals['remaining_leaves']:
 440             del(vals['remaining_leaves'])
 441         return super(hr_employee, self).create(cr, uid, vals, context=context)
 442
 443     def _set_remaining_days(self, cr, uid, empl_id, name, value, arg, context=None):
 444         employee = self.browse(cr, uid, empl_id, context=context)
 445         diff = value - employee.remaining_leaves
 446         type_obj = self.pool.get('hr.holidays.status')
 447         holiday_obj = self.pool.get('hr.holidays')
 448         # Find for holidays status
 449         status_ids = type_obj.search(cr, uid, [('limit', '=', False)], context=context)
 450         if len(status_ids) != 1 :
 451             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)))
 452         status_id = status_ids and status_ids[0] or False
 453         if not status_id:
 454             return False
 455         if diff > 0:
 456             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)
 457         elif diff < 0:
 458             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)
 459         else:
 460             return False
 461         wf_service = netsvc.LocalService("workflow")
 462         wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'confirm', cr)
 463         wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'validate', cr)
 464         wf_service.trg_validate(uid, 'hr.holidays', leave_id, 'second_validate', cr)
 465         return True
 466
 467     def _get_remaining_days(self, cr, uid, ids, name, args, context=None):
 468         cr.execute("""SELECT
 469                 sum(h.number_of_days) as days,
 470                 h.employee_id
 471             from
 472                 hr_holidays h
 473                 join hr_holidays_status s on (s.id=h.holiday_status_id)
 474             where
 475                 h.state='validate' and
 476                 s.limit=False and
 477                 h.employee_id in (%s)
 478             group by h.employee_id"""% (','.join(map(str,ids)),) )
 479         res = cr.dictfetchall()
 480         remaining = {}
 481         for r in res:
 482             remaining[r['employee_id']] = r['days']
 483         for employee_id in ids:
 484             if not remaining.get(employee_id):
 485                 remaining[employee_id] = 0.0
 486         return remaining
 487
 488     def _get_leave_status(self, cr, uid, ids, name, args, context=None):
 489         holidays_obj = self.pool.get('hr.holidays')
 490         holidays_id = holidays_obj.search(cr, uid,
 491            [('employee_id', 'in', ids), ('date_from','<=',time.strftime('%Y-%m-%d')),
 492            ('date_to','>=',time.strftime('%Y-%m-%d')),('type','=','remove'),('state','not in',('cancel','refuse'))],
 493            context=context)
 494         result = {}
 495         for id in ids:
 496             result[id] = {
 497                 'current_leave_state': False,
 498                 'current_leave_id': False,
 499                 'date_from':False,
 500                 'date_to':False,
 501             }
 502         for holiday in self.pool.get('hr.holidays').browse(cr, uid, holidays_id, context=context):
 503             result[holiday.employee_id.id]['date_from'] = holiday.date_from
 504             result[holiday.employee_id.id]['date_to'] = holiday.date_to
 505             result[holiday.employee_id.id]['current_leave_state'] = holiday.state
 506             result[holiday.employee_id.id]['current_leave_id'] = holiday.holiday_status_id.id
 507         return result
 508
 509     _columns = {
 510         '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.'),
 511         'current_leave_state': fields.function(_get_leave_status, multi="leave_status", string="Current Leave Status", type="selection",
 512             selection=[('draft', 'New'), ('confirm', 'Waiting Approval'), ('refuse', 'Refused'),
 513             ('validate1', 'Waiting Second Approval'), ('validate', 'Approved'), ('cancel', 'Cancelled')]),
 514         'current_leave_id': fields.function(_get_leave_status, multi="leave_status", string="Current Leave Type",type='many2one', relation='hr.holidays.status'),
 515         'date_from': fields.function(_get_leave_status, multi='leave_status', type='date', string='From Date'),
 516         'date_to': fields.function(_get_leave_status, multi='leave_status', type='date', string='To Date'),
 517         'last_login': fields.related('user_id', 'date', type='datetime', string='Latest Connection', readonly=1)
 518     }
 519
 520 hr_employee()
 521
 522 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: