mirror of
https://gitlab.com/hibou-io/hibou-odoo/suite.git
synced 2025-01-20 12:37:31 +02:00
90 lines
3.4 KiB
Python
Executable File
90 lines
3.4 KiB
Python
Executable File
from datetime import datetime
|
|
from collections import defaultdict
|
|
from odoo import api, models
|
|
from odoo.exceptions import ValidationError
|
|
from odoo.tools import DEFAULT_SERVER_DATETIME_FORMAT
|
|
|
|
|
|
class HrPayslip(models.Model):
|
|
_inherit = 'hr.payslip'
|
|
|
|
@api.model
|
|
def get_worked_day_lines(self, contracts, date_from, date_to):
|
|
def create_empty_worked_lines(employee, contract, date_from, date_to):
|
|
attn = {
|
|
'name': 'Attendance',
|
|
'sequence': 10,
|
|
'code': 'ATTN',
|
|
'number_of_days': 0.0,
|
|
'number_of_hours': 0.0,
|
|
'contract_id': contract.id,
|
|
}
|
|
|
|
valid_attn = [
|
|
('employee_id', '=', employee.id),
|
|
('check_in', '>=', date_from),
|
|
('check_in', '<=', date_to),
|
|
]
|
|
return attn, valid_attn
|
|
|
|
work = []
|
|
for contract in contracts:
|
|
worked_attn, valid_attn = create_empty_worked_lines(
|
|
contract.employee_id,
|
|
contract,
|
|
date_from,
|
|
date_to
|
|
)
|
|
days = set()
|
|
for attn in self.env['hr.attendance'].search(valid_attn):
|
|
if not attn.check_out:
|
|
raise ValidationError('This pay period must not have any open attendances.')
|
|
if attn.worked_hours:
|
|
# Avoid in/outs
|
|
attn_start_time = datetime.strptime(attn.check_in, DEFAULT_SERVER_DATETIME_FORMAT)
|
|
attn_iso = attn_start_time.isocalendar()
|
|
if not attn_iso in days:
|
|
worked_attn['number_of_days'] += 1
|
|
days.add(attn_iso)
|
|
worked_attn['number_of_hours'] += attn.worked_hours
|
|
worked_attn['number_of_hours'] = round(worked_attn['number_of_hours'], 2)
|
|
work.append(worked_attn)
|
|
|
|
res = super(HrPayslip, self).get_worked_day_lines(contracts, date_from, date_to)
|
|
res.extend(work)
|
|
return res
|
|
|
|
@api.multi
|
|
def hour_break_down(self, code):
|
|
"""
|
|
:param code: what kind of worked days you need aggregated
|
|
:return: dict: keys are isocalendar tuples, values are hours.
|
|
"""
|
|
self.ensure_one()
|
|
if code == 'ATTN':
|
|
attns = self.env['hr.attendance'].search([
|
|
('employee_id', '=', self.employee_id.id),
|
|
('check_in', '>=', self.date_from),
|
|
('check_in', '<=', self.date_to),
|
|
])
|
|
day_values = defaultdict(float)
|
|
for attn in attns:
|
|
if not attn.check_out:
|
|
raise ValidationError('This pay period must not have any open attendances.')
|
|
if attn.worked_hours:
|
|
# Avoid in/outs
|
|
attn_start_time = datetime.strptime(attn.check_in, DEFAULT_SERVER_DATETIME_FORMAT)
|
|
attn_iso = attn_start_time.isocalendar()
|
|
day_values[attn_iso] += attn.worked_hours
|
|
return day_values
|
|
elif hasattr(super(HrPayslip, self), 'hours'):
|
|
return super(HrPayslip, self).hours(code)
|
|
|
|
@api.multi
|
|
def hours_break_down_week(self, code):
|
|
days = self.hour_break_down(code)
|
|
weeks = defaultdict(float)
|
|
for isoday, hours in days.items():
|
|
weeks[isoday[1]] += hours
|
|
return weeks
|