From 1152a43af52287a406903d30fd6c3c20e369159e Mon Sep 17 00:00:00 2001
From: Yohann D'ANELLO <yohann.danello@gmail.com>
Date: Wed, 22 Apr 2020 17:53:42 +0200
Subject: [PATCH] Extract mailing list registrations

---
 .../commands/extract_ml_registrations.py      | 36 +++++++++++++++++++
 apps/wei/management/commands/wei_algorithm.py |  6 ++--
 2 files changed, 40 insertions(+), 2 deletions(-)
 create mode 100644 apps/wei/management/commands/extract_ml_registrations.py

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..cdc42f39
--- /dev/null
+++ b/apps/wei/management/commands/extract_ml_registrations.py
@@ -0,0 +1,36 @@
+# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from django.core.management import BaseCommand
+from django.db.models import Q
+
+from ...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=["events", "art", "sport"], default="events",
+                            help='Select the type of the mailing list (default events)')
+        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["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
index d639117e..1c910e4c 100644
--- a/apps/wei/management/commands/wei_algorithm.py
+++ b/apps/wei/management/commands/wei_algorithm.py
@@ -2,13 +2,15 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 
 from django.core.management import BaseCommand
-from django.utils.translation import gettext_lazy as _
 
 from ...forms import CurrentSurvey
 
 
 class Command(BaseCommand):
-    help = _("Attribute to each first year member a bus for the WEI")
+    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()