Add export actions to payment_requests and participations

Author: MoeeinAali

Diff for: backend/participant/admin.py

@@ -1,5 +1,7 @@
 from django.contrib import admin
 from core.admin import ExportCSVMixin
+import csv
+from django.http import HttpResponse
 
 from participant.models import *
 
@@ -16,18 +18,51 @@ def get_national_code(self, obj):
 
 
 @admin.register(Participation)
-class ParticipationAdmin(admin.ModelAdmin, ExportCSVMixin):
+class ParticipationAdmin(admin.ModelAdmin):
     list_display = ('participant', 'participant_email', 'plan')
     list_filter = ('plan', 'plan__event__order',)
     autocomplete_fields = ('participant', 'info',)
     search_fields = ('participant__user__email', 'participant__info__national_code',)
-    actions = ["export_as_csv"]
+    actions = ["export_data"]
 
     def participant_email(self, obj):
         return obj.participant.user.email if obj.participant and obj.participant.user else ""
 
     participant_email.short_description = "Email"
 
+    def export_data(self, request, queryset):
+        meta = self.model._meta
+        response = HttpResponse(content_type='text/csv')
+        response['Content-Disposition'] = 'attachment; filename=participation_data.csv'
+        writer = csv.writer(response)
+
+        headers = [
+            'Participation Id',
+            'Email',
+            'First Name',
+            'Last Name',
+            'National Code',
+            'Phone Number',
+            'Plan'
+        ]
+        writer.writerow(headers)
+
+        for obj in queryset:
+            row = [
+                obj.id if obj.id else '',
+                obj.participant.user.email if obj.participant and obj.participant.user else '',
+                obj.participant.info.first_name if obj.participant and obj.participant.info else '',
+                obj.participant.info.last_name if obj.participant and obj.participant.info else '',
+                obj.participant.info.national_code if obj.participant and obj.participant.info else '',
+                obj.participant.info.phone_number if obj.participant and obj.participant.info else '',
+                str(obj.plan) if obj.plan else ''
+            ]
+            writer.writerow(row)
+
+        return response
+
+    export_data.short_description = "Export Registration Data As CSV"
+
 
 @admin.register(ParticipantInfo)
 class ParticipantInfoAdmin(admin.ModelAdmin):

Diff for: backend/payment/admin.py

@@ -5,6 +5,7 @@
 from django.shortcuts import redirect, render
 from django.urls import path
 from core.admin import ExportCSVMixin
+from django.http import HttpResponse
 
 from payment.models import PaymentDiscount, PaymentRequest
 from core.models import Event
@@ -53,7 +54,7 @@ def get_urls(self):
 class PaymentRequestAdmin(admin.ModelAdmin, ExportCSVMixin):
     list_display = ('participant', 'timestamp', 'paid', 'discount_code', 'order_id', 'base_price', 'paid_price')
     list_filter = ('paid', 'discount__code')
-    actions = ["export_as_csv"]
+    actions = ["export_as_csv", "calculate_discount_usage"]
 
     def base_price(self, obj):
         return obj.get_price()[0]
@@ -64,4 +65,41 @@ def paid_price(self, obj):
     def discount_code(self, obj):
         return obj.discount.code if obj.discount is not None else None
 
+    def calculate_discount_usage(self, request, queryset):
+        response = HttpResponse(content_type='text/csv')
+        response['Content-Disposition'] = 'attachment; filename=discount_usage.csv'
+        writer = csv.writer(response)
+
+        headers = [
+            'Discount Code',
+            'Event',
+            'Total Usage',
+            'Successful Payments'
+        ]
+        writer.writerow(headers)
+
+        discount_codes = PaymentDiscount.objects.filter(
+            paymentrequest__in=queryset
+        ).distinct()
+
+        for discount in discount_codes:
+            total_usage = queryset.filter(discount=discount).count()
+            
+            successful_payments = queryset.filter(
+                discount=discount,
+                paid=True
+            ).count()
+
+            row = [
+                discount.code,
+                str(discount.event) if discount.event else 'No Event',
+                total_usage,
+                successful_payments
+            ]
+            writer.writerow(row)
+
+        return response
+
+    calculate_discount_usage.short_description = "Calculate Discount Code Usage"
+
 admin.site.register(PaymentRequest, PaymentRequestAdmin)