Web system can export poll results to csv file.

1 files changed, 12 insertions, 0 deletions

diff --git a/webapp/polls/views.py b/webapp/polls/views.py
index 3487845..10ed773 100644
--- a/webapp/polls/views.py
+++ b/webapp/polls/views.py
@@ -323,6 +323,18 @@ def opt_thumb(request, poll_id, img_name):
     )
 
 
+def csv_download(request, poll_id):
+
+    poll_result = Poll.get(poll_id).get_result()
+
+    # Download csv file
+    response = HttpResponse(
+        poll_result.to_csv(), content_type='text/csv;')
+    response['Content-Disposition'] = (
+        'attachment; filename="%s_result.csv"' % poll_id)
+    return response
+
+
 class UnploadPollResultFormView(TemplateView):
 
     template_name = "poll-result-form.html"