1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU Affero General Public License as
9 # published by the Free Software Foundation, either version 3 of the
10 # License, or (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU Affero General Public License for more details.
17 # You should have received a copy of the GNU Affero General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 ##############################################################################
22 from datetime import datetime, timedelta
25 from new import classobj
26 from osv import fields, osv
27 from tools.translate import _
29 from itertools import groupby
30 from operator import itemgetter
33 class resource_calendar(osv.osv):
34 _name = "resource.calendar"
35 _description = "Resource Calendar"
37 'name' : fields.char("Name", size=64, required=True),
38 'company_id' : fields.many2one('res.company', 'Company', required=False),
39 'attendance_ids' : fields.one2many('resource.calendar.attendance', 'calendar_id', 'Working Time'),
40 'manager' : fields.many2one('res.users', 'Workgroup manager'),
43 'company_id': lambda self, cr, uid, context: self.pool.get('res.company')._company_default_get(cr, uid, 'resource.calendar', context=context)
46 def working_hours_on_day(self, cr, uid, resource_calendar_id, day, context=None):
48 @param resource_calendar_id: resource.calendar browse record
49 @param day: datetime object
50 @return: returns the working hours (as float) men should work on the given day if is in the attendance_ids of the resource_calendar_id (i.e if that day is a working day), returns 0.0 otherwise
53 for working_day in resource_calendar_id.attendance_ids:
54 if (int(working_day.dayofweek) + 1) == day.isoweekday():
55 res += working_day.hour_to - working_day.hour_from
58 def _get_leaves(self, cr, uid, id, resource):
59 resource_cal_leaves = self.pool.get('resource.calendar.leaves')
62 resource_leave_ids = resource_cal_leaves.search(cr, uid, [('calendar_id','=',id), '|', ('resource_id','=',False), ('resource_id','=',resource)])
63 #res_leaves = resource_cal_leaves.read(cr, uid, resource_leave_ids, ['date_from', 'date_to'])
64 res_leaves = resource_cal_leaves.browse(cr, uid, resource_leave_ids)
66 for leave in res_leaves:
67 dtf = datetime.strptime(leave.date_from, '%Y-%m-%d %H:%M:%S')
68 dtt = datetime.strptime(leave.date_to, '%Y-%m-%d %H:%M:%S')
70 [dt_leave.append((dtf + timedelta(days=x)).strftime('%Y-%m-%d')) for x in range(int(no.days + 1))]
75 def interval_min_get(self, cr, uid, id, dt_from, hours, resource=False):
78 return [(dt_from - timedelta(hours=td), dt_from)]
79 dt_leave = self._get_leaves(cr, uid, id, resource)
84 current_hour = dt_from.hour
85 while (todo>0) and maxrecur:
86 cr.execute("select hour_from,hour_to from resource_calendar_attendance where dayofweek='%s' and calendar_id=%s order by hour_from desc", (dt_from.weekday(),id))
87 for (hour_from,hour_to) in cr.fetchall():
89 if (hour_from<current_hour) and (todo>0):
90 m = min(hour_to, current_hour)
91 if (m-hour_from)>todo:
93 dt_check = dt_from.strftime('%Y-%m-%d')
94 for leave in dt_leave:
96 dt_check = datetime.strptime(dt_check, '%Y-%m-%d') + timedelta(days=1)
101 d1 = datetime(dt_from.year, dt_from.month, dt_from.day, int(math.floor(hour_from)), int((hour_from%1) * 60))
102 d2 = datetime(dt_from.year, dt_from.month, dt_from.day, int(math.floor(m)), int((m%1) * 60))
103 result.append((d1, d2))
104 current_hour = hour_from
105 todo -= (m-hour_from)
106 dt_from -= timedelta(days=1)
112 # def interval_get(self, cr, uid, id, dt_from, hours, resource=False, byday=True):
113 def interval_get_multi(self, cr, uid, date_and_hours_by_cal, resource=False, byday=True):
115 lst.sort(key=itemgetter(key))
116 grouped = groupby(lst, itemgetter(key))
117 return dict([(k, [v for v in itr]) for k, itr in grouped])
120 cr.execute("select calendar_id, dayofweek, hour_from, hour_to from resource_calendar_attendance order by hour_from")
121 hour_res = cr.dictfetchall()
122 hours_by_cal = group(hour_res, 'calendar_id')
126 for d, hours, id in date_and_hours_by_cal:
127 dt_from = datetime.strptime(d, '%Y-%m-%d %H:%M:%S')
130 results[(d, hours, id)] = [(dt_from, dt_from + timedelta(hours=td))]
133 dt_leave = self._get_leaves(cr, uid, id, resource)
137 current_hour = dt_from.hour
138 while (todo>0) and maxrecur:
139 for (hour_from,hour_to) in [(item['hour_from'], item['hour_to']) for item in hours_by_cal[id] if item['dayofweek'] == str(dt_from.weekday())]:
141 if (hour_to>current_hour) and (todo>0):
142 m = max(hour_from, current_hour)
145 dt_check = dt_from.strftime('%Y-%m-%d')
146 for leave in dt_leave:
147 if dt_check == leave:
148 dt_check = datetime.strptime(dt_check, '%Y-%m-%d') + timedelta(days=1)
153 d1 = datetime(dt_from.year, dt_from.month, dt_from.day, int(math.floor(m)), int((m%1) * 60))
154 d2 = datetime(dt_from.year, dt_from.month, dt_from.day, int(math.floor(hour_to)), int((hour_to%1) * 60))
155 result.append((d1, d2))
156 current_hour = hour_to
157 todo -= (hour_to - m)
158 dt_from += timedelta(days=1)
161 results[(d, hours, id)] = result
164 def interval_get(self, cr, uid, id, dt_from, hours, resource=False, byday=True):
165 res = self.interval_get_multi(cr, uid, [(dt_from.strftime('%Y-%m-%d %H:%M:%S'), hours, id)], resource, byday)[(dt_from.strftime('%Y-%m-%d %H:%M:%S'), hours, id)]
168 def interval_hours_get(self, cr, uid, id, dt_from, dt_to, resource=False):
171 dt_leave = self._get_leaves(cr, uid, id, resource)
174 current_hour = dt_from.hour
176 while (dt_from <= dt_to):
177 cr.execute("select hour_from,hour_to from resource_calendar_attendance where dayofweek='%s' and calendar_id=%s order by hour_from", (dt_from.weekday(),id))
179 for (hour_from,hour_to) in der:
180 if hours != 0.0:#For first time of the loop only,hours will be 0
181 current_hour = hour_from
183 if (hour_to>=current_hour):
184 dt_check = dt_from.strftime('%Y-%m-%d')
185 for leave in dt_leave:
186 if dt_check == leave:
187 dt_check = datetime.strptime(dt_check, "%Y-%m-%d") + timedelta(days=1)
194 d2 = datetime(dt_from.year, dt_from.month, dt_from.day, int(math.floor(hour_to)), int((hour_to%1) * 60))
196 if hours != 0.0:#For first time of the loop only,hours will be 0
197 d1 = datetime(dt_from.year, dt_from.month, dt_from.day, int(math.floor(current_hour)), int((current_hour%1) * 60))
199 if dt_from.day == dt_to.day:
200 if hour_from <= dt_to.hour <= hour_to:
203 hours += (d2-d1).seconds
204 dt_from = datetime(dt_from.year, dt_from.month, dt_from.day, int(math.floor(current_hour)), int((current_hour%1) * 60)) + timedelta(days=1)
211 class resource_calendar_attendance(osv.osv):
212 _name = "resource.calendar.attendance"
213 _description = "Work Detail"
216 'name' : fields.char("Name", size=64, required=True),
217 'dayofweek': fields.selection([('0','Monday'),('1','Tuesday'),('2','Wednesday'),('3','Thursday'),('4','Friday'),('5','Saturday'),('6','Sunday')], 'Day of week', required=True),
218 'date_from' : fields.date('Starting date'),
219 'hour_from' : fields.float('Work from', size=8, required=True, help="Working time will start from"),
220 'hour_to' : fields.float("Work to", size=8, required=True, help="Working time will end at"),
221 'calendar_id' : fields.many2one("resource.calendar", "Resource's Calendar", required=True),
224 _order = 'dayofweek, hour_from'
229 resource_calendar_attendance()
231 def convert_timeformat(time_string):
232 split_list = str(time_string).split('.')
233 hour_part = split_list[0]
234 mins_part = split_list[1]
235 round_mins = int(round(float(mins_part) * 60,-2))
236 converted_string = hour_part + ':' + str(round_mins)[0:2]
237 return converted_string
239 class resource_resource(osv.osv):
240 _name = "resource.resource"
241 _description = "Resource Detail"
243 'name' : fields.char("Name", size=64, required=True),
244 'code': fields.char('Code', size=16),
245 'active' : fields.boolean('Active', help="If the active field is set to False, it will allow you to hide the resource record without removing it."),
246 'company_id' : fields.many2one('res.company', 'Company'),
247 'resource_type': fields.selection([('user','Human'),('material','Material')], 'Resource Type', required=True),
248 'user_id' : fields.many2one('res.users', 'User', help='Related user name for the resource to manage its access.'),
249 'time_efficiency' : fields.float('Efficiency factor', size=8, required=True, help="This field depict the efficiency of the resource to complete tasks. e.g resource put alone on a phase of 5 days with 5 tasks assigned to him, will show a load of 100% for this phase by default, but if we put a efficency of 200%, then his load will only be 50%."),
250 'calendar_id' : fields.many2one("resource.calendar", "Working Time", help="Define the schedule of resource"),
253 'resource_type' : 'user',
254 'time_efficiency' : 1,
256 'company_id': lambda self, cr, uid, context: self.pool.get('res.company')._company_default_get(cr, uid, 'resource.resource', context=context)
260 def copy(self, cr, uid, id, default=None, context=None):
263 if not default.get('name', False):
264 default['name'] = self.browse(cr, uid, id, context=context).name + _(' (copy)')
265 return super(resource_resource, self).copy(cr, uid, id, default, context)
267 def generate_resources(self, cr, uid, user_ids, calendar_id, context=None):
269 Return a list of Resource Class objects for the resources allocated to the phase.
272 user_pool = self.pool.get('res.users')
273 for user in user_pool.browse(cr, uid, user_ids, context=context):
274 resource_objs[user.id] = {
280 resource_ids = self.search(cr, uid, [('user_id', '=', user.id)], context=context)
282 for resource in self.browse(cr, uid, resource_ids, context=context):
283 resource_objs[user.id]['efficiency'] = resource.time_efficiency
284 resource_cal = resource.calendar_id.id
286 leaves = self.compute_vacation(cr, uid, calendar_id, resource.id, resource_cal, context=context)
287 resource_objs[user.id]['vacation'] += list(leaves)
290 def compute_vacation(self, cr, uid, calendar_id, resource_id=False, resource_calendar=False, context=None):
292 Compute the vacation from the working calendar of the resource.
293 @param calendar_id : working calendar of the project
294 @param resource_id : resource working on phase/task
295 @param resource_calendar : working calendar of the resource
297 resource_calendar_leaves_pool = self.pool.get('resource.calendar.leaves')
300 leave_ids = resource_calendar_leaves_pool.search(cr, uid, ['|', ('calendar_id', '=', calendar_id),
301 ('calendar_id', '=', resource_calendar),
302 ('resource_id', '=', resource_id)
305 leave_ids = resource_calendar_leaves_pool.search(cr, uid, [('calendar_id', '=', calendar_id),
306 ('resource_id', '=', False)
308 leaves = resource_calendar_leaves_pool.read(cr, uid, leave_ids, ['date_from', 'date_to'], context=context)
309 for i in range(len(leaves)):
310 dt_start = datetime.strptime(leaves[i]['date_from'], '%Y-%m-%d %H:%M:%S')
311 dt_end = datetime.strptime(leaves[i]['date_to'], '%Y-%m-%d %H:%M:%S')
312 no = dt_end - dt_start
313 [leave_list.append((dt_start + timedelta(days=x)).strftime('%Y-%m-%d')) for x in range(int(no.days + 1))]
317 def compute_working_calendar(self, cr, uid, calendar_id=False, context=None):
319 Change the format of working calendar from 'Openerp' format to bring it into 'Faces' format.
320 @param calendar_id : working calendar of the project
323 # Calendar is not specified: working days: 24/7
324 return [('fri', '8:0-12:0','13:0-17:0'), ('thu', '8:0-12:0','13:0-17:0'), ('wed', '8:0-12:0','13:0-17:0'),
325 ('mon', '8:0-12:0','13:0-17:0'), ('tue', '8:0-12:0','13:0-17:0')]
326 resource_attendance_pool = self.pool.get('resource.calendar.attendance')
327 time_range = "8:00-8:00"
329 week_days = {"0": "mon", "1": "tue", "2": "wed","3": "thu", "4": "fri", "5": "sat", "6": "sun"}
334 week_ids = resource_attendance_pool.search(cr, uid, [('calendar_id', '=', calendar_id)], context=context)
335 weeks = resource_attendance_pool.read(cr, uid, week_ids, ['dayofweek', 'hour_from', 'hour_to'], context=context)
336 # Convert time formats into appropriate format required
337 # and create a list like [('mon', '8:00-12:00'), ('mon', '13:00-18:00')]
341 if week_days.has_key(week['dayofweek']):
342 day = week_days[week['dayofweek']]
343 wk_days[week['dayofweek']] = week_days[week['dayofweek']]
344 hour_from_str = convert_timeformat(week['hour_from'])
345 hour_to_str = convert_timeformat(week['hour_to'])
346 res_str = hour_from_str + '-' + hour_to_str
347 wktime_list.append((day, res_str))
348 # Convert into format like [('mon', '8:00-12:00', '13:00-18:00')]
349 for item in wktime_list:
350 if wk_time.has_key(item[0]):
351 wk_time[item[0]].append(item[1])
353 wk_time[item[0]] = [item[0]]
354 wk_time[item[0]].append(item[1])
355 for k,v in wk_time.items():
356 wktime_cal.append(tuple(v))
357 # Add for the non-working days like: [('sat, sun', '8:00-8:00')]
358 for k, v in wk_days.items():
359 if week_days.has_key(k):
361 for v in week_days.itervalues():
362 non_working += v + ','
364 wktime_cal.append((non_working[:-1], time_range))
367 #TODO: Write optimized alogrothem for resource availability. : Method Yet not implemented
368 def check_availability(self, cr, uid, ids, start, end, context=None):
376 class resource_calendar_leaves(osv.osv):
377 _name = "resource.calendar.leaves"
378 _description = "Leave Detail"
380 'name' : fields.char("Name", size=64),
381 'company_id' : fields.related('calendar_id','company_id',type='many2one',relation='res.company',string="Company", store=True, readonly=True),
382 'calendar_id' : fields.many2one("resource.calendar", "Working time"),
383 'date_from' : fields.datetime('Start Date', required=True),
384 'date_to' : fields.datetime('End Date', required=True),
385 'resource_id' : fields.many2one("resource.resource", "Resource", help="If empty, this is a generic holiday for the company. If a resource is set, the holiday/leave is only for this resource"),
388 def check_dates(self, cr, uid, ids, context=None):
389 leave = self.read(cr, uid, ids[0], ['date_from', 'date_to'])
390 if leave['date_from'] and leave['date_to']:
391 if leave['date_from'] > leave['date_to']:
396 (check_dates, 'Error! leave start-date must be lower then leave end-date.', ['date_from', 'date_to'])
399 def onchange_resource(self,cr, uid, ids, resource, context=None):
402 resource_pool = self.pool.get('resource.resource')
403 result['calendar_id'] = resource_pool.browse(cr, uid, resource, context=context).calendar_id.id
404 return {'value': result}
405 return {'value': {'calendar_id': []}}
407 resource_calendar_leaves()
409 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: