添加功能

yuthon_salary/wizard/salary_compute_wizard.py

@@ -73,9 +73,10 @@ class SalaryComputeWizard(models.Model):
             else:
                 end_day = 28
             end_date = datetime(year, month, end_day).strftime('%Y-%m-%d')
-            payslip_run_id = self.env['hr.payslip.run'].search([('name', '=', payslip_run_name)], limit=1)
+            payslip_run_id = self.env['hr.payslip.run'].search([('name', '=', payslip_run_name), ('company_id', '=', employee.company_id.id)], limit=1)
             if not payslip_run_id:
                 payslip_run_id = self.env['hr.payslip.run'].create({
+                    'company_id': employee.company_id.id,
                     'name': payslip_run_name,
                     'date_start': start_date,
                     'date_end': end_date,
@@ -87,8 +88,8 @@ class SalaryComputeWizard(models.Model):
                 ('date_from', '<=', end_date),
                 ('date_to', '>=', start_date)
             ])
-            # if existing_payslip:
-            #     raise UserError( f"员工 {employee.name} 在 {start_date} 到 {end_date} 日期范围内已有工资单记录，无法创建新的工资单。")
+            if existing_payslip:
+                raise UserError( f"员工 {employee.name} 在 {start_date} 到 {end_date} 日期范围内已有工资单记录，无法创建新的工资单。")
             payslip = self.env['hr.payslip'].create({
                 'is_bold':is_bold,
                 'name': employee.name,