Initial commit of *new* hr_payroll_timesheet and hr_payroll_timesheet_old for 11.0

hr_payroll_timesheet_old/hr_payslip.py

@@ -0,0 +1,89 @@
+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