Export csv evenements

Author: CelestinHuet

orgues/templates/orgues/orgue_list.html

@@ -47,6 +47,12 @@ <h5>Quelques conseils pour éditer la photo principale d'une fiche d'orgue</h5>
                 {% endif %}
                 <hr>
 
+                {% if perms.orgues.view_user %}
+                <a href="{% url 'orgues:evenement-csv' %}" class="btn btn-warning rounded-pill mt-3 btn-block">
+                    Export des événements en csv</a>
+                {% endif %}
+                <hr>
+
                 {% if perms.orgues.add_orgue %}
                 <a href="{% url 'orgues:orgue-create' %}" class="btn btn-info rounded-pill mt-3 btn-block">
                     Créer un nouvel orgue</a>

orgues/urls.py

@@ -6,6 +6,7 @@
 urlpatterns = [
     path('orgues/', v.OrgueList.as_view(), name='orgue-list'),
     path('orgues/csv/', v.OrgueExport.as_view(), name='orgue-csv'),
+    path('evenements/csv/', v.EvenementExport.as_view(), name='evenement-csv'),
     path('recherche/', v.OrgueSearch.as_view(), name='orgue-search'),
     path('stats.json', v.Stats.as_view(), name='orgue-stats-js'),
     path('carte/', v.OrgueCarte.as_view(), name='orgue-carte'),

orgues/views.py

@@ -13,7 +13,8 @@
 from django.core import serializers
 from django.core.paginator import Paginator
 from django.db import connection, transaction
-from django.db.models import Q, Count, Case, When, IntegerField, Value
+from django.db.models import Q, Count, Case, When, IntegerField, Value, F, Aggregate, CharField
+from django.db.models.functions import Concat
 from django.forms import modelformset_factory
 from django.http import JsonResponse, Http404, HttpResponse
 from django.shortcuts import get_object_or_404, redirect, render
@@ -1684,6 +1685,63 @@ def get(self, request, *args, **kwargs):
         return response
 
 
+class GroupConcat(Aggregate):
+    function = 'GROUP_CONCAT'
+    template = "%(function)s(%(expressions)s, '%(separator)s')"
+
+    def __init__(self, expression, separator=', ', **extra):
+        super().__init__(
+            expression,
+            separator=separator,
+            output_field=CharField(),
+            **extra
+        )
+
+class EvenementExport(FabView):
+    permission_required = 'orgues.view_user'
+
+    def get(self, request, *args, **kwargs):
+        response = HttpResponse(content_type='text/csv')
+        response.write(u'\ufeff'.encode('utf8'))
+        response['Content-Disposition'] = 'attachment;filename=evenements_orgue_{}.csv'.format(
+            datetime.today().strftime("%Y-%m-%d"))
+        columns = [
+            "annee",
+            "annee_fin",
+            "nom_orgue",
+            "circa",
+            "type",
+            "nom_facteurs",
+            "nom_manufactures",
+        ]
+        writer = csv.DictWriter(response, delimiter=';', fieldnames=columns)
+
+        # header from verbose_names
+        row = {}
+        for column in columns:
+            try:
+                field = Evenement._meta.get_field(column)
+                row[column] = field.verbose_name
+            except:
+                # Pour les colonnes "virtuelles", on met un nom lisible par défaut
+                row[column] = column.replace('_', ' ').capitalize()
+        writer.writerow(row)
+        writer.writerows(
+            Evenement.objects.annotate(
+                nom_orgue = Concat(
+                    F('orgue__designation'),
+                    Value(' '),
+                    F('orgue__edifice'),
+                    Value(' '),
+                    F('orgue__commune')
+                ),
+                nom_facteurs = GroupConcat('facteurs__nom', separator=', '),
+                nom_manufactures = GroupConcat('manufactures__nom', separator=', '),
+            ).values(*columns).order_by("annee")
+        )
+        return response
+
+
 class Dashboard(FabView):
     """
     Dashboard d'avancement réservé aux admins