add new module

hr_employee_shift/models/hr_employee_contract.py

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+from odoo.exceptions import Warning
+from odoo import models, fields, api, _
+
+
+class HrEmployeeContract(models.Model):
+    _inherit = 'hr.contract'
+
+    shift_schedule = fields.One2many('hr.shift.schedule', 'rel_hr_schedule', string="Shift Schedule", help="Shift schedule")
+    working_hours = fields.Many2one('resource.calendar', string='Working Schedule', help="Working hours")
+    department_id = fields.Many2one('hr.department', string="Department", help="Department",
+                                    required=True)
+
+
+class HrSchedule(models.Model):
+    _name = 'hr.shift.schedule'
+
+    start_date = fields.Date(string="Date From", required=True, help="Starting date for the shift")
+    end_date = fields.Date(string="Date To", required=True, help="Ending date for the shift")
+    rel_hr_schedule = fields.Many2one('hr.contract')
+    hr_shift = fields.Many2one('resource.calendar', string="Shift", required=True, help="Shift")
+    company_id = fields.Many2one('res.company', string='Company', help="Company")
+
+    @api.onchange('start_date', 'end_date')
+    def get_department(self):
+        """Adding domain to  the hr_shift field"""
+        hr_department = None
+        if self.start_date:
+            hr_department = self.rel_hr_schedule.department_id.id
+        return {
+            'domain': {
+                'hr_shift': [('hr_department', '=', hr_department)]
+            }
+        }
+
+    
+    def write(self, vals):
+        self._check_overlap(vals)
+        return super(HrSchedule, self).write(vals)
+
+    @api.model
+    def create(self, vals):
+        self._check_overlap(vals)
+        return super(HrSchedule, self).create(vals)
+
+    def _check_overlap(self, vals):
+        if vals.get('start_date', False) and vals.get('end_date', False):
+            shifts = self.env['hr.shift.schedule'].search([('rel_hr_schedule', '=', vals.get('rel_hr_schedule'))])
+            for each in shifts:
+                if each != shifts[-1]:
+                    if each.end_date >= vals.get('start_date') or each.start_date >= vals.get('start_date'):
+                        raise Warning(_('The dates may not overlap with one another.'))
+            if vals.get('start_date') > vals.get('end_date'):
+                raise Warning(_('Start date should be less than end date.'))
+        return True
+