From 6dea6801e26e24f7c65671321c3fe64678ef2435 Mon Sep 17 00:00:00 2001
From: zhouzhou <123@qq.com>
Date: Wed, 18 Dec 2024 18:07:00 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../views/hr_attendance_views.xml             |   4 +
 yuthon_wechat/models/hr_attendance_detail.py  | 179 +++++++++---------
 2 files changed, 94 insertions(+), 89 deletions(-)

diff --git a/yuthon_hr_attendance/views/hr_attendance_views.xml b/yuthon_hr_attendance/views/hr_attendance_views.xml
index 0db0c285c..541008031 100644
--- a/yuthon_hr_attendance/views/hr_attendance_views.xml
+++ b/yuthon_hr_attendance/views/hr_attendance_views.xml
@@ -48,6 +48,7 @@
                 <field name="work_time_type" optional="hidden"/>
                 <field name="check_in" optional="show" widget="time"/>
                 <field name="check_out" optional="show" widget="time"/>
+                <field name="shift_id" optional="show"/>
                 <field name="shift_name" optional="show"/>
                 <field name="is_approval" column_invisible="1"/>
                 <button name="button_leave" type="object" string="请假" class="oe_highlight"
@@ -81,6 +82,7 @@
                 <field name="noon_duty" optional="show" widget="time"/>
                 <field name="night_duty" optional="show" widget="time"/>
                 <field name="check_out" optional="show" widget="time"/>
+                <field name="shift_id" optional="show"/>
                 <field name="shift_name" optional="show"/>
                 <field name="is_approval" column_invisible="1"/>
                 <button name="button_leave" type="object" string="请假" class="oe_highlight"
@@ -113,6 +115,7 @@
                 <field name="check_in" optional="show" widget="time"/>
                 <field name="noon_duty" optional="show" widget="time"/>
                 <field name="check_out" optional="show" widget="time"/>
+                <field name="shift_id" optional="show"/>
                 <field name="shift_name" optional="show"/>
                 <field name="is_approval" column_invisible="1"/>
                 <button name="button_leave" type="object" string="请假" class="oe_highlight"
@@ -145,6 +148,7 @@
                 <field name="check_in" optional="show" widget="time"/>
                 <field name="noon_duty" optional="show" widget="time"/>
                 <field name="check_out" optional="show" widget="time"/>
+                <field name="shift_id" optional="show"/>
                 <field name="shift_name" optional="show"/>
                 <field name="is_approval" column_invisible="1"/>
                 <button name="button_leave" type="object" string="请假" class="oe_highlight"
diff --git a/yuthon_wechat/models/hr_attendance_detail.py b/yuthon_wechat/models/hr_attendance_detail.py
index 24abc164e..25e37145d 100644
--- a/yuthon_wechat/models/hr_attendance_detail.py
+++ b/yuthon_wechat/models/hr_attendance_detail.py
@@ -266,96 +266,97 @@ class HrAttendanceDetail(models.Model):
             exception_type = data.get("exception_type", "")
             will_datetime_start = checkin_time.replace(hour=min_hour, minute=min_minute)
             will_datetime_end = checkin_time.replace(hour=max_hour, minute=max_minute)
-            # if not hr_employee_obj.default_planning_role_id:
-            if exception_type != "未打卡":
-                if checkin_time and checkin_type == "上班打卡" and exception_type != "未打卡" and not attendance_obj.check_in:
-                    attendance_obj.sudo().write({
-                        "sch_check_in_time": sch_checkin_time,
-                        "check_in": checkin_time,
-                        "is_late_card": checkin_time > will_datetime_start,
+            if not hr_employee_obj.default_planning_role_id:
+                if exception_type != "未打卡":
+                    if checkin_time and checkin_type == "上班打卡" and exception_type != "未打卡" and not attendance_obj.check_in:
+                        attendance_obj.sudo().write({
+                            "sch_check_in_time": sch_checkin_time,
+                            "check_in": checkin_time,
+                            "is_late_card": checkin_time > will_datetime_start,
+                        })
+                    elif checkin_time and checkin_type == "下班打卡" and exception_type != "未打卡" and not attendance_obj.check_out:
+                        attendance_obj.sudo().write({
+                            "sch_check_out_time": sch_checkin_time,
+                            "check_out": checkin_time,
+                            "is_early_card": will_datetime_end > checkin_time,
+                        })
+                    elif attendance_obj.check_in and checkin_time and checkin_type == "上班打卡" and exception_type != "未打卡" and checkin_time < attendance_obj.check_in:
+                        attendance_obj.sudo().write({
+                            "sch_check_in_time": sch_checkin_time,
+                            "check_in": checkin_time,
+                            "is_late_card": checkin_time > will_datetime_start,
+                        })
+                    elif attendance_obj.check_out and checkin_time and checkin_type == "下班打卡" and exception_type != "未打卡" and checkin_time > attendance_obj.check_out:
+                        attendance_obj.sudo().write({
+                            "sch_check_out_time": sch_checkin_time,
+                            "check_out": checkin_time,
+                            "is_early_card": will_datetime_end > checkin_time,
+                        })
+                    elif checkin_time and exception_type != "未打卡" and not attendance_obj.check_in and not attendance_obj.check_out:
+                        attendance_obj.sudo().write({
+                            "sch_check_in_time": sch_checkin_time,
+                            "check_in": checkin_time,
+                            "is_late_card": checkin_time > will_datetime_start,
+                        })
+                    elif checkin_time and exception_type != "未打卡" and attendance_obj.check_in and not attendance_obj.check_out:
+                        attendance_obj.sudo().write({
+                            "sch_check_out_time": sch_checkin_time,
+                            "check_out": checkin_time,
+                            "is_early_card": will_datetime_end > checkin_time,
+                        })
+                    elif attendance_obj.is_early_card or attendance_obj.is_late_card:
+                        attendance_obj.sudo().write({"is_abnormal": True})
+                else:
+                    attendance_obj.sudo().write({"is_no_card": True, "is_abnormal": True})
+            elif hr_employee_obj.default_planning_role_id.name == '保安部':
+                current_datetime_start = datetime.datetime.combine(datetime.datetime.strptime(current_data, '%Y-%m-%d'), datetime.time.min)
+                current_datetime_end = datetime.datetime.combine(datetime.datetime.strptime(current_data, '%Y-%m-%d'), datetime.time.max)
+                planning_slot_id = self.env['planning.slot'].search([('resource_id', '=', hr_employee_obj.resource_id.id),
+                                                                     ('start_datetime', '>=', current_datetime_start),
+                                                                     ('start_datetime', '<=', current_datetime_end)])
+                if planning_slot_id:
+                    attendance_obj = self.env['hr.attendance'].sudo().create({
+                        'employee_id': hr_employee_obj.id,
+                        "userid": userid,
+                        "current_data": current_data,
+                        "shift_id": planning_slot_id.id,
+                        "attendance_type": "wx_work"
                     })
-                elif checkin_time and checkin_type == "下班打卡" and exception_type != "未打卡" and not attendance_obj.check_out:
-                    attendance_obj.sudo().write({
-                        "sch_check_out_time": sch_checkin_time,
-                        "check_out": checkin_time,
-                        "is_early_card": will_datetime_end > checkin_time,
-                    })
-                elif attendance_obj.check_in and checkin_time and checkin_type == "上班打卡" and exception_type != "未打卡" and checkin_time < attendance_obj.check_in:
-                    attendance_obj.sudo().write({
-                        "sch_check_in_time": sch_checkin_time,
-                        "check_in": checkin_time,
-                        "is_late_card": checkin_time > will_datetime_start,
-                    })
-                elif attendance_obj.check_out and checkin_time and checkin_type == "下班打卡" and exception_type != "未打卡" and checkin_time > attendance_obj.check_out:
-                    attendance_obj.sudo().write({
-                        "sch_check_out_time": sch_checkin_time,
-                        "check_out": checkin_time,
-                        "is_early_card": will_datetime_end > checkin_time,
-                    })
-                elif checkin_time and exception_type != "未打卡" and not attendance_obj.check_in and not attendance_obj.check_out:
-                    attendance_obj.sudo().write({
-                        "sch_check_in_time": sch_checkin_time,
-                        "check_in": checkin_time,
-                        "is_late_card": checkin_time > will_datetime_start,
-                    })
-                elif checkin_time and exception_type != "未打卡" and attendance_obj.check_in and not attendance_obj.check_out:
-                    attendance_obj.sudo().write({
-                        "sch_check_out_time": sch_checkin_time,
-                        "check_out": checkin_time,
-                        "is_early_card": will_datetime_end > checkin_time,
-                    })
-                elif attendance_obj.is_early_card or attendance_obj.is_late_card:
-                    attendance_obj.sudo().write({"is_abnormal": True})
-            else:
-                attendance_obj.sudo().write({"is_no_card": True, "is_abnormal": True})
-            # else:
-            #     current_datetime_start = datetime.datetime.combine(datetime.datetime.strptime(current_data, '%Y-%m-%d'), datetime.time.min)
-            #     current_datetime_end = datetime.datetime.combine(datetime.datetime.strptime(current_data, '%Y-%m-%d'), datetime.time.max)
-            #     planning_slot_id = self.env['planning.slot'].search([('resource_id', '=', hr_employee_obj.resource_id.id),
-            #                                                          ('start_datetime', '>=', current_datetime_start),
-            #                                                          ('start_datetime', '<=', current_datetime_end)])
-            #     if planning_slot_id:
-            #         if planning_slot_id.department_id.name == '保安部':
-            #             attendance_obj = self.env['hr.attendance'].sudo().create({
-            #                 'employee_id': hr_employee_obj.id,
-            #                 "userid": userid,
-            #                 "current_data": current_data,
-            #                 "shift_id": planning_slot_id.id,
-            #                 "attendance_type": "wx_work"
-            #             })
-            #             if exception_type != "未打卡":
-            #                 if checkin_time and checkin_type == "上班打卡" and not attendance_obj.check_in:
-            #                     attendance_obj.sudo().write({
-            #                         "check_in": checkin_time,
-            #                         "is_late_card": checkin_time > planning_slot_id.start_datetime,
-            #                     })
-            #                 elif checkin_time and checkin_type == "下班打卡" and not attendance_obj.check_out:
-            #                     attendance_obj.sudo().write({
-            #                         "check_out": checkin_time,
-            #                         "is_early_card": planning_slot_id.end_datetime > checkin_time,
-            #                     })
-            #                 elif attendance_obj.check_in and checkin_time and checkin_type == "上班打卡" and checkin_time < attendance_obj.check_in:
-            #                     attendance_obj.sudo().write({
-            #                         "check_in": checkin_time,
-            #                         "is_late_card": checkin_time > planning_slot_id.start_datetime,
-            #                     })
-            #                 elif attendance_obj.check_out and checkin_time and checkin_type == "下班打卡" and checkin_time > attendance_obj.check_out:
-            #                     attendance_obj.sudo().write({
-            #                         "check_out": checkin_time,
-            #                         "is_early_card": planning_slot_id.end_datetime > checkin_time,
-            #                     })
-            #                 elif checkin_time and not attendance_obj.check_in and not attendance_obj.check_out:
-            #                     attendance_obj.sudo().write({
-            #                         "check_in": checkin_time,
-            #                         "is_late_card": checkin_time > planning_slot_id.start_datetime,
-            #                     })
-            #                 elif checkin_time and attendance_obj.check_in and not attendance_obj.check_out:
-            #                     attendance_obj.sudo().write({
-            #                         "check_out": checkin_time,
-            #                         "is_early_card": planning_slot_id.end_datetime > checkin_time,
-            #                     })
-            #                 elif attendance_obj.is_early_card or attendance_obj.is_late_card:
-            #                     attendance_obj.sudo().write({"is_abnormal": True})
+                    if exception_type != "未打卡":
+                        if checkin_time and checkin_type == "上班打卡" and not attendance_obj.check_in:
+                            attendance_obj.sudo().write({
+                                "check_in": checkin_time,
+                                "is_late_card": checkin_time > planning_slot_id.start_datetime,
+                            })
+                        elif checkin_time and checkin_type == "下班打卡" and not attendance_obj.check_out:
+                            attendance_obj.sudo().write({
+                                "check_out": checkin_time,
+                                "is_early_card": planning_slot_id.end_datetime > checkin_time,
+                            })
+                        elif attendance_obj.check_in and checkin_time and checkin_type == "上班打卡" and checkin_time < attendance_obj.check_in:
+                            attendance_obj.sudo().write({
+                                "check_in": checkin_time,
+                                "is_late_card": checkin_time > planning_slot_id.start_datetime,
+                            })
+                        elif attendance_obj.check_out and checkin_time and checkin_type == "下班打卡" and checkin_time > attendance_obj.check_out:
+                            attendance_obj.sudo().write({
+                                "check_out": checkin_time,
+                                "is_early_card": planning_slot_id.end_datetime > checkin_time,
+                            })
+                        elif checkin_time and not attendance_obj.check_in and not attendance_obj.check_out:
+                            attendance_obj.sudo().write({
+                                "check_in": checkin_time,
+                                "is_late_card": checkin_time > planning_slot_id.start_datetime,
+                            })
+                        elif checkin_time and attendance_obj.check_in and not attendance_obj.check_out:
+                            attendance_obj.sudo().write({
+                                "check_out": checkin_time,
+                                "is_early_card": planning_slot_id.end_datetime > checkin_time,
+                            })
+                        elif attendance_obj.is_early_card or attendance_obj.is_late_card:
+                            attendance_obj.sudo().write({"is_abnormal": True})
+                    else:
+                        attendance_obj.sudo().write({"is_no_card": True, "is_abnormal": True})
                     # elif planning_slot_id.department_id.name == '工程部':
                     #     attendance_obj = self.env['hr.attendance'].sudo().create({
                     #         'employee_id': hr_employee_obj.id,