Login

Admin action for a generic "CSV Export"

Author:
javinievas
Posted:
February 22, 2011
Language:
Python
Version:
1.2
Score:
6 (after 6 ratings)

Based on #2020

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=[...])]
    • Added UTF-8 encoding support
    • Bugs fixed
 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
import csv
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/1697/
        """
        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([unicode(getattr(obj, field)).encode("utf-8","replace") for field in field_names])
        return response
    export_as_csv.short_description = description
    return export_as_csv

More like this

  1. Template tag - list punctuation for a list of items by shapiromatron 10 months, 1 week ago
  2. JSONRequestMiddleware adds a .json() method to your HttpRequests by cdcarter 10 months, 2 weeks ago
  3. Serializer factory with Django Rest Framework by julio 1 year, 5 months ago
  4. Image compression before saving the new model / work with JPG, PNG by Schleidens 1 year, 6 months ago
  5. Help text hyperlinks by sa2812 1 year, 6 months ago

Comments

macmind (on August 18, 2011):

This might help for folks that need this ordered by the field list given in the admin module: 

writer = csv.DictWriter(response,fields)
writer.writeheader()

for obj in queryset:
    writer.writerow(dict(zip(fields,[unicode(getattr(obj, field)).encode("utf-8","replace") for field in fields])))

#

allaby (on November 14, 2011):

I need ordering preserved as mentioned by macmind, but this I'm getting the error "dictionary update sequence element #0 has length 1; 2 is required".

Does anyone have any thoughts on how I can fix this?

#

tatlar (on January 29, 2014):

macmind's code works, but you must use the code macmind posted exactly. So, replace:

writer = csv.writer(response)

if header:

writer.writerow(list(field_names))

with:

writer = csv.DictWriter(response,fields)

writer.writeheader()

and also notice that the

for obj in queryset:

loop uses fields not field_names. These are critical modifications.

#

tatlar (on February 7, 2014):

Note that the writer.writeheader() method is only available in Python 2.7+. Python 2.6 and lower do not support this and you will get an AttributeError exception thrown.

#

jaygauten (on February 10, 2015):

if you experienced an error regarding mimetype just change this part:

response = HttpResponse(mimetype='text/csv')

to:

response = HttpResponse(content_type='text/csv')

for as mimetype is deprecated in django1.7

#

Please login first before commenting.