276 lines
9.2 KiB
Python
276 lines
9.2 KiB
Python
#This file is part of Tryton. The COPYRIGHT file at the top level of this repository contains the full copyright notices and license terms.
|
|
"Timesheet Line"
|
|
|
|
from trytond.osv import fields, OSV
|
|
from trytond.wizard import Wizard, WizardOSV
|
|
import datetime
|
|
|
|
|
|
class Line(OSV):
|
|
'Timesheet Line'
|
|
_name = 'timesheet.line'
|
|
_description = __doc__
|
|
|
|
employee = fields.Many2One('company.employee', 'Employee', required=True,
|
|
select=1, domain="[('company', '=', company)]")
|
|
date = fields.Date('Date', required=True, select=1)
|
|
hours = fields.Float('Hours', digits=(16, 2), required=True)
|
|
work = fields.Many2One('timesheet.work', 'Work',
|
|
required=True, select=1, domain=[
|
|
('type', '!=', 'view'),
|
|
('state', '!=', 'closed'),
|
|
])
|
|
description = fields.Char('Description')
|
|
|
|
def __init__(self):
|
|
super(Line, self).__init__()
|
|
self._sql_constraints += [
|
|
('check_move_hours_pos',
|
|
'CHECK(hours >= 0.0)', 'Hours field must be positive'),
|
|
]
|
|
|
|
def default_employee(self, cursor, user_id, context=None):
|
|
user_obj = self.pool.get('res.user')
|
|
employee_obj = self.pool.get('company.employee')
|
|
|
|
if context is None:
|
|
context = {}
|
|
employee_id = None
|
|
if context.get('employee'):
|
|
employee_id = context['employee']
|
|
else:
|
|
user = user_obj.browse(cursor, user_id, user_id, context=context)
|
|
if user.employee:
|
|
employee_id = user.employee.id
|
|
if employee_id:
|
|
return employee_obj.name_get(cursor, user_id, employee_id,
|
|
context=context)[0]
|
|
return False
|
|
|
|
def default_date(self, cursor, user, context=None):
|
|
if context is None:
|
|
context = {}
|
|
if context.get('date'):
|
|
return context['date']
|
|
return datetime.date.today()
|
|
|
|
def view_header_get(self, cursor, user, value, view_type='form',
|
|
context=None):
|
|
if not context.get('employee'):
|
|
return False
|
|
employee_obj = self.pool.get('company.employee')
|
|
employee = employee_obj.browse(cursor, user, context['employee'],
|
|
context=context)
|
|
return value + " (" + employee.name + ")"
|
|
|
|
Line()
|
|
|
|
|
|
class EnterLinesInit(WizardOSV):
|
|
_name = 'timesheet.enter_lines.init'
|
|
employee = fields.Many2One('company.employee', 'Employee', required=True,
|
|
domain="[('company', '=', company)]")
|
|
date = fields.Date('Date', required=True)
|
|
|
|
def default_employee(self, cursor, user, context=None):
|
|
line_obj = self.pool.get('timesheet.line')
|
|
return line_obj.default_employee(cursor, user, context=context)
|
|
|
|
def default_date(self, cursor, user, context=None):
|
|
line_obj = self.pool.get('timesheet.line')
|
|
return line_obj.default_date(cursor, user, context=context)
|
|
|
|
EnterLinesInit()
|
|
|
|
|
|
class EnterLines(Wizard):
|
|
'Enter Lines'
|
|
_name = 'timesheet.enter_lines'
|
|
states = {
|
|
'init': {
|
|
'result': {
|
|
'type': 'form',
|
|
'object': 'timesheet.enter_lines.init',
|
|
'state': [
|
|
('end', 'Cancel', 'tryton-cancel'),
|
|
('enter', 'Enter', 'tryton-ok', True),
|
|
],
|
|
},
|
|
},
|
|
'enter': {
|
|
'result': {
|
|
'type': 'action',
|
|
'action': '_action_enter_lines',
|
|
'state': 'end',
|
|
},
|
|
}
|
|
}
|
|
|
|
def _action_enter_lines(self, cursor, user, data, context=None):
|
|
model_data_obj = self.pool.get('ir.model.data')
|
|
act_window_obj = self.pool.get('ir.action.act_window')
|
|
employee_obj = self.pool.get('company.employee')
|
|
|
|
model_data_ids = model_data_obj.search(cursor, user, [
|
|
('fs_id', '=', 'act_line_form'),
|
|
('module', '=', 'timesheet'),
|
|
], limit=1, context=context)
|
|
model_data = model_data_obj.browse(cursor, user, model_data_ids[0],
|
|
context=context)
|
|
res = act_window_obj.read(cursor, user, model_data.db_id, context=context)
|
|
res['domain'] = str([
|
|
('employee', '=', data['form']['employee']),
|
|
('date', '=', data['form']['date']),
|
|
])
|
|
res['context'] = str({
|
|
'employee': data['form']['employee'],
|
|
'date': data['form']['date'],
|
|
})
|
|
|
|
if data['form']['employee']:
|
|
employee_name = employee_obj.name_get(
|
|
cursor, user, data['form']['employee'], context=context)
|
|
res['name'] += " - " + employee_name[0][1]
|
|
|
|
return res
|
|
|
|
EnterLines()
|
|
|
|
|
|
class HoursEmployee(OSV):
|
|
'Hours per Employee'
|
|
_name = 'timesheet.hours_employee'
|
|
_description = __doc__
|
|
|
|
employee = fields.Many2One('company.employee', 'Employee', select=1)
|
|
hours = fields.Float('Hours', digits=(16, 2))
|
|
|
|
def table_query(self, context=None):
|
|
if context is None:
|
|
context = {}
|
|
clause = ' '
|
|
args = []
|
|
if context.get('start_date'):
|
|
clause += 'AND date >= %s '
|
|
args.append(context['start_date'])
|
|
if context.get('end_date'):
|
|
clause += 'AND date <= %s '
|
|
args.append(context['end_date'])
|
|
return ('SELECT DISTINCT(employee) AS id, employee, ' \
|
|
'SUM(COALESCE(hours, 0)) AS hours ' \
|
|
'FROM timesheet_line ' \
|
|
'WHERE True ' \
|
|
+ clause + \
|
|
'GROUP BY employee', args)
|
|
|
|
HoursEmployee()
|
|
|
|
|
|
class OpenHoursEmployeeInit(WizardOSV):
|
|
_name = 'timesheet.open_hours_employee.init'
|
|
start_date = fields.Date('Start Date')
|
|
end_date = fields.Date('End Date')
|
|
|
|
OpenHoursEmployeeInit()
|
|
|
|
|
|
class OpenHoursEmployee(Wizard):
|
|
'Open Hours per Employee'
|
|
_name = 'timesheet.open_hours_employee'
|
|
states = {
|
|
'init': {
|
|
'result': {
|
|
'type': 'form',
|
|
'object': 'timesheet.open_hours_employee.init',
|
|
'state': [
|
|
('end', 'Cancel', 'tryton-cancel'),
|
|
('open', 'Open', 'tryton-ok', True),
|
|
],
|
|
},
|
|
},
|
|
'open': {
|
|
'result': {
|
|
'type': 'action',
|
|
'action': '_action_open',
|
|
'state': 'end',
|
|
},
|
|
},
|
|
}
|
|
|
|
def _action_open(self, cursor, user, data, context=None):
|
|
model_data_obj = self.pool.get('ir.model.data')
|
|
act_window_obj = self.pool.get('ir.action.act_window')
|
|
|
|
model_data_ids = model_data_obj.search(cursor, user, [
|
|
('fs_id', '=', 'act_hours_employee_form'),
|
|
('module', '=', 'timesheet'),
|
|
], limit=1, context=context)
|
|
model_data = model_data_obj.browse(cursor, user, model_data_ids[0],
|
|
context=context)
|
|
res = act_window_obj.read(cursor, user, model_data.db_id, context=context)
|
|
res['context'] = str({
|
|
'start_date': data['form']['start_date'],
|
|
'end_date': data['form']['end_date'],
|
|
})
|
|
return res
|
|
|
|
OpenHoursEmployee()
|
|
|
|
|
|
class HoursEmployeeWeekly(OSV):
|
|
'Hours per Employee per Week'
|
|
_name = 'timesheet.hours_employee_weekly'
|
|
_description = __doc__
|
|
|
|
year = fields.Integer('Year', select=1)
|
|
week = fields.Integer('Week', select=1)
|
|
employee = fields.Many2One('company.employee', 'Employee', select=1)
|
|
hours = fields.Float('Hours', digits=(16, 2), select=1)
|
|
|
|
def __init__(self):
|
|
super(HoursEmployeeWeekly, self).__init__()
|
|
self._order.insert(0, ('year', 'DESC'))
|
|
self._order.insert(1, ('week', 'DESC'))
|
|
self._order.insert(2, ('employee', 'ASC'))
|
|
|
|
def table_query(self, context=None):
|
|
return ('SELECT EXTRACT(WEEK FROM date) + ' \
|
|
'EXTRACT(YEAR FROM date) * 100 + ' \
|
|
'employee * 1000000 AS id, ' \
|
|
'EXTRACT(YEAR FROM date) AS year, ' \
|
|
'EXTRACT(WEEK FROM date) AS week, employee, ' \
|
|
'SUM(COALESCE(hours, 0)) AS hours ' \
|
|
'FROM timesheet_line ' \
|
|
'GROUP BY year, week, employee', [])
|
|
|
|
HoursEmployeeWeekly()
|
|
|
|
|
|
class HoursEmployeeMonthly(OSV):
|
|
'Hours per Employee per Month'
|
|
_name = 'timesheet.hours_employee_monthly'
|
|
_description = __doc__
|
|
|
|
year = fields.Integer('Year', select=1)
|
|
month = fields.Integer('Month', select=1)
|
|
employee = fields.Many2One('company.employee', 'Employee', select=1)
|
|
hours = fields.Float('Hours', digits=(16, 2), select=1)
|
|
|
|
def __init__(self):
|
|
super(HoursEmployeeMonthly, self).__init__()
|
|
self._order.insert(0, ('year', 'DESC'))
|
|
self._order.insert(1, ('month', 'DESC'))
|
|
self._order.insert(2, ('employee', 'ASC'))
|
|
|
|
def table_query(self, context=None):
|
|
return ('SELECT EXTRACT(MONTH FROM date) + ' \
|
|
'EXTRACT(YEAR FROM date) * 100 + ' \
|
|
'employee * 1000000 AS id, ' \
|
|
'EXTRACT(YEAR FROM date) AS year, ' \
|
|
'EXTRACT(MONTH FROM date) AS month, employee, ' \
|
|
'SUM(COALESCE(hours, 0)) AS hours ' \
|
|
'FROM timesheet_line ' \
|
|
'GROUP BY year, month, employee', [])
|
|
|
|
HoursEmployeeMonthly()
|