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 "", 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)