Changeset f23bc9a for asadb/groups

Timestamp:

Mar 31, 2012, 5:08:15 PM (14 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master, space-access, stable, stage

Children:

1b10de0

Parents:

6f85f2a

git-author:

Alex Dehnert <adehnert@…> (03/31/12 17:08:15)

git-committer:

Alex Dehnert <adehnert@…> (03/31/12 17:08:15)

Message:

Support CSV output

File:

• asadb/groups/views.py (modified) (5 diffs)

asadb/groups/views.py

@@ -2 +2 @@
 
 import collections
+import csv
 import datetime
 
@@ -14 +15 @@
 from django.template import Context, Template
 from django.template.loader import get_template
-from django.http import Http404, HttpResponseRedirect
+from django.http import HttpResponse, Http404, HttpResponseRedirect
 from django.core.urlresolvers import reverse
 from django.core.mail import EmailMessage, mail_admins
@@ -787 +788 @@
     )
 
+    _format_choices = [
+        ('html/inline',     "Web (HTML)", ),
+        ('csv/inline',      "Spreadsheet (CSV) --- in browser", ),
+        ('csv/download',    "Spreadsheet (CSV) --- download", ),
+    ]
+    output_format = forms.fields.ChoiceField(choices=_format_choices, widget=forms.RadioSelect, initial='html')
+
     class Meta:
         fieldsets = [
@@ -793 +801 @@
                 'fields': ['name', 'abbreviation', 'activity_category', 'group_class', 'group_status', 'group_funding', ],
             }),
-            ('display', {
+            ('fields', {
                 'legend': 'Data to display',
                 'fields': ['basic_fields', ],
+            }),
+            ('final', {
+                'legend': 'Final options',
+                'fields': ['output_format', ],
             }),
         ]
@@ -817 +829 @@
     if run_report:
         basic_fields = form.cleaned_data['basic_fields']
+        output_format, output_disposition = form.cleaned_data['output_format'].split('/')
         col_labels = [form.basic_fields_labels[field] for field in basic_fields]
         for group in groups_filterset.qs:
             group_data = [getattr(group, field) for field in basic_fields]
             report_groups.append(group_data)
-    else:
-        pass
+
+        if output_format == 'csv':
+            if output_disposition == 'download':
+                mimetype = 'text/csv'
+            else:
+                # Firefox, at least, downloads text/csv regardless
+                mimetype = 'text/plain'
+            response = HttpResponse(mimetype=mimetype)
+            if output_disposition == 'download':
+                response['Content-Disposition'] = 'attachment; filename=asa-db-report.csv'
+            writer = csv.writer(response)
+            writer.writerow(col_labels)
+            for row in report_groups: writer.writerow(row)
+            return response
 
     context = {