← Back to branch summary

~zaber/openupgrade-addons/missing-import

« back to all changes in this revision

Viewing changes to resource/resource.py

[ADD]:added resource module

Show diffs side-by-side

added added

removed removed

Lines of Context:
resource/resource.py
 
1
# -*- coding: utf-8 -*-
 
2
##############################################################################
 
3
#
 
4
#    OpenERP, Open Source Management Solution
 
5
#    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
 
6
#
 
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.
 
11
#
 
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.
 
16
#
 
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/>.
 
19
#
 
20
##############################################################################
 
21
import mx.DateTime
 
22
import time
 
23
import math
 
24
from osv import fields, osv
 
25
from tools.translate import _
 
26
 
 
27
class resource_calendar(osv.osv):
 
28
    _name = "resource.calendar"
 
29
    _description = "Resource Calendar"
 
30
    _columns = {
 
31
        'name' : fields.char("Name", size=64, required=True),
 
32
        'company_id' : fields.many2one('res.company', 'Company', required=True),
 
33
        'week_id' : fields.one2many('resource.calendar.week', 'calendar_id', 'Working Time'),
 
34
        'manager' : fields.many2one('res.users', 'Workgroup manager'),
 
35
    }
 
36
    _defaults = {
 
37
        'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'resource.calendar', c)
 
38
    }
 
39
    def interval_min_get(self, cr, uid, id, dt_from, hours , resource=0):
 
40
        dt_leave = []
 
41
        if not id:
 
42
            return [(dt_from-mx.DateTime.RelativeDateTime(hours=int(hours)*3), dt_from)]
 
43
        if resource:
 
44
            resource_leave_ids = self.pool.get('resource.calendar.leaves').search(cr,uid,[('resource_id','=',resource)])
 
45
            if resource_leave_ids:
 
46
                res_leaves = self.pool.get('resource.calendar.leaves').read(cr,uid,resource_leave_ids,['date_from','date_to'])
 
47
                print 'Leave Details',res_leaves
 
48
                for i in range(len(res_leaves)):
 
49
                    dtf = mx.DateTime.strptime(res_leaves[i]['date_from'],'%Y-%m-%d %H:%M:%S')
 
50
                    dtt = mx.DateTime.strptime(res_leaves[i]['date_to'],'%Y-%m-%d %H:%M:%S')
 
51
                    leave_days = ((dtt - dtf).days) + 1
 
52
                    print 'No Of Leaves:::::::::',int(leave_days)
 
53
                    for x in range(int(leave_days)):
 
54
                        dt_leave.append((dtf + mx.DateTime.RelativeDateTime(days=x)).strftime('%Y-%m-%d'))
 
55
                    dt_leave.sort()
 
56
            print 'Sorted Leave Dates::::',dt_leave
 
57
        todo = hours
 
58
        cycle = 0
 
59
        result = []
 
60
        maxrecur = 100
 
61
        current_hour = dt_from.hour
 
62
        while (todo>0) and maxrecur:
 
63
            cr.execute("select hour_from,hour_to from resource_calendar_week where dayofweek='%s' and calendar_id=%s order by hour_from", (dt_from.day_of_week,id))
 
64
            for (hour_from,hour_to) in cr.fetchall():
 
65
                    if (hour_to>current_hour) and (todo>0):
 
66
                        m = max(hour_from, current_hour)
 
67
                        if (hour_to-m)>todo:
 
68
                            hour_to = m+todo
 
69
                        d1 = mx.DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,int(math.floor(m)),int((m%1) * 60))
 
70
                        d2 = mx.DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,int(math.floor(hour_to)),int((hour_to%1) * 60))
 
71
                        dt1 = d1.strftime('%Y-%m-%d')
 
72
                        dt2 = d2.strftime('%Y-%m-%d')
 
73
                        print 'Date 1',dt1
 
74
                        print 'Date 2',dt2
 
75
                        for i in range(len(dt_leave)):
 
76
                            if dt1 == dt_leave[i]:
 
77
                                print 'Leave::::::::'
 
78
                                dt_from += mx.DateTime.RelativeDateTime(days=1)
 
79
                                d1 = mx.DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,int(math.floor(m)),int((m%1) * 60))
 
80
                                d2 = mx.DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,int(math.floor(hour_to)),int((hour_to%1) * 60))
 
81
                                dt1 = d1.strftime('%Y-%m-%d')
 
82
                                dt2 = d2.strftime('%Y-%m-%d')
 
83
                        result.append((d1, d2))
 
84
                        current_hour = hour_to
 
85
                        todo -= (hour_to - m)
 
86
            dt_from += mx.DateTime.RelativeDateTime(days=1)
 
87
            current_hour = 0
 
88
            maxrecur -= 1
 
89
        return result
 
90
 
 
91
    def interval_get(self, cr, uid, id, dt_from, hours, resource=0, byday=True):
 
92
        dt_leave = []
 
93
        if not id:
 
94
            return [(dt_from,dt_from+mx.DateTime.RelativeDateTime(hours=int(hours)*3))]
 
95
        if resource:
 
96
            resource_leave_ids = self.pool.get('resource.calendar.leaves').search(cr,uid,[('resource_id','=',resource)])
 
97
            if resource_leave_ids:
 
98
                res_leaves = self.pool.get('resource.calendar.leaves').read(cr,uid,resource_leave_ids,['date_from','date_to'])
 
99
                print 'Leave Details',res_leaves
 
100
                for i in range(len(res_leaves)):
 
101
                    dtf = mx.DateTime.strptime(res_leaves[i]['date_from'],'%Y-%m-%d %H:%M:%S')
 
102
                    dtt = mx.DateTime.strptime(res_leaves[i]['date_to'],'%Y-%m-%d %H:%M:%S')
 
103
                    no = dtt - dtf
 
104
                    leave_days = no.days + 1
 
105
                    print 'No Of Leaves:::::::::',int(leave_days)
 
106
                    for x in range(int(leave_days)):
 
107
                        dt_leave.append((dtf + mx.DateTime.RelativeDateTime(days=x)).strftime('%Y-%m-%d'))
 
108
                    dt_leave.sort()
 
109
            print 'Sorted Leave Dates::::',dt_leave
 
110
        todo = hours
 
111
        cycle = 0
 
112
        result = []
 
113
        maxrecur = 100
 
114
        current_hour = dt_from.hour
 
115
        while (todo>0) and maxrecur:
 
116
            cr.execute("select hour_from,hour_to from resource_calendar_week where dayofweek='%s' and calendar_id=%s order by hour_from", (dt_from.day_of_week,id))
 
117
            for (hour_from,hour_to) in cr.fetchall():
 
118
                    if (hour_to>current_hour) and (todo>0):
 
119
                        m = max(hour_from, current_hour)
 
120
                        if (hour_to-m)>todo:
 
121
                            hour_to = m+todo
 
122
                        d1 = mx.DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,int(math.floor(m)),int((m%1) * 60))
 
123
                        d2 = mx.DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,int(math.floor(hour_to)),int((hour_to%1) * 60))
 
124
                        dt1 = d1.strftime('%Y-%m-%d')
 
125
                        dt2 = d2.strftime('%Y-%m-%d')
 
126
                        print 'Date 1',dt1
 
127
                        print 'Date 2',dt2
 
128
                        for i in range(len(dt_leave)):
 
129
                            if dt1 == dt_leave[i]:
 
130
                                print 'Leave::::::::'
 
131
                                dt_from += mx.DateTime.RelativeDateTime(days=1)
 
132
                                d1 = mx.DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,int(math.floor(m)),int((m%1) * 60))
 
133
                                d2 = mx.DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,int(math.floor(hour_to)),int((hour_to%1) * 60))
 
134
                                dt1 = d1.strftime('%Y-%m-%d')
 
135
                                dt2 = d2.strftime('%Y-%m-%d')
 
136
                        result.append((d1, d2))
 
137
                        current_hour = hour_to
 
138
                        todo -= (hour_to - m)
 
139
            dt_from += mx.DateTime.RelativeDateTime(days=1)
 
140
            current_hour = 0
 
141
            maxrecur -= 1
 
142
        return result
 
143
 
 
144
resource_calendar()
 
145
 
 
146
class resource_calendar_week(osv.osv):
 
147
    _name = "resource.calendar.week"
 
148
    _description = "Work Detail"
 
149
    _columns = {
 
150
        'name' : fields.char("Name", size=64, required=True),
 
151
        'dayofweek': fields.selection([('0','Monday'),('1','Tuesday'),('2','Wednesday'),('3','Thursday'),('4','Friday'),('5','Saturday'),('6','Sunday')], 'Day of week'),
 
152
        'date_from' : fields.date('Starting date'),
 
153
        'hour_from' : fields.float('Work from', size=8, required=True),
 
154
        'hour_to' : fields.float("Work to", size=8, required=True),
 
155
        'calendar_id' : fields.many2one("resource.calendar", "Resource's Calendar", required=True),
 
156
    }
 
157
    _order = 'dayofweek, hour_from'
 
158
resource_calendar_week()
 
159
 
 
160
class resource_resource(osv.osv):
 
161
    _name = "resource.resource"
 
162
    _description = "Resource Detail"
 
163
    _columns = {
 
164
        'name' : fields.char("Name", size=64, required=True),
 
165
        'code': fields.char('Code', size=16),
 
166
        'active' : fields.boolean('Active', help="If the active field is set to true, it will allow you to hide the resource record without removing it."),
 
167
        'company_id' : fields.many2one('res.company', 'Company', required=True),
 
168
        'resource_type': fields.selection([('user','Human'),('material','Material')], 'Resource Type', required=True),
 
169
        'user_id' : fields.many2one('res.users', 'User'),
 
170
        'time_efficiency' : fields.float('Efficiency factor', size=8, help="This field depict the efficency of the ressource to complete tasks. e.g  ressource 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%."),
 
171
        'calendar_id' : fields.many2one("resource.calendar", "Working time", required=True, help="Define the schedule of resource"),
 
172
    }
 
173
    _defaults = {
 
174
        'resource_type' : lambda *a: 'user',
 
175
        'time_efficiency' : lambda *a: 1,
 
176
        'active' : lambda *a: True,
 
177
        'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'resource.resource', c)
 
178
    }
 
179
resource_resource()
 
180
 
 
181
class resource_calendar_leaves(osv.osv):
 
182
    _name = "resource.calendar.leaves"
 
183
    _description = "Leave Detail"
 
184
    _columns = {
 
185
        'name' : fields.char("Name", size=64),
 
186
        'company_id' : fields.related('calendar_id','company_id',type='many2one',relation='res.company',string="Company",required=True),
 
187
        #'company_id':fields.many2one('res.company', 'Company', required=True),
 
188
        'calendar_id' : fields.many2one("resource.calendar", "Working time"),
 
189
        'date_from' : fields.datetime('Start Date', required=True),
 
190
        'date_to' : fields.datetime('End Date', required=True),
 
191
        '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"),
 
192
 
 
193
    }
 
194
resource_calendar_leaves()
 
195
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: