diff options
Diffstat (limited to 'addons/hr_work_entry/models/resource.py')
| -rw-r--r-- | addons/hr_work_entry/models/resource.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/addons/hr_work_entry/models/resource.py b/addons/hr_work_entry/models/resource.py new file mode 100644 index 00000000..ac7fa6c9 --- /dev/null +++ b/addons/hr_work_entry/models/resource.py @@ -0,0 +1,29 @@ +# -*- coding:utf-8 -*- +# Part of Odoo. See LICENSE file for full copyright and licensing details. + +from odoo import models, fields + + +class ResourceCalendarAttendance(models.Model): + _inherit = 'resource.calendar.attendance' + + def _default_work_entry_type_id(self): + return self.env.ref('hr_work_entry.work_entry_type_attendance', raise_if_not_found=False) + + work_entry_type_id = fields.Many2one('hr.work.entry.type', 'Work Entry Type', default=_default_work_entry_type_id) + + def _copy_attendance_vals(self): + res = super()._copy_attendance_vals() + res['work_entry_type_id'] = self.work_entry_type_id.id + return res + + +class ResourceCalendarLeave(models.Model): + _inherit = 'resource.calendar.leaves' + + work_entry_type_id = fields.Many2one('hr.work.entry.type', 'Work Entry Type') + + def _copy_leave_vals(self): + res = super()._copy_leave_vals() + res['work_entry_type_id'] = self.work_entry_type_id.id + return res |
