Allow generalization of CSV export

1 files changed, 12 insertions, 8 deletions

diff --git a/chimere/admin.py b/chimere/admin.py
index c36fd26..d492a4c 100644
--- a/chimere/admin.py
+++ b/chimere/admin.py
@@ -84,14 +84,18 @@ def export_to_shapefile(modeladmin, request, queryset):
     return response
 export_to_shapefile.short_description = _(u"Export to Shapefile")
 
-def export_to_csv(modeladmin, request, queryset):
-    u"""
-    Export data to CSV
-    """
-    filename, result = CSVManager.export(queryset)
-    response = HttpResponse(result, mimetype='text/csv')
-    response['Content-Disposition'] = 'attachment; filename=%s' % filename
-    return response
+def _export_to_csv(cols=[]):
+    def func(modeladmin, request, queryset):
+        u"""
+        Export data to CSV
+        """
+        filename, result = CSVManager.export(queryset, cols=cols)
+        response = HttpResponse(result, mimetype='text/csv')
+        response['Content-Disposition'] = 'attachment; filename=%s' % filename
+        return response
+    return func
+
+export_to_csv = _export_to_csv()
 export_to_csv.short_description = _(u"Export to CSV")
 
 def managed_modified(modeladmin, request, queryset):