Export CSV for proposal Model in the Admin Panel (#787)

cricketkid · web-flow · commit b3a113ffbcd3 · 2023-10-09T12:41:51.000+05:30
* Export CSV for proposal

* removed spaces

* removed spaces
diff --git a/.gitignore b/.gitignore
@@ -27,6 +27,7 @@ var/
 *.egg-info/
 .installed.cfg
 *.egg
+.idea/*
 
 # PyInstaller
 #  Usually these files are written by a python script from a template
diff --git a/junction/proposals/admin.py b/junction/proposals/admin.py
@@ -10,10 +10,32 @@
 from junction.conferences import service
 from junction.conferences.models import ConferenceProposalReviewer
 from junction.proposals.models import ProposalSection
+from junction.base.constants import ProposalStatus, ProposalReviewStatus
 
+from django.http import HttpResponse
 from . import models
 
 
+class ExportMixin:
+    def export_as_csv(self, request, queryset):
+        import csv
+
+        response = HttpResponse(content_type='text/csv')
+        response['Content-Disposition'] = 'attachment; filename="exported_data.csv"'
+
+        writer = csv.writer(response)
+        writer.writerow(['Proposal Info', 'Author Info', 'Author Email', 'Conference', 'Status', 'Review Status'])
+
+        for item in queryset:
+            writer.writerow([self.proposal_info(item), self.author_info(item), self.author_email(item),
+                             item.conference.name, dict(ProposalStatus.CHOICES).get(item.status),
+                             dict(ProposalReviewStatus.CHOICES).get(item.review_status)])
+
+        return response
+
+    export_as_csv.short_description = "Export To CSV"
+
+
 @admin.register(models.ProposalSection)
 class ProposalSectionAdmin(AuditAdmin):
     list_display = ("name", "active") + AuditAdmin.list_display
@@ -32,26 +54,27 @@ def get_queryset(self, request):
         return qs.filter(
             conference_reviewer__conference__in=[m.conference for m in moderators]
         )
-    
+
     def get_form(self, request, obj=None, **kwargs):
         form = super(ProposalSectionReviewerAdmin, self).get_form(request, obj, **kwargs)
-        form.base_fields['conference_reviewer'].queryset = ConferenceProposalReviewer.objects.all().order_by('-created_at')
+        form.base_fields['conference_reviewer'].queryset = ConferenceProposalReviewer.objects.all().order_by(
+            '-created_at')
         form.base_fields['proposal_section'].queryset = ProposalSection.objects.all().order_by('-created_at')
         return form
 
 
 @admin.register(models.ProposalType)
 class ProposalTypeAdmin(AuditAdmin):
     list_display = (
-        "name",
-        "active",
-        "start_date",
-        "end_date",
+       "name",
+       "active",
+       "start_date",
+       "end_date",
     ) + AuditAdmin.list_display
 
 
 @admin.register(models.Proposal)
-class ProposalAdmin(TimeAuditAdmin, SimpleHistoryAdmin):
+class ProposalAdmin(TimeAuditAdmin, SimpleHistoryAdmin, ExportMixin):
     list_display = (
         "proposal_info",
         "author_info",
@@ -72,6 +95,8 @@ class ProposalAdmin(TimeAuditAdmin, SimpleHistoryAdmin):
         TextField: {"widget": AdminPagedownWidget},
     }
 
+    actions = ["export_as_csv"]
+
     def proposal_info(self, obj):
         return "%s (%s)" % (obj.title, obj.proposal_type)
 
