From 8b90380866b2ce4ed55bbc9a181463bcd1282ccd Mon Sep 17 00:00:00 2001
From: Yohann D'ANELLO <yohann.danello@gmail.com>
Date: Thu, 23 Apr 2020 21:13:29 +0200
Subject: [PATCH] Add script to export the WEI registrations as CSV format
---
README.md | 12 +++
.../commands/export_wei_registrations.py | 87 +++++++++++++++++++
2 files changed, 99 insertions(+)
create mode 100644 management/commands/export_wei_registrations.py
diff --git a/README.md b/README.md
index 0a7b177..7548965 100644
--- a/README.md
+++ b/README.md
@@ -41,6 +41,18 @@
options). Si non renseigné, il s'agit du dernier WEI.
Par défaut, si `--type` est non renseigné, la liste des adhérents BDE est renvoyée.
+
+- `extract_wei_registrations [--year|-y YEAR] [--bus|-b BUS] [--team|-t TEAM] [--sep SEP]` :
+
+ Récupère la liste des inscriptions au WEI et l'exporte au format CSV. Arguments possibles, optionnels :
+
+ * `--year YEAR` : sélectionne l'année du WEI. Par défaut, il s'agit du dernier WEI ayant eu lieu.
+ * `--bus BUS` : filtre par bus, en récupérant uniquement les inscriptions sur un bus. Par défaut, on affiche
+ tous les bus.
+ * `--team TEAM` : filtre par équipe, en récupérant uniquement les inscriptions sur une équipe. Par défaut, on
+ affiche toutes les équipes. Entrer `"none"` filtre les inscriptions sans équipe (chefs de bus, ...)
+ * `--sep` : définit le caractère de séparation des colonnes du fichier CSV. Par défaut, il s'agit du caractère `|`.
+ Merci de ne pas rentrer plus d'un caractère.
## Shell
diff --git a/management/commands/export_wei_registrations.py b/management/commands/export_wei_registrations.py
new file mode 100644
index 0000000..fb0398e
--- /dev/null
+++ b/management/commands/export_wei_registrations.py
@@ -0,0 +1,87 @@
+# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from django.core.management import BaseCommand, CommandError
+from django.db.models import Q
+from django.db.models.functions import Lower
+
+from wei.models import WEIClub, Bus, BusTeam, WEIMembership
+
+
+class Command(BaseCommand):
+ help = "Export WEI registrations."
+
+ def add_arguments(self, parser):
+ parser.add_argument('--bus', '-b', choices=[bus.name for bus in Bus.objects.all()], type=str, default=None,
+ help='Filter by bus')
+ parser.add_argument('--team', '-t', choices=[team.name for team in BusTeam.objects.all()], type=str,
+ default=None, help='Filter by team. Type "none" if you want to select the members '
+ + 'that are not in a team.')
+ parser.add_argument('--year', '-y', type=int, default=None,
+ help='Select the year of the concerned WEI. Default: last year')
+ parser.add_argument('--sep', type=str, default='|',
+ help='Select the CSV separator.')
+
+ def handle(self, *args, **options):
+ year = options["year"]
+ if year:
+ try:
+ wei = WEIClub.objects.get(year=year)
+ except WEIClub.DoesNotExist:
+ raise CommandError("The WEI of year {:d} does not exist.".format(year,))
+ else:
+ wei = WEIClub.objects.order_by('-year').first()
+
+ bus = options["bus"]
+ if bus:
+ try:
+ bus = Bus.objects.filter(wei=wei).get(name=bus)
+ except Bus.DoesNotExist:
+ raise CommandError("The bus {} does not exist or does not belong to the WEI {}.".format(bus, wei.name,))
+
+ team = options["team"]
+ if team:
+ if team.lower() == "none":
+ team = 0
+ else:
+ try:
+ team = BusTeam.objects.filter(Q(bus=bus) | Q(wei=wei)).get(name=team)
+ bus = team.bus
+ except BusTeam.DoesNotExist:
+ raise CommandError("The bus {} does not exist or does not belong to the bus {} neither the wei {}."
+ .format(team, bus.name if bus else "<None>", wei.name,))
+
+ qs = WEIMembership.objects
+ qs = qs.filter(club=wei).order_by(
+ Lower('bus__name'),
+ Lower('team__name'),
+ 'roles',
+ Lower('user__last_name'),
+ Lower('user__first_name'),
+ ).distinct()
+
+ if bus:
+ qs = qs.filter(bus=bus)
+
+ if team is not None:
+ qs = qs.filter(team=team if team else None)
+
+ sep = options["sep"]
+
+ self.stdout.write("Nom|Prénom|Date de naissance|Genre|Département|Année|Section|Bus|Équipe|Rôles"
+ .replace(sep, sep))
+
+ for membership in qs.all():
+ user = membership.user
+ registration = membership.registration
+ bus = membership.bus
+ team = membership.team
+ s = user.last_name
+ s += sep + user.first_name
+ s += sep + str(registration.birth_date)
+ s += sep + registration.get_gender_display()
+ s += sep + str(user.profile.ens_year) + "A"
+ s += sep + bus.name
+ s += sep + (team.name if team else "--")
+ s += sep + ", ".join(role.name for role in membership.roles.all())
+ self.stdout.write(s)
--
GitLab