-
-
Save michaelchiche/933ff4bb238b8f30e91f371a7b42efd1 to your computer and use it in GitHub Desktop.
"Export to CSV" action for django admin
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import unicodecsv | |
| from django.http import HttpResponse | |
| def export_as_csv_action(description="Export selected objects as CSV file", | |
| fields=None, exclude=None, header=True): | |
| """ | |
| This function returns an export csv action | |
| 'fields' and 'exclude' work like in django ModelForm | |
| 'header' is whether or not to output the column names as the first row | |
| """ | |
| def export_as_csv(modeladmin, request, queryset): | |
| opts = modeladmin.model._meta | |
| if not fields: | |
| field_names = [field.name for field in opts.fields] | |
| else: | |
| field_names = fields | |
| response = HttpResponse(mimetype='text/csv') | |
| response['Content-Disposition'] = 'attachment; filename=%s.csv' % unicode(opts).replace('.', '_') | |
| writer = unicodecsv.writer(response, encoding='utf-8') | |
| if header: | |
| writer.writerow(field_names) | |
| for obj in queryset: | |
| row = [getattr(obj, field)() if callable(getattr(obj, field)) else getattr(obj, field) for field in field_names] | |
| writer.writerow(row) | |
| return response | |
| export_as_csv.short_description = description | |
| return export_as_csv |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from django.contrib import admin | |
| from repsf.map.models import Location | |
| from django.http import HttpResponse, HttpResponseForbidden | |
| from actions import export_as_csv_action | |
| class SpamAdmin(admin.ModelAdmin): | |
| list_display = ['wink','nudge'] | |
| actions = [export_as_csv_action("CSV Export", fields=['wink','nudge'])] | |
| admin.site.register(Location, LocationAdmin) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment