Merge pull request #1342 from glogiotatidis/issue-1204-csv-export

[Fix #1204] Add CSV Export action to metrics related adminmodels.

snippets/base/admin/actions.py

@@ -1,8 +1,30 @@
+import csv
+from datetime import datetime
+
 from django.db import transaction
+from django.http import HttpResponse
 
 
 @transaction.atomic
 def duplicate_snippets_action(modeladmin, request, queryset):
     for snippet in queryset:
         snippet.duplicate(request.user)
 duplicate_snippets_action.short_description = 'Duplicate selected snippets'  # noqa
+
+
+def export_as_csv(modeladmin, request, queryset):
+    """Adapted from https://books.agiliq.com/projects/django-admin-cookbook/en/latest/export.html"""
+    meta = modeladmin.model._meta
+    field_names = [field.name for field in meta.fields]
+    filename = f'{meta}-{datetime.today().strftime("%Y-%m-%d-%H-%M")}'
+
+    response = HttpResponse(content_type='text/csv')
+    response['Content-Disposition'] = 'attachment; filename={}.csv'.format(filename)
+    writer = csv.writer(response)
+
+    writer.writerow(field_names)
+    for obj in queryset:
+        writer.writerow([getattr(obj, field) for field in field_names])
+
+    return response
+export_as_csv.short_description = 'Export Selected to CSV'  # noqa

snippets/base/admin/adminmodels.py

@@ -1562,6 +1562,9 @@ class JobDailyPerformanceAdmin(admin.ModelAdmin):
             ),
         }),
     ]
+    actions = [
+        actions.export_as_csv,
+    ]
 
     def has_add_permission(self, request):
         return False
@@ -1585,6 +1588,9 @@ class DailyImpressionsAdmin(admin.ModelAdmin):
             ),
         }),
     ]
+    actions = [
+        actions.export_as_csv,
+    ]
 
     def has_add_permission(self, request):
         return False