From 56b10caba5b5afcc028b14a8fe364f6b3e277c85 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Thibault=20Delavall=C3=A9e?= Date: Wed, 24 Jul 2013 13:21:30 +0200 Subject: [PATCH 1/6] [IMP] hr_holidays: also calculates virutal remaining leaves bzr revid: tde@openerp.com-20130724112130-3tsptifx6er4yww4 --- addons/hr_holidays/hr_holidays.py | 61 ++++++++++++++----------------- 1 file changed, 28 insertions(+), 33 deletions(-) diff --git a/addons/hr_holidays/hr_holidays.py b/addons/hr_holidays/hr_holidays.py index f9eee9cd243..adf6e70e01f 100644 --- a/addons/hr_holidays/hr_holidays.py +++ b/addons/hr_holidays/hr_holidays.py @@ -22,11 +22,10 @@ ############################################################################## import datetime -import time -from itertools import groupby -from operator import attrgetter, itemgetter - import math +import time +from operator import attrgetter + from openerp import tools from openerp.osv import fields, osv from openerp.tools.translate import _ @@ -36,40 +35,37 @@ class hr_holidays_status(osv.osv): _name = "hr.holidays.status" _description = "Leave Type" - def get_days(self, cr, uid, ids, employee_id, return_false, context=None): - 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""", - [employee_id, tuple(ids)]) - result = sorted(cr.dictfetchall(), key=lambda x: x['holiday_status_id']) - grouped_lines = dict((k, [v for v in itr]) for k, itr in groupby(result, itemgetter('holiday_status_id'))) - res = {} - for record in self.browse(cr, uid, ids, context=context): - res[record.id] = {} - max_leaves = leaves_taken = 0 - if not return_false: - if record.id in grouped_lines: - leaves_taken = -sum([item['number_of_days'] for item in grouped_lines[record.id] if item['type'] == 'remove']) - max_leaves = sum([item['number_of_days'] for item in grouped_lines[record.id] if item['type'] == 'add']) - res[record.id]['max_leaves'] = max_leaves - res[record.id]['leaves_taken'] = leaves_taken - res[record.id]['remaining_leaves'] = max_leaves - leaves_taken - return res + def get_days(self, cr, uid, ids, employee_id, context=None): + result = dict((id, dict(max_leaves=0, leaves_taken=0, remaining_leaves=0, + virtual_remaining_leaves=0)) for id in ids) + holiday_ids = self.pool['hr.holidays'].search(cr, uid, [('employee_id', '=', employee_id), + ('state', 'in', ['confirm', 'validate1', 'validate']), + ('holiday_status_id', 'in', ids) + ], context=context) + for holiday in self.pool['hr.holidays'].browse(cr, uid, holiday_ids, context=context): + status_dict = result[holiday.holiday_status_id.id] + if holiday.type == 'add': + status_dict['virtual_remaining_leaves'] += holiday.number_of_days + if holiday.state == 'validate': + status_dict['max_leaves'] += holiday.number_of_days + status_dict['remaining_leaves'] += holiday.number_of_days + elif holiday.type == 'remove': # number of days is negative + status_dict['virtual_remaining_leaves'] += holiday.number_of_days + if holiday.state == 'validate': + status_dict['leaves_taken'] -= holiday.number_of_days + status_dict['remaining_leaves'] += holiday.number_of_days + return result def _user_left_days(self, cr, uid, ids, name, args, context=None): - return_false = False employee_id = False - res = {} - if context and context.has_key('employee_id'): - if not context['employee_id']: - return_false = True + if context and 'employee_id' in context: employee_id = context['employee_id'] else: - employee_ids = self.pool.get('hr.employee').search(cr, uid, [('user_id','=',uid)], context=context) + employee_ids = self.pool.get('hr.employee').search(cr, uid, [('user_id', '=', uid)], context=context) if employee_ids: employee_id = employee_ids[0] - else: - return_false = True if employee_id: - res = self.get_days(cr, uid, ids, employee_id, return_false, context=context) + res = self.get_days(cr, uid, ids, employee_id, context=context) else: res = dict.fromkeys(ids, {'leaves_taken': 0, 'remaining_leaves': 0, 'max_leaves': 0}) return res @@ -84,6 +80,7 @@ class hr_holidays_status(osv.osv): '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'), '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'), 'remaining_leaves': fields.function(_user_left_days, string='Remaining Leaves', help='Maximum Leaves Allowed - Leaves Already Taken', multi='user_left_days'), + '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'), 'double_validation': fields.boolean('Apply Double Validation', help="When selected, the Allocation/Leave Requests for this type require a second validation to be approved."), } _defaults = { @@ -92,8 +89,6 @@ class hr_holidays_status(osv.osv): } def name_get(self, cr, uid, ids, context=None): - if not ids: - return [] res = [] for record in self.browse(cr, uid, ids, context=context): name = record.name @@ -425,7 +420,7 @@ class hr_holidays(osv.osv): for record in self.browse(cr, uid, ids): if record.holiday_type == 'employee' and record.type == 'remove': if record.employee_id and not record.holiday_status_id.limit: - 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'] + leaves_rest = holi_status_obj.get_days( cr, uid, [record.holiday_status_id.id], record.employee_id.id, False)[record.holiday_status_id.id]['virtual_remaining_leaves'] if leaves_rest < record.number_of_days_temp: 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)) return True From a87af992bba24aa736c9701243f37373f4cfae9e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Thibault=20Delavall=C3=A9e?= Date: Wed, 24 Jul 2013 14:27:54 +0200 Subject: [PATCH 2/6] [IMP] hr_holidays: cleaning + improve check_holidays bzr revid: tde@openerp.com-20130724122754-59xa1k8vo81dsb7b --- addons/hr_holidays/hr_holidays.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/addons/hr_holidays/hr_holidays.py b/addons/hr_holidays/hr_holidays.py index adf6e70e01f..6dc58bbdb24 100644 --- a/addons/hr_holidays/hr_holidays.py +++ b/addons/hr_holidays/hr_holidays.py @@ -416,13 +416,16 @@ class hr_holidays(osv.osv): return True def check_holidays(self, cr, uid, ids, context=None): - holi_status_obj = self.pool.get('hr.holidays.status') - for record in self.browse(cr, uid, ids): - if record.holiday_type == 'employee' and record.type == 'remove': - if record.employee_id and not record.holiday_status_id.limit: - leaves_rest = holi_status_obj.get_days( cr, uid, [record.holiday_status_id.id], record.employee_id.id, False)[record.holiday_status_id.id]['virtual_remaining_leaves'] - if leaves_rest < record.number_of_days_temp: - 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)) + for record in self.browse(cr, uid, ids, context=context): + if record.holiday_type != 'employee' or record.type != 'remove' or not record.employee_id or record.holiday_status_id.limit: + continue + 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] + if leave_days['remaining_leaves'] < record.number_of_days_temp: + raise osv.except_osv(_('Warning!'), + _('There are not enough remaining days available in %s for employee %s.') % (record.holiday_status_id.name, record.employee_id.name)) + if leave_days['virtual_remaining_leaves'] < record.number_of_days_temp: + raise osv.except_osv(_('Warning!'), + _('Other pending requests already book too much days in %s for employee %s.') % (record.holiday_status_id.name, record.employee_id.name)) return True # ----------------------------- From 7f1e1c72468b41c80823a27077b9ac21374e4084 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Thibault=20Delavall=C3=A9e?= Date: Mon, 5 Aug 2013 13:30:23 +0200 Subject: [PATCH 3/6] [IMP] hr_holidays: workflow update starts in confirm from confirm, possible to get back to draft with reset signal, calling old set_to_draft, now renamed holidays_reset, in the same workflow from draft, possible to go to confirm bzr revid: tde@openerp.com-20130805113023-1cs1s18dd5m4mi20 --- addons/hr_holidays/hr_holidays.py | 8 +++---- addons/hr_holidays/hr_holidays_data.xml | 2 +- addons/hr_holidays/hr_holidays_view.xml | 7 +++--- addons/hr_holidays/hr_holidays_workflow.xml | 25 +++++++++++++++------ addons/hr_holidays/test/test_hr_holiday.yml | 2 +- 5 files changed, 27 insertions(+), 17 deletions(-) diff --git a/addons/hr_holidays/hr_holidays.py b/addons/hr_holidays/hr_holidays.py index 6dc58bbdb24..b75d2538e4e 100644 --- a/addons/hr_holidays/hr_holidays.py +++ b/addons/hr_holidays/hr_holidays.py @@ -165,7 +165,7 @@ class hr_holidays(osv.osv): } _defaults = { 'employee_id': _employee_get, - 'state': 'draft', + 'state': 'confirm', 'type': 'remove', 'user_id': lambda obj, cr, uid, context: uid, 'holiday_type': 'employee' @@ -306,18 +306,16 @@ class hr_holidays(osv.osv): raise osv.except_osv(_('Warning!'),_('You cannot modify a leave request that has been approved. Contact a human resource manager.')) return super(hr_holidays, self).write(cr, uid, ids, vals, context=context) - def set_to_draft(self, cr, uid, ids, context=None): + def holidays_reset(self, cr, uid, ids, context=None): self.write(cr, uid, ids, { 'state': 'draft', 'manager_id': False, 'manager_id2': False, }) - self.delete_workflow(cr, uid, ids) - self.create_workflow(cr, uid, ids) to_unlink = [] for record in self.browse(cr, uid, ids, context=context): for record2 in record.linked_request_ids: - self.set_to_draft(cr, uid, [record2.id], context=context) + self.holidays_reset(cr, uid, [record2.id], context=context) to_unlink.append(record2.id) if to_unlink: self.unlink(cr, uid, to_unlink, context=context) diff --git a/addons/hr_holidays/hr_holidays_data.xml b/addons/hr_holidays/hr_holidays_data.xml index 2b370f08821..703fa74a2ee 100644 --- a/addons/hr_holidays/hr_holidays_data.xml +++ b/addons/hr_holidays/hr_holidays_data.xml @@ -49,7 +49,7 @@ To Approve hr.holidays - Request created and waiting confirmation + Request confirmed and waiting approval Approved diff --git a/addons/hr_holidays/hr_holidays_view.xml b/addons/hr_holidays/hr_holidays_view.xml index 85bbcb96f94..9a91a2198ef 100644 --- a/addons/hr_holidays/hr_holidays_view.xml +++ b/addons/hr_holidays/hr_holidays_view.xml @@ -54,10 +54,11 @@
+
@@ -102,7 +103,7 @@