fix: changed migration to update only the current year balance

Author: fatmaebrahim

server/cshr/migrations/0036_update_excuse_balance.py

@@ -1,10 +1,12 @@
 from django.db import migrations, models
+from django.utils import timezone
 
 
 def update_excuse_balance(apps, schema_editor):
     UserVacationBalance = apps.get_model("cshr", "UserVacationBalance")
+    current_year = timezone.now().year
 
-    for user_balance in UserVacationBalance.objects.select_related("total_days", "remaining_days").all():
+    for user_balance in UserVacationBalance.objects.filter(year=current_year).select_related("total_days", "remaining_days"):
         if user_balance.total_days:
             user_balance.total_days.excuse = 12
             user_balance.total_days.save(update_fields=["excuse"])
@@ -15,8 +17,9 @@ def update_excuse_balance(apps, schema_editor):
 
 def reverse_excuse_balance(apps, schema_editor):
     UserVacationBalance = apps.get_model("cshr", "UserVacationBalance")
+    current_year = timezone.now().year
 
-    for user_balance in UserVacationBalance.objects.select_related("total_days", "remaining_days").all():
+    for user_balance in UserVacationBalance.objects.filter(year=current_year).select_related("total_days", "remaining_days"):
         if user_balance.total_days:
             user_balance.total_days.excuse = 6
             user_balance.total_days.save(update_fields=["excuse"])