From 0100604798156b8c23248c5d39c6ff69d6c84dfd Mon Sep 17 00:00:00 2001
From: Raphael Michel <mail@raphaelmichel.de>
Date: Tue, 7 Aug 2018 10:38:26 +0200
Subject: [PATCH] Allow to download waiting list

---
 .../pretixcontrol/waitinglist/index.html      |  7 ++-
 src/pretix/control/views/waitinglist.py       | 56 ++++++++++++++++++-
 2 files changed, 58 insertions(+), 5 deletions(-)

diff --git a/src/pretix/control/templates/pretixcontrol/waitinglist/index.html b/src/pretix/control/templates/pretixcontrol/waitinglist/index.html
index a8deb2f8ec..01800aa796 100644
--- a/src/pretix/control/templates/pretixcontrol/waitinglist/index.html
+++ b/src/pretix/control/templates/pretixcontrol/waitinglist/index.html
@@ -116,6 +116,9 @@
             </select>
         {% endif %}
         <button class="btn btn-primary" type="submit">{% trans "Filter" %}</button>
+        <a href="?{% url_replace request "download" "yes" %}"
+                class="btn btn-default"><i class="fa fa-download"></i>
+            {% trans "Download list" %}</a>
     </form>
     </p>
     <form method="post" action="?next={{ request.get_full_path|urlencode }}">
@@ -179,15 +182,13 @@
                                 </button>
                             {% endif %}
                         </td>
-                        <td>
+                        <td class="text-right">
                             {% if not e.voucher %}
                                 <a href="{% url "control:event.orders.waitinglist.delete" organizer=request.event.organizer.slug event=request.event.slug entry=e.id %}?next={{ request.get_full_path|urlencode }}" class="btn btn-danger btn-sm"><i class="fa fa-trash"></i></a>
                             {% else %}
                                 <span class="btn btn-danger btn-sm disabled"><i class="fa fa-trash"></i></span>
                             {% endif %}
                         </td>
-                        <td class="text-right">
-                        </td>
                     </tr>
                 {% endfor %}
                 </tbody>
diff --git a/src/pretix/control/views/waitinglist.py b/src/pretix/control/views/waitinglist.py
index f4cb879bec..07a2bedef2 100644
--- a/src/pretix/control/views/waitinglist.py
+++ b/src/pretix/control/views/waitinglist.py
@@ -1,13 +1,16 @@
+import csv
+import io
+
 from django.contrib import messages
 from django.db import transaction
 from django.db.models import F, Q, Sum
 from django.db.models.functions import Coalesce
-from django.http import Http404, HttpResponseRedirect
+from django.http import Http404, HttpResponse, HttpResponseRedirect
 from django.shortcuts import redirect
 from django.urls import reverse
 from django.utils.http import is_safe_url
 from django.utils.timezone import now
-from django.utils.translation import ugettext_lazy as _
+from django.utils.translation import pgettext, ugettext_lazy as _
 from django.views import View
 from django.views.generic import ListView
 from django.views.generic.edit import DeleteView
@@ -158,6 +161,55 @@ class WaitingListView(EventPermissionRequiredMixin, PaginationMixin, ListView):
         )
         return qs['s']
 
+    def get(self, request, *args, **kwargs):
+        if request.GET.get("download", "") == "yes":
+            return self._download_csv()
+        return super().get(request, *args, **kwargs)
+
+    def _download_csv(self):
+        output = io.StringIO()
+        writer = csv.writer(output, quoting=csv.QUOTE_NONNUMERIC, delimiter=",")
+
+        headers = [
+            _('E-mail address'), _('Product'), _('On list since'), _('Status'), _('Voucher code'),
+            _('Language')
+        ]
+        if self.request.event.has_subevents:
+            headers.append(pgettext('subevent', 'Date'))
+        writer.writerow(headers)
+
+        for w in self.get_queryset():
+            if w.item:
+                if w.variation:
+                    prod = '%s – %s' % (str(w.item), str(w.variation))
+                else:
+                    prod = '%s' % str(w.item)
+            if w.voucher:
+                if w.voucher.redeemed >= w.voucher.max_usages:
+                    status = _('Voucher redeemed')
+                elif not w.voucher.is_active():
+                    status = _('Voucher expired')
+                else:
+                    status = _('Voucher assigned')
+            else:
+                status = _('Waiting')
+
+            row = [
+                w.email,
+                prod,
+                w.created.isoformat(),
+                status,
+                w.voucher.code if w.voucher else '',
+                w.locale,
+            ]
+            if self.request.event.has_subevents:
+                row.append(str(w.subevent))
+            writer.writerow(row)
+
+        r = HttpResponse(output.getvalue().encode("utf-8"), content_type='text/csv')
+        r['Content-Disposition'] = 'attachment; filename="waitinglist.csv"'
+        return r
+
 
 class EntryDelete(EventPermissionRequiredMixin, DeleteView):
     model = WaitingListEntry