fix left short rebate bug

DaemonLab · Nov 19, 2024 · e3aa828 · e3aa828
1 parent 081553d
commit e3aa828
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/home/admin.py b/home/admin.py
@@ -790,7 +790,7 @@ def export_as_csv(self, request, queryset):
         """
         Export action available in the admin page
         """
-        resource = RebateBillsResource()
+        resource = StudentBillsResource()
         dataset = resource.export(queryset)
         response = HttpResponse(dataset.csv, content_type="text/csv")
         response["Content-Disposition"] = 'attachment; filename="RebateAutumn.csv"'
@@ -1050,9 +1050,10 @@ def Add(self, request, queryset: list[LeftShortRebate]):
                         upper_cap_check = max_days_rebate(
                             student_obj, start_date, end_date, period
                         )
-                        if upper_cap_check == 0:
-                            continue
-                        end_date = start_date + timedelta(days=upper_cap_check - 1)
+                        if upper_cap_check > 0:
+                            end_date = start_date + timedelta(
+                                days=(upper_cap_check - 1)
+                            )
                         short_rebate = Rebate(
                             email=student_obj,
                             allocation_id=allocation,