Change mandaptry to method

employee.py

@@ -31,6 +31,25 @@ class Employee(metaclass=PoolMeta):
             else:
                 return self.contract.end_date or None
 
+    def create_mandatory_wages(self, employees_ids):
+        MandatoryWage = Pool().get('staff.payroll.mandatory_wage')
+        employees = Employee.browse(employees_ids)
+        values = []
+        for employee in employees:
+            if employee.category and employee.category.wages_default:
+                current_wages_ids = [m.wage_type.id
+                                     for m in employee.mandatory_wages]
+                for wage in employee.category.wages_default:
+                    if wage.id in current_wages_ids:
+                        continue
+                    val = {
+                        'employee': employee.id,
+                        'wage_type': wage.id,
+                    }
+                    values.append(val)
+        if values:
+            MandatoryWage.create(values)
+
 
 class MandatoryWage(ModelSQL, ModelView):
     "Mandatory Wage"
@@ -73,24 +92,7 @@ class CreateMandatoryWages(Wizard):
     create_wages = StateTransition()
 
     def transition_create_wages(self):
-        pool = Pool()
-        Employee = pool.get('company.employee')
-        MandatoryWage = pool.get('staff.payroll.mandatory_wage')
-
+        Employee = Pool().get('company.employee')
         employees_ids = Transaction().context['active_ids']
-        employees = Employee.browse(employees_ids)
-        values = []
-        for employee in employees:
-            if employee.category and employee.category.wages_default:
-                current_wages_ids = [m.wage_type.id
-                                     for m in employee.mandatory_wages]
-                for wage in employee.category.wages_default:
-                    if wage.id in current_wages_ids:
-                        continue
-                    val = {
-                        'employee': employee.id,
-                        'wage_type': wage.id,
-                    }
-                    values.append(val)
-        MandatoryWage.create(values)
+        Employee.create_mandatory_wages(employees_ids)
         return 'end'