Merge pull request #19 from mastertech/improvement/admin-export

Adicionar ação no admin para exportar linhas selecionadas para csv

Author: ericbgauch

nisia_cadastro/admin.py

@@ -1,6 +1,7 @@
 from django.contrib import admin
 from django.utils.translation import gettext_lazy as _
 from .models import Registered
+from .export import ExportCsvMixin
 
 
 class HasWrittenStoryListFilter(admin.SimpleListFilter):
@@ -21,9 +22,10 @@ def queryset(self, request, queryset):
             return queryset.filter(story__exact='')
 
 
-class RegisteredAdmin(admin.ModelAdmin):
+class RegisteredAdmin(admin.ModelAdmin, ExportCsvMixin):
     list_display = ('name', 'email', 'role', 'has_written_story', 'registered_at')
     list_filter = ('role', HasWrittenStoryListFilter, 'registered_at')
+    actions = ["export_as_csv"]
 
 admin.site.empty_value_display = 'desconhecido'
 admin.site.register(Registered, RegisteredAdmin)

nisia_cadastro/export.py

@@ -0,0 +1,22 @@
+from django.http import HttpResponse
+from django.utils.translation import gettext_lazy as _
+import csv
+
+
+class ExportCsvMixin:
+    def export_as_csv(self, request, queryset):
+        meta = self.model._meta
+        field_names = [field.name for field in meta.fields]
+
+        response = HttpResponse(content_type='text/csv')
+        response['Content-Disposition'] = 'attachment; filename={}.csv'.format(meta)
+        writer = csv.writer(response)
+
+        writer.writerow(field_names)
+        for obj in queryset:
+            row = writer.writerow([getattr(obj, field) for field in field_names])
+
+        return response
+
+    export_as_csv.short_description = _('Exportar Selecionados')
+