Use periods property on PayModifier

uktrade · Dec 17, 2024 · a9771fc · a9771fc
1 parent b4cb1a0
commit a9771fc
Showing 1 changed file with 1 addition and 17 deletions.
diff --git a/payroll/services/payroll.py b/payroll/services/payroll.py
@@ -345,23 +345,7 @@ def get_pay_modifiers_data(
         financial_year=financial_year,
     )
     for obj in qs:
-        yield PayModifiers(
-            id=obj.pk,
-            pay_modifiers=[
-                obj.apr,
-                obj.may,
-                obj.jun,
-                obj.jul,
-                obj.aug,
-                obj.sep,
-                obj.oct,
-                obj.nov,
-                obj.dec,
-                obj.jan,
-                obj.feb,
-                obj.mar,
-            ],
-        )
+        yield PayModifiers(id=obj.pk, pay_modifiers=obj.periods)
 
 
 @transaction.atomic