添加功能

yuthon_will_task/models/inherit_room_booking.py

@@ -28,6 +28,7 @@ class InheritRoomBooking(models.Model):
         new_record = super(InheritRoomBooking, self).create(vals)
         employee_list = []
         booking = new_record
+
         # ① 组织者
         if booking.organizer_id:
             employee_list.append(booking.organizer_id.employee_id.id)
@@ -54,7 +55,7 @@ class InheritRoomBooking(models.Model):
                 role_employees = self.env['hr.employee'].search([
                     '|',
                     ('user_role_id', 'in', role_group_roles.ids),
-                    ('user_role_ids', 'in', role_group_roles.ids)
+                    ('user_ids', 'in', role_group_roles.ids)
                 ])
                 employee_list.extend(role_employees.ids)
         # ⑦ 主持人
@@ -63,6 +64,7 @@ class InheritRoomBooking(models.Model):
 
         # 去重
         employee_list = list(set(employee_list))
+
         def get_hour_selection(datetime_obj):
             if not datetime_obj:
                 return None
@@ -92,22 +94,23 @@ class InheritRoomBooking(models.Model):
                     return f"{hour % 24}.75"
 
         start_datetime = new_record.start_datetime
-        start_date = start_datetime.date()
+        start_date = start_datetime.date() if start_datetime else False
         request_hour_from = get_hour_selection(new_record.start_datetime)
         request_hour_to = get_hour_selection(new_record.stop_datetime)
 
-        # ====================== 创建任务 ======================
+        task_time_vals = vals.get('task_time_ids', [])
         task_id = self.env['yuthon.will.task'].sudo().create({
             'name': new_record.name,
             'employee_ids': [(6, 0, employee_list)],
-            'task_time_ids': [(6, 0, new_record.task_time_ids.ids)] if new_record.task_time_ids else [],
+            'task_time_ids': task_time_vals,
             'start_date': start_date,
             'request_hour_from': request_hour_from,
             'request_hour_to': request_hour_to,
-            'address': new_record.room_id.name,
+            'address': new_record.room_id.name if new_record.room_id else '',
             'room_booking_id': new_record.id,
             'task_type': 'meet',
             'task_state': 'will',
         })
+
         task_id._onchange_employee_ids()
         return new_record