From 110ef79951e0937b496ba4ee6227953af281bd2d Mon Sep 17 00:00:00 2001 From: Pierre-antoine Comby Date: Thu, 14 May 2020 15:16:10 +0200 Subject: [PATCH] move wei script to wei app --- apps/scripts | 2 +- .../commands/export_wei_registrations.py | 89 +++++++++++++++++++ .../commands/extract_ml_registrations.py | 52 +++++++++++ apps/wei/management/commands/wei_algorithm.py | 15 ++++ 4 files changed, 157 insertions(+), 1 deletion(-) create mode 100644 apps/wei/management/commands/export_wei_registrations.py create mode 100644 apps/wei/management/commands/extract_ml_registrations.py create mode 100644 apps/wei/management/commands/wei_algorithm.py diff --git a/apps/scripts b/apps/scripts index f0aa4269..64e8e88e 160000 --- a/apps/scripts +++ b/apps/scripts @@ -1 +1 @@ -Subproject commit f0aa426950b9b867bf99233795e260871be2cb99 +Subproject commit 64e8e88ed3ab505b2cfb0529817812f3b7990767 diff --git a/apps/wei/management/commands/export_wei_registrations.py b/apps/wei/management/commands/export_wei_registrations.py new file mode 100644 index 00000000..f76852c8 --- /dev/null +++ b/apps/wei/management/commands/export_wei_registrations.py @@ -0,0 +1,89 @@ +# 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'), + 'user__profile__promotion', + 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 + user.profile.get_department_display() + s += sep + str(user.profile.ens_year) + "A" + s += sep + user.profile.section_generated + s += sep + bus.name + s += sep + (team.name if team else "--") + s += sep + ", ".join(role.name for role in membership.roles.filter(~Q(name="Adhérent WEI")).all()) + self.stdout.write(s) diff --git a/apps/wei/management/commands/extract_ml_registrations.py b/apps/wei/management/commands/extract_ml_registrations.py new file mode 100644 index 00000000..b67bf10b --- /dev/null +++ b/apps/wei/management/commands/extract_ml_registrations.py @@ -0,0 +1,52 @@ +# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay +# SPDX-License-Identifier: GPL-3.0-or-later + +from datetime import date + +from django.core.management import BaseCommand +from django.db.models import Q +from member.models import Membership, Club +from wei.models import WEIClub + + +class Command(BaseCommand): + help = "Get mailing list registrations from the last wei. " \ + "Usage: manage.py extract_ml_registrations -t {events,art,sport}. " \ + "You can write this into a file with a pipe, then paste the document into your mail manager." + + def add_arguments(self, parser): + parser.add_argument('--type', '-t', choices=["members", "clubs", "events", "art", "sport"], default="members", + help='Select the type of the mailing list (default members)') + parser.add_argument('--year', '-y', type=int, default=None, + help='Select the year of the concerned WEI. Default: last year') + + def handle(self, *args, **options): + if options["type"] == "members": + for membership in Membership.objects.filter( + club__name="BDE", + date_start__lte=date.today(), + date_end__gte=date.today(), + ).all(): + self.stdout.write(membership.user.email) + return + + if options["type"] == "clubs": + for club in Club.objects.all(): + self.stdout.write(club.email) + return + + if options["year"] is None: + wei = WEIClub.objects.order_by('-year').first() + else: + wei = WEIClub.objects.filter(year=options["year"]) + if wei.exists(): + wei = wei.get() + else: + wei = WEIClub.objects.order_by('-year').first() + self.stderr.write(self.style.WARNING("Warning: there was no WEI in year " + str(options["year"]) + ". " + + "Assuming the last WEI (year " + str(wei.year) + ")")) + q = Q(ml_events_registration=True) if options["type"] == "events" else Q(ml_art_registration=True)\ + if options["type"] == "art" else Q(ml_sport_registration=True) + registrations = wei.users.filter(q) + for registration in registrations.all(): + self.stdout.write(registration.user.email) diff --git a/apps/wei/management/commands/wei_algorithm.py b/apps/wei/management/commands/wei_algorithm.py new file mode 100644 index 00000000..01640720 --- /dev/null +++ b/apps/wei/management/commands/wei_algorithm.py @@ -0,0 +1,15 @@ +# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay +# SPDX-License-Identifier: GPL-3.0-or-later + +from django.core.management import BaseCommand +from wei.forms import CurrentSurvey + + +class Command(BaseCommand): + help = "Attribute to each first year member a bus for the WEI" + + def handle(self, *args, **options): + """ + Run the WEI algorithm to attribute a bus to each first year member. + """ + CurrentSurvey.get_algorithm_class()().run_algorithm()