Login

Admin action for a generic "CSV Export" (fix for unicode)

Author:
__alexander__
Posted:
December 1, 2013
Language:
Python
Version:
1.5
Tags:
admin export csv action
Score:
0 (after 0 ratings)

Based on #2369

Save the snippet as actions.py within your django app, and then add an action on any model you want in it's ModelAdmin definition.

Example usage:

from actions import export_as_csv_action

class YourModelAdmin(admin.ModelAdmin): list_display = (...) list_filter = [...] actions = [export_as_csv_action("CSV Export", fields=[...])]

  • Unicode fix (requires unidecode)
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import csv
from unidecode import unidecode

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):
        """
        Generic csv export admin action.
        based on http://djangosnippets.org/snippets/2369/
        """
        opts = modeladmin.model._meta
        field_names = set([field.name for field in opts.fields])
        if fields:
            fieldset = set(fields)
            field_names = field_names & fieldset
        elif exclude:
            excludeset = set(exclude)
            field_names = field_names - excludeset

        response = HttpResponse(mimetype='text/csv')
        response['Content-Disposition'] = 'attachment; filename=%s.csv' % unicode(opts).replace('.', '_')

        writer = csv.writer(response)
        if header:
            writer.writerow(list(field_names))
        for obj in queryset:
            writer.writerow([unidecode(unicode(getattr(obj, field))) for field in field_names])
        return response
    export_as_csv.short_description = description
    return export_as_csv

More like this

  1. Admin action for a generic "CSV Export" by javinievas 4 years ago
  2. Admin action for a "CSV Export" with ManyToManyField by CarlosRodriguez 8 months, 1 week ago
  3. Generic admin action export selected rows to excel by jordic 3 years, 5 months ago
  4. Django Admin CSV Export Mixin by Ceran 1 year, 10 months ago
  5. Generic CSV Export by zbyte64 6 years, 8 months ago

Comments

Please login first before commenting.