diff --git a/yuthon_hr_attendance/models/hr_attendance.py b/yuthon_hr_attendance/models/hr_attendance.py index 749ab8a8d..e38533bc6 100644 --- a/yuthon_hr_attendance/models/hr_attendance.py +++ b/yuthon_hr_attendance/models/hr_attendance.py @@ -1158,17 +1158,17 @@ class HrAttendance(models.Model): else: record.shift_name = '行政班次' - # @api.model - # def create(self, vals): - # record = super(HrAttendance, self).create(vals) - # # 未来日期不处理班次(保持原逻辑) - # if 'current_data' in vals: - # current_data = fields.Date.to_date(vals['current_data']) - # today = fields.Date.today() - # if current_data > today: - # return record - # record._set_shift_name() - # return record + @api.model + def create(self, vals): + record = super(HrAttendance, self).create(vals) + # 未来日期不处理班次(保持原逻辑) + if 'current_data' in vals: + current_data = fields.Date.to_date(vals['current_data']) + today = fields.Date.today() + if current_data > today: + return record + record._set_shift_name() + return record def write(self, vals): result = super(HrAttendance, self).write(vals)