1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
|
# -*- encoding: utf-8 -*-
##############################################################################
#
# OpenERP, Open Source Management Solution
# Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
# $Id$
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
from mx import DateTime
import time
from osv import fields, osv
class hr_timesheet_group(osv.osv):
_name = "hr.timesheet.group"
_description = "Timesheet"
_columns = {
'name' : fields.char("Group name", size=64, required=True),
'timesheet_id' : fields.one2many('hr.timesheet', 'tgroup_id', 'Timesheet'),
'manager' : fields.many2one('res.users', 'Workgroup manager'),
}
#
# TODO: improve; very slow !
# bug if transition to another period
#
def interval_get(self, cr, uid, id, dt_from, hours, byday=True):
if not id:
return [(dt_from,dt_from+DateTime.RelativeDateTime(hours=int(hours)*3))]
todo = hours
cycle = 0
result = []
while todo>0:
cr.execute('select hour_from,hour_to from hr_timesheet where dayofweek=%d and tgroup_id=%d order by hour_from', (dt_from.day_of_week,id))
for (hour_from,hour_to) in cr.fetchall():
h1,m1 = map(int,hour_from.split(':'))
h2,m2 = map(int,hour_to.split(':'))
d1 = DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,h1,m1)
d2 = DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day,h2,m2)
if dt_from<d2:
date1 = max(dt_from,d1)
if date1+DateTime.RelativeDateTime(hours=todo)<=d2:
result.append((date1, date1+DateTime.RelativeDateTime(hours=todo)))
todo = 0
else:
todo -= (d2-date1).hours
result.append((date1, d2))
dt_from = DateTime.DateTime(dt_from.year,dt_from.month,dt_from.day)+DateTime.RelativeDateTime(days=1)
cycle+=1
if cycle>7 and todo==hours:
return [(dt_from,dt_from+DateTime.RelativeDateTime(hours=hours*3))]
if byday:
i = 1
while i<len(result):
if (result[i][0]-result[i-1][1]).days<1:
result[i-1]=(result[i-1][0],result[i][1])
del result[i]
else:
i+=1
return result
hr_timesheet_group()
class hr_employee_category(osv.osv):
_name = "hr.employee.category"
_description = "Employee Category"
_columns = {
'name' : fields.char("Category", size=64, required=True),
'parent_id': fields.many2one('hr.employee.category', 'Parent category', select=True),
'child_ids': fields.one2many('hr.employee.category', 'parent_id', 'Childs Categories')
}
hr_employee_category()
class hr_employee_department(osv.osv):
_name = "hr.employee.department"
_description = "Departments"
_columns = {
'name' : fields.char("Department", size=64),
'dept_id' : fields.one2many('hr.employee.position', 'department', 'Department ID'),
}
hr_employee_department()
class hr_employee_position(osv.osv):
_name = "hr.employee.position"
_description = "Position Available"
_columns = {
'name' : fields.char("Position", required=True, size=64),
'department' : fields.many2one('hr.employee.department', 'Department'),
}
_order = 'department desc'
hr_employee_position()
class hr_employee(osv.osv):
def _get_basic_salary(self, cr, uid, ids, prop, unknow_none, unknow_dict):
id_set=",".join(map(str,ids))
cr.execute("SELECT s.id,COALESCE(SUM(s.basic),0)::decimal(16,2) AS basic FROM hr_employee s WHERE s.id IN ("+id_set+") GROUP BY s.id ")
res=dict(cr.fetchall())
return res
def _get_gross_salary(self, cr, uid, ids, prop, unknow_none, unknow_dict):
id_set=",".join(map(str,ids))
cr.execute("SELECT s.id,COALESCE(SUM(l.value),0)::decimal(16,2) AS gross FROM hr_employee s LEFT OUTER JOIN hr_employee_salary_allowance l ON (s.id=l.employee_id) WHERE s.id IN ("+id_set+") GROUP BY s.id ")
allowance=dict(cr.fetchall())
basic = self._get_basic_salary(cr, uid, ids, prop, unknow_none, unknow_dict)
res = {}
for id in ids:
res[id] = basic.get(id,0.0) + allowance.get(id,0.0)
return res
def _get_net_deduction(self, cr, uid, ids, prop, unknow_none, unknow_dict):
id_set=",".join(map(str,ids))
cr.execute("SELECT s.id,COALESCE(SUM(l.value),0)::decimal(16,2) AS n_deduction FROM hr_employee s LEFT OUTER JOIN hr_employee_salary_deduction l ON (s.id=l.deduction_id) WHERE s.id IN ("+id_set+") GROUP BY s.id ")
res=dict(cr.fetchall())
return res
def _get_net_salary(self, cr, uid, ids, prop, unknow_none, unknow_dict):
gross = self._get_gross_salary(cr, uid, ids, prop, unknow_none, unknow_dict)
net_deduct = self._get_net_deduction(cr, uid, ids, prop, unknow_none, unknow_dict)
res = {}
for id in ids:
res[id] = gross.get(id,0.0) - net_deduct.get(id,0.0)
return res
def _get_employee_department(self, cr, uid, ids, prop, unknow_none, unknow_dict):
id_set=",".join(map(str,ids))
cr.execute("SELECT s.id,COALESCE(SUM(l.value),0)::decimal(16,2) AS n_deduction FROM hr_employee s LEFT OUTER JOIN hr_employee_salary_deduction l ON (s.id=l.deduction_id) WHERE s.id IN ("+id_set+") GROUP BY s.id ")
res=dict(cr.fetchall())
return res
_name = "hr.employee"
_description = "Employee"
_columns = {
'name' : fields.char("Employee", size=128, required=True),
'dob' : fields.date("Date of Birth"),
'sex' : fields.selection([('male','Male'), ('female','Female')], 'Sex'),
'active' : fields.boolean('Active'),
'company_id': fields.many2one('res.company', 'Company'),
'address_id': fields.many2one('res.partner.address', 'Contact address'),
'state' : fields.selection([('absent', 'Absent'), ('present', 'Present')], 'Attendance', readonly=True),
'started' : fields.date("Started on"),
'notes': fields.text('Notes'),
'attendances' : fields.one2many('hr.attendance', 'employee_id', "Employee's attendances"),
'holidays' : fields.one2many('hr.holidays', 'employee_id', "Employee's holidays"),
'workgroups' : fields.many2many('hr.timesheet.group', 'hr_timesheet_employee_rel', 'emp_id', 'tgroup_id', "Employee's work team"),
'user_id' : fields.many2one('res.users', 'Tiny ERP User'),
'category_id' : fields.many2one('hr.employee.category', 'Category'),
'regime' : fields.float('Workhours by week'),
'holiday_max' : fields.integer("Number of holidays"),
'parent_id': fields.many2one('hr.employee', 'Boss', select=True),
'child_ids': fields.one2many('hr.employee', 'parent_id','Subordinates'),
'bank_account': fields.char("Bank Account", size=16),
'position': fields.many2one('hr.employee.position', 'Positions', select=True),
'department': fields.many2one('hr.employee.department', 'Department', select=True),
'payelements' : fields.one2many('employee.setup.payelements', 'employee_id', 'Pay Elements'),
'tax_declaration': fields.one2many('employee.tax.declarations', 'employee_id', 'Tax Declarations'),
'specify_incr': fields.one2many('employee.specify.increment', 'name', 'Increment'),
}
_defaults = {
'active' : lambda *a: True,
'state' : lambda *a: 'absent',
#'basic' : lambda *a: 0.0,
}
def sign_change(self, cr, uid, ids, context={}, dt=False):
for emp in self.browse(cr, uid, ids):
if not self._action_check(cr, uid, emp.id, dt, context):
raise osv.except_osv('Warning', 'You tried to sign with a date anterior to another event !\nTry to contact the administrator to correct attendances.')
res = {'action':'action', 'employee_id':emp.id}
if dt:
res['name'] = dt
att_id = self.pool.get('hr.attendance').create(cr, uid, res)
return True
def sign_out(self, cr, uid, ids, context={}, dt=False, *args):
for emp in self.browse(cr, uid, ids):
if not self._action_check(cr, uid, emp.id, dt, context):
raise osv.except_osv('Warning', 'You tried to sign out with a date anterior to another event !\nTry to contact the administrator to correct attendances.')
res = {'action':'sign_out', 'employee_id':emp.id}
if dt:
res['name'] = dt
att_id = self.pool.get('hr.attendance').create(cr, uid, res)
self.write(cr, uid, [emp.id], {'state':'absent'})
return True
def _action_check(self, cr, uid, emp_id, dt=False,context={}):
cr.execute('select max(name) from hr_attendance where employee_id=%d', (emp_id,))
res = cr.fetchone()
return not (res and (res[0]>=(dt or time.strftime('%Y-%m-%d %H:%M:%S'))))
def sign_in(self, cr, uid, ids, context={}, dt=False, *args):
for emp in self.browse(cr, uid, ids):
if not self._action_check(cr, uid, emp.id, dt, context):
raise osv.except_osv('Warning', 'You tried to sign in with a date anterior to another event !\nTry to contact the administrator to correct attendances.')
res = {'action':'sign_in', 'employee_id':emp.id}
if dt:
res['name'] = dt
self.pool.get('hr.attendance').create(cr, uid, res)
self.write(cr, uid, [emp.id], {'state':'present'})
return True
def onchange_position(self, cr, uid, ids, state, position):
#result = {'value': {'department': 1}}
result = {'value': {'department': False}}
if position:
res = self.pool.get('hr.employee.position').browse(cr, uid, position)
result['value']['department'] = res.department.id
return result
def compute_value(self, cr, uid, emp_pay, emp_payelements, payelements):
if emp_pay.element_name.element_specific == True :
print emp_pay.name,' emp_pay.element_name.element_specific ',emp_pay.element_name.element_specific
dep_payelem_ids = emp_payelements.search(cr, uid, [('element_name','=',emp_pay.dependent_on.id)])
dep_payelems = emp_payelements.browse(cr, uid, dep_payelem_ids)
for dep_payelem in dep_payelems:
if dep_payelem.cal_value < 0:
self.compute_value(cr, uid, dep_payelem, emp_payelements, payelements)
if emp_pay.formulae == "percent" :
cal_val = dep_payelem.cal_value*(float(emp_pay.value)/100)
emp_payelements.write(cr, uid, [emp_pay.id], {'cal_value': cal_val})
if emp_pay.formulae == "addition" :
cal_val = dep_payelem.cal_value + emp_pay.value
emp_payelements.write(cr, uid, [emp_pay.id], {'cal_value': cal_val})
if emp_pay.formulae == "subtraction" :
cal_val = dep_payelem.cal_value - emp_pay.value
emp_payelements.write(cr, uid, [emp_pay.id], {'cal_value': cal_val})
if emp_pay.formulae == "multiply" :
cal_val = dep_payelem.cal_value * emp_pay.value
emp_payelements.write(cr, uid, [emp_pay.id], {'cal_value': cal_val})
print 'cal-value : ',cal_val
else:
emp_payelements.write(cr, uid, [emp_pay.id], {'cal_value': emp_pay.value})
return True
def calculate_value(self, cr, uid, ids, context={}, dt=False, *args):
emps = self.browse(cr, uid, ids)
for emp in emps:
emp_payelements = self.pool.get("employee.setup.payelements")
emp_pay_ids = emp_payelements.search(cr, uid, [('employee_id','=',emp.id)])
emp_pays = emp_payelements.browse(cr, uid, emp_pay_ids)
payelements = self.pool.get("payroll.setup.payelements")
for emp_pay in emp_pays:
self.compute_value(cr, uid, emp_pay, emp_payelements, payelements)
return True
hr_employee()
class hr_timesheet(osv.osv):
_name = "hr.timesheet"
_description = "Timesheet Line"
_columns = {
'name' : fields.char("Name", size=64, required=True),
'dayofweek': fields.selection([('0','Monday'),('1','Tuesday'),('2','Wednesday'),('3','Thursday'),('4','Friday'),('5','Saturday'),('6','Sunday')], 'Day of week'),
'date_from' : fields.date('Starting date'),
'hour_from' : fields.char('Work from', size=8),
'hour_to' : fields.char("Work to", size=8),
'tgroup_id' : fields.many2one("hr.timesheet.group", "Employee's timesheet group", select=True),
}
_order = 'dayofweek, hour_from'
hr_timesheet()
class hr_action_reason(osv.osv):
_name = "hr.action.reason"
_description = "Action reason"
_columns = {
'name' : fields.char('Reason', size=64, required=True),
'action_type' : fields.selection([('sign_in', 'Sign in'), ('sign_out', 'Sign out')], "Action's type"),
}
_defaults = {
'action_type' : lambda *a: 'sign_in',
}
hr_action_reason()
def _employee_get(obj,cr,uid,context={}):
ids = obj.pool.get('hr.employee').search(cr, uid, [('user_id','=', uid)])
if ids:
return ids[0]
return False
class hr_attendance(osv.osv):
_name = "hr.attendance"
_description = "Attendance"
_columns = {
'name' : fields.datetime('Date'),
'action' : fields.selection([('sign_in', 'Sign In'), ('sign_out', 'Sign Out'),('action','Action')], 'Action'),
'action_desc' : fields.many2one("hr.action.reason", "Action reason", domain="[('action_type', '=', action)]"),
'employee_id' : fields.many2one('hr.employee', 'Employee',required=True, select=True),
}
_defaults = {
'name' : lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
'employee_id' : _employee_get,
}
def _altern_si_so(self, cr, uid, ids):
for id in ids:
sql = '''
select action, name
from hr_attendance as att
where employee_id = (select employee_id from hr_attendance where id=%s)
and action in ('sign_in','sign_out')
and name <= (select name from hr_attendance where id=%s)
order by name desc
limit 2
''' % (id, id)
cr.execute(sql)
atts = cr.fetchall()
if not ((len(atts)==1 and atts[0][0] == 'sign_in') or (atts[0][0] != atts[1][0] and atts[0][1] != atts[1][1])):
return False
return True
_constraints = [(_altern_si_so, 'Error: Sign in (resp. Sign out) must follow Sign out (resp. Sign in)', ['action'])]
_order = 'name desc'
hr_attendance()
class hr_holidays_status(osv.osv):
_name = "hr.holidays.status"
_description = "Holidays Status"
_columns = {
'name' : fields.char('Holiday Status', size=64, required=True, translate=True),
}
hr_holidays_status()
class hr_holidays(osv.osv):
_name = "hr.holidays"
_description = "Holidays"
_columns = {
'name' : fields.char('Description', required=True, size=64),
'date_from' : fields.datetime('Vacation start day', required=True),
'date_to' : fields.datetime('Vacation end day'),
'holiday_status' : fields.many2one("hr.holidays.status", "Holiday's Status"),
'employee_id' : fields.many2one('hr.employee', 'Employee', select=True),
}
_defaults = {
'employee_id' : _employee_get
}
_order = 'date_from desc'
hr_holidays()
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|