Download sellings as csv

club/templates/club/club_sellings.jinja

@@ -7,6 +7,7 @@
     {% csrf_token %}
     {{ form }}
     <p><input type="submit" value="{% trans %}Show{% endtrans %}" /></p>
+    <p><input type="submit" value="{% trans %}Download as cvs{% endtrans %}" formaction="{{ url('club:sellings_csv', club_id=object.id) }}"/></p>
 </form>
 <p>
 {% trans %}Quantity: {% endtrans %}{{ total_quantity }} {% trans %}units{% endtrans %}<br/>

club/urls.py

@@ -10,6 +10,7 @@ urlpatterns = [
     url(r'^(?P<club_id>[0-9]+)/members$', ClubMembersView.as_view(), name='club_members'),
     url(r'^(?P<club_id>[0-9]+)/elderlies$', ClubOldMembersView.as_view(), name='club_old_members'),
     url(r'^(?P<club_id>[0-9]+)/sellings$', ClubSellingView.as_view(), name='club_sellings'),
+    url(r'^(?P<club_id>[0-9]+)/sellings/csv$', ClubSellingCSVView.as_view(), name='sellings_csv'),
     url(r'^(?P<club_id>[0-9]+)/prop$', ClubEditPropView.as_view(), name='club_prop'),
     url(r'^(?P<club_id>[0-9]+)/tools$', ClubToolsView.as_view(), name='tools'),
     url(r'^membership/(?P<membership_id>[0-9]+)/set_old$', MembershipSetOldView.as_view(), name='membership_set_old'),

club/views.py

@@ -4,10 +4,11 @@ from django.views.generic import ListView, DetailView
 from django.views.generic.edit import UpdateView, CreateView
 from django.forms import CheckboxSelectMultiple
 from django.core.exceptions import ValidationError
-from django.http import HttpResponseRedirect
+from django.http import HttpResponseRedirect, HttpResponse
 from django.core.urlresolvers import reverse
 from django.utils import timezone
 from django.utils.translation import ugettext_lazy as _
+from django.utils.translation import ugettext as _t
 from django.conf import settings
 from ajax_select.fields import AutoCompleteSelectField
 
@@ -203,6 +204,36 @@ class ClubSellingView(ClubTabsMixin, CanEditMixin, DetailView):
         kwargs['form'] = form
         return kwargs
 
+class ClubSellingCSVView(ClubSellingView):
+    """
+    Generate sellings in csv for a given period
+    """
+
+    def get(self, request, *args, **kwargs):
+        import csv
+        response = HttpResponse(content_type='text/csv')
+        self.object = self.get_object()
+        name = _("Sellings") + "_" + self.object.name + ".csv"
+        response['Content-Disposition'] = 'filename=' + name
+        kwargs = self.get_context_data(**kwargs)
+        writer = csv.writer(response, delimiter=";", lineterminator='\n', quoting=csv.QUOTE_ALL)
+
+        writer.writerow([_t('Date'),_t('Counter'),_t('Barman'),_t('Customer'),_t('Label'),
+                         _t('Quantity'), _t('Total'),_t('Payment method')])
+        for o in kwargs['result']:
+            row = [o.date, o.counter]
+            if o.seller:
+                row.append(o.seller.get_display_name())
+            else: row.append('')
+            if o.customer:
+                row.append(o.customer.user.get_display_name())
+            else: row.append('')
+            row = row +[o.label, o.quantity, o.quantity * o.unit_price,
+                       o.get_payment_method_display()]
+            writer.writerow(row)
+
+        return response
+
 class ClubEditView(ClubTabsMixin, CanEditMixin, UpdateView):
     """
     Edit a Club's main informations (for the club's members)