a95825ede8
- Things are suggested in python3 porting guide. https://docs.djangoproject.com/en/1.8/topics/python3/ 1. adding ```from django.utils.encoding import python_2_unicode_compatible``` 2. ``` __str__``` instead of ```__unicode__``` https://docs.djangoproject.com/en/1.8/topics/python3/#str-and-unicode-methods 3. Adding ```from __future__ import unicode_literals``` at the top of your Python modules https://docs.djangoproject.com/en/1.8/topics/python3/#unicode-literals 4. Removing the `u` prefix before unicode strings; https://docs.djangoproject.com/en/1.8/topics/python3/#unicode-literals - also closed #66 Signed-off-by: Hiroshi Miura <miurahr@linux.com>
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
from __future__ import unicode_literals
|
|
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
|
|
if fields:
|
|
fieldset = set(fields)
|
|
field_names = fieldset
|
|
elif exclude:
|
|
excludeset = set(exclude)
|
|
field_names = field_names - excludeset
|
|
response = HttpResponse(content_type="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
|