admin_action (#96)

home/admin.py

@@ -4,6 +4,8 @@
 For more information please see: https://docs.djangoproject.com/en/4.1/ref/contrib/admin/
 """
 
+import csv
+
 from django.contrib import admin
 from django.http import HttpResponse
 from import_export.admin import ImportExportMixin, ImportExportModelAdmin
@@ -466,7 +468,7 @@ def get_queryset(self, request):
     def name(self, obj):
         return obj.email.name
 
-    actions = ["export_as_csv", "disapprove", "approve"]
+    actions = ["export_as_csv", "disapprove", "approve", "export_rebate_total"]
 
     @admin.action(description="Disapprove the students")
     def disapprove(self, request, queryset):
@@ -496,6 +498,21 @@ def export_as_csv(self, request, queryset):
         response["Content-Disposition"] = 'attachment; filename="Rebate.csv"'
         return response
 
+    def export_rebate_total(modeladmin, request, queryset):
+        total_days = 0
+        for obj in queryset:
+            total_days += (obj.end_date - obj.start_date).days + 1
+
+        # Create the HttpResponse object with the appropriate CSV header.
+        response = HttpResponse(content_type="text/csv")
+        response["Content-Disposition"] = 'attachment; filename="Rebate.csv"'
+
+        writer = csv.writer(response)
+        writer.writerow(["Total Days"])
+        writer.writerow([total_days])
+
+        return response
+
     export_as_csv.short_description = "Export Rebate details to CSV"