initial commit 2

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