diff --git a/apps/activity/migrations/0005_alter_opener_options_alter_opener_opener.py b/apps/activity/migrations/0005_alter_opener_options_alter_opener_opener.py new file mode 100644 index 00000000..c09500e1 --- /dev/null +++ b/apps/activity/migrations/0005_alter_opener_options_alter_opener_opener.py @@ -0,0 +1,24 @@ +# Generated by Django 4.2.15 on 2024-08-28 08:00 + +from django.db import migrations, models +import django.db.models.deletion + + +class Migration(migrations.Migration): + + dependencies = [ + ('note', '0006_trust'), + ('activity', '0004_opener'), + ] + + operations = [ + migrations.AlterModelOptions( + name='opener', + options={'verbose_name': 'Opener', 'verbose_name_plural': 'Openers'}, + ), + migrations.AlterField( + model_name='opener', + name='opener', + field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='activity_responsible', to='note.note', verbose_name='Opener'), + ), + ] diff --git a/apps/food/migrations/0005_alter_food_polymorphic_ctype.py b/apps/food/migrations/0005_alter_food_polymorphic_ctype.py new file mode 100644 index 00000000..5473bffc --- /dev/null +++ b/apps/food/migrations/0005_alter_food_polymorphic_ctype.py @@ -0,0 +1,20 @@ +# Generated by Django 4.2.15 on 2024-08-28 08:00 + +from django.db import migrations, models +import django.db.models.deletion + + +class Migration(migrations.Migration): + + dependencies = [ + ('contenttypes', '0002_remove_content_type_name'), + ('food', '0004_auto_20240813_2358'), + ] + + operations = [ + migrations.AlterField( + model_name='food', + name='polymorphic_ctype', + field=models.ForeignKey(editable=False, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='polymorphic_%(app_label)s.%(class)s_set+', to='contenttypes.contenttype'), + ), + ] diff --git a/apps/note/migrations/0007_alter_note_polymorphic_ctype_and_more.py b/apps/note/migrations/0007_alter_note_polymorphic_ctype_and_more.py new file mode 100644 index 00000000..8d0e8a19 --- /dev/null +++ b/apps/note/migrations/0007_alter_note_polymorphic_ctype_and_more.py @@ -0,0 +1,25 @@ +# Generated by Django 4.2.15 on 2024-08-28 08:00 + +from django.db import migrations, models +import django.db.models.deletion + + +class Migration(migrations.Migration): + + dependencies = [ + ('contenttypes', '0002_remove_content_type_name'), + ('note', '0006_trust'), + ] + + operations = [ + migrations.AlterField( + model_name='note', + name='polymorphic_ctype', + field=models.ForeignKey(editable=False, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='polymorphic_%(app_label)s.%(class)s_set+', to='contenttypes.contenttype'), + ), + migrations.AlterField( + model_name='transaction', + name='polymorphic_ctype', + field=models.ForeignKey(editable=False, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='polymorphic_%(app_label)s.%(class)s_set+', to='contenttypes.contenttype'), + ), + ] diff --git a/apps/treasury/migrations/0009_alter_sogecredit_transactions.py b/apps/treasury/migrations/0009_alter_sogecredit_transactions.py new file mode 100644 index 00000000..e3a46576 --- /dev/null +++ b/apps/treasury/migrations/0009_alter_sogecredit_transactions.py @@ -0,0 +1,19 @@ +# Generated by Django 4.2.15 on 2024-08-28 08:00 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('note', '0007_alter_note_polymorphic_ctype_and_more'), + ('treasury', '0008_auto_20240322_0045'), + ] + + operations = [ + migrations.AlterField( + model_name='sogecredit', + name='transactions', + field=models.ManyToManyField(blank=True, related_name='+', to='note.membershiptransaction', verbose_name='membership transactions'), + ), + ] diff --git a/apps/wei/forms/registration.py b/apps/wei/forms/registration.py index c851248d..cf0579aa 100644 --- a/apps/wei/forms/registration.py +++ b/apps/wei/forms/registration.py @@ -81,6 +81,11 @@ class WEIChooseBusForm(forms.Form): class WEIMembershipForm(forms.ModelForm): + caution_check = forms.BooleanField( + required=False, + label=_("Caution check given"), + ) + roles = forms.ModelMultipleChoiceField( queryset=WEIRole.objects, label=_("WEI Roles"), @@ -149,6 +154,7 @@ class WEIMembership1AForm(WEIMembershipForm): """ Used to confirm registrations of first year members without choosing a bus now. """ + caution_check = None roles = None def clean(self): diff --git a/apps/wei/forms/surveys/__init__.py b/apps/wei/forms/surveys/__init__.py index 0e0c37e2..733f7c37 100644 --- a/apps/wei/forms/surveys/__init__.py +++ b/apps/wei/forms/surveys/__init__.py @@ -2,11 +2,11 @@ # SPDX-License-Identifier: GPL-3.0-or-later from .base import WEISurvey, WEISurveyInformation, WEISurveyAlgorithm -from .wei2023 import WEISurvey2023 +from .wei2024 import WEISurvey2024 __all__ = [ 'WEISurvey', 'WEISurveyInformation', 'WEISurveyAlgorithm', 'CurrentSurvey', ] -CurrentSurvey = WEISurvey2023 +CurrentSurvey = WEISurvey2024 diff --git a/apps/wei/forms/surveys/wei2024.py b/apps/wei/forms/surveys/wei2024.py new file mode 100644 index 00000000..ef3f37f9 --- /dev/null +++ b/apps/wei/forms/surveys/wei2024.py @@ -0,0 +1,366 @@ +# Copyright (C) 2018-2024 by BDE ENS Paris-Saclay +# SPDX-License-Identifier: GPL-3.0-or-later + +import json + +from functools import lru_cache + +from django import forms +from django.utils.safestring import mark_safe +from django.db import transaction +from django.db.models import Q + +from .base import WEISurvey, WEISurveyInformation, WEISurveyAlgorithm, WEIBusInformation +from ...models import WEIMembership + + +buses_descr = [ + [ + "Magi[Kar]p 🐙🎼đŸŽČ", "#ef5568", 1, + """Vous l'aurez compris au nom du bus, l'ambiance est aux jeux et Ă  la culture geek ! Ici, vous trouverez une ambiance + calme avec une bonne dose d'autodĂ©rision et de second degrĂ©. Que vous ayez besoin de beaucoup dormir pour tenir la soirĂ©e + du lendemain, ou que vous souhaitiez faire nuit blanche pour jouer toute la nuit, vous pouvez nous rejoindre. Votre voix + n'y survivra peut-ĂȘtre pas Ă  force de chanter. PS : les meilleurs cocktails du WEI sont chez nous, Ă  dĂ©guster, pas Ă  + siphonner !""", + ], + [ + "Va[car]me đŸŽ·đŸŽđŸ”Š", "#fd7a28", 3, + """Ici, c'est le bus du bruit. Si vous voulez rĂ©veiller les autres bus en musique, apprendre de merveilleuses mĂ©lodies au kazoo tout le week-end, ou simplement profiter d'une bonne ambiance musicale, le BDA et la F[ENS]foire sont lĂ  pour vous. Vous pourrez Ă©galement goĂ»ter au cĂ©lĂšbre cocktail de la fanfare, concoctĂ© pour l'occasion par les tout nouveaux "meilleurs artisans v*********** de France" ! Alors que vous soyez artiste dans l'Ăąme ou que vous souhaitiez juste faire le plus grand Vacarme, rejoignez-nous !""", + ], + [ + "[Kar]aĂŻbes đŸïžđŸŽâ€â˜ ïžđŸ„„", "#a5cfdd", 3, + """Ahoy, explorateurs du WEI ! Le bus Karaibes t’invite Ă  une traversĂ©e sous les tropiques, oĂč l’ambiance est toujours au beau fixe ! ☀đŸč Ici, c’est soleil, rhum, et bonne humeur assurĂ©e : une atmosphĂšre de vacances oĂč l’on se laisse porter par la chaleur humaine et la fĂȘte. Que tu sois un pirate en quĂȘte de sensations fortes ou un amateur de chill avec un cocktail Ă  la main, tu seras Ă  ta place dans notre bus. Les soirĂ©es seront marquĂ©es par des rythmes tropicaux qui te feront vibrer jusqu’à l’aube. PrĂȘt Ă  embarquer pour une aventure inoubliable avec les meilleurs matelots du WEI ? On t’attend sur le pont du Karaibes pour lever l’ancre ensemble !""", + ], + [ + "[Kar]di [Bus] đŸŽ™ïžđŸ’…", "#e46398", 2.5, + """Bienvenue Ă  bord du Kardi Bus, la seul, l’unique, l’inimitable pĂ©pite de ce weekend d’intĂ©gration ! InspirĂ© par les + icĂŽnes suprĂȘmes de la pop culture telles les Bratz, les Winx et autres Mean Girls, notre bus est un sanctuaire de style, + d’audace et de pur plaisir. A nos cotĂ©s attends toi Ă  siroter tes meilleurs Cosmo, sex on the Beach et autres cocktails + de maxi pĂ©tasse tout en papotant entre copains copines ! Si tu rejoins le Kardi Bus, tu entres dans un monde oĂč tu + pourras te dĂ©hancher sur du BeyoncĂ©, Britney, Aya et autres reines de la pop ! À trĂšs vite, les futures stars du Kardi + Bus !""", + ], + [ + "Sparta[bus] đŸș🐒🏉", "#ebdac2", 5, + """Dans notre bus, on vous donne un avant goĂ»t des plus grandes assos de l'ENS : les Kyottes et l'Aspique (clubs de rugby + fĂ©minin et masculin, mais pas que). Bien entendu, qui dit rugby dit les copaings, le pastaga et la Pena Bayona, mais vous + verrez par vous mĂȘme qu'on est ouvert⋅e Ă  toutes propositions quand il s'agit de faire la fĂȘte. Pour les casse-cous comme + pour les plus calmes, vous trouverez au bus Aspique-Kyottes les 2A+ qui vous feront kiffer votre WEI.""", + ], + [ + "Zanzo[Bus] đŸ€©đŸ‘œđŸ”", "#FFFF", 3, + """Dans un entre-trois bien senti entre zinzinerie, enfance et vieillerie, le Zanzo[BUS] est un concentrĂ© de fun mĂȘlĂ© Ă  + de la dinguerie Ă  gogo. N'hĂ©sitez plus et rejoignez-nous pour un WEI toujours plus dĂ©jantĂ© !""", + ], + [ + "Bran[Kar] đŸčđŸ„ł", "#6da1ac", 4, + """Si vous ne connaissez pas le Bran[Kar], c’est comme une grande famille qui fait un apĂ©ro, qui se bourre un peu la + gueule en discutant des heures autour d’une table remplie de bouffe et de super bons cocktails (la plupart des + barmen/barwomen du bus sont les barmans de Shakens), sauf qu’on est un bus du Wei (vous comprendrez bien le nom de notre + bus en voyant l’état de certain·e·s). Il nous arrive de faire quelques conneries, mais surtout de jouer au BiĂšre-pong en + musique !""", + ], + [ + "Techno [kar]ade đŸ”ŠđŸš©", "#8065a3", 3, + """Avis Ă  tous·tes les gauchos, amoureux·ses de la fĂȘte et des manifs : le Techno [kar]ade vous ouvre grand ses bras pour + finir en beautĂ© votre premiĂšre intĂ©. PrĂ©parez-vous Ă  vous abreuver de cocktails (savamment Ă©laborĂ©s) Ă  la vibration d’un + systĂšme son fabriquĂ© pour l’occasion. Des sets technos Ă  « Mon pĂšre Ă©tait tellement de gauche » en passant par « Female + Body », le car accueillant les meilleures DJs du plateau saura animer le trajet aussi bien que les soirĂ©es. Si alcool et + musique seront au rendez-vous, les maĂźtres mots sont sĂ©curitĂ© et inclusivitĂ©. Qui que vous soyez et quelle que soit votre + maniĂšre de vous amuser, notre objectif est que vous vous sentiez Ă  l’aise pour rencontrer au mieux les 1A, les 2A et les + (nombreux⋅ses) 3A+ qui auront rĂ©pondu Ă  l’appel. Bref, rejoignez-nous, on est super cools :)""" + ], + [ + "[Bus]ka-P đŸ„‡đŸ»đŸŽ€", "#7c4768", 4.5, + """Booska-p, c’est le « site N°1 du Rap français ». Le [Bus]ka-p ? Le bus N°1 sur l’ambiance au WEI. Les nuits vont ĂȘtre + courtes, les cocktails vont couler Ă  flots : tout sera rĂ©uni pour vivre un week-end dont tu te souviendras toute ta vie. + Au programme pas un seul temps mort et un maximum de rencontres pour bien commencer ta premiĂšre annĂ©e Ă  l’ENS. Et bien + entendu, le tout accompagnĂ© des meilleurs sons, de Jul Ă  Aya, en passant par ABBA et Sexion d’Assaut. Bref, si tu veux + vivre un WEI d’anthologie et faire la fĂȘte, de jour comme de nuit, nous t’accueillons avec plaisir !""", + ], +] + + + +def print_bus(i): + return f"""

{buses_descr[i][0]}


+ AlcoolomĂštre : {buses_descr[i][2]} / 5 đŸ»

{buses_descr[i][3]}
""" + + +def print_all_buses(): + liste = [print_bus(i) for i in range(len(buses_descr))] + return "



".join(liste) + + +def get_number_comment(i): + if i == 1: + return "MĂȘme pas en rĂȘve" + elif i == 2: + return "Pas envie" + elif i == 3: + return "Mouais..." + elif i == 4: + return "Pourquoi pas !" + elif i == 5: + return "Ce bus ou rien !!!" + else: + return "" + + +WORDS = { + "recap": + [ + """ChĂšr⋅e 1A, te voilĂ  arrivĂ©â‹…e au moment fatidique du choix de ton bus !


+ Ton bus est constitué des gens avec qui tu passeras la majorité de ton temps : que ce soit le voyage d'aller et de + retour et les différentes activité qu'ils pourront te proposer tout au long du WEI donc choisis le bien ! +

Tu trouveras ci-dessous la liste de tous les bus ainsi qu'une description dĂ©taillĂ©e de ces derniers. + Prends ton temps pour Ă©tudier chacun d'eux et quand tu te sens prĂȘt⋅e, appuie sur le bouton « J'ai pris connaissance + des bus » pour continuer +
(pas besoin d'apprendre par cƓur chaque bus, la description de chaque bus te sera rappeler avant de lui attribuer + une note !)



""" + print_all_buses(), + { + "1": "J'ai pris connaissance des diffĂ©rents bus et me sent fin prĂȘt Ă  choisir celui qui me convient le mieux !", + } + ] +} + +WORDS.update({ + f"bus{id}": [print_bus(id), {i: f"{get_number_comment(i)} ({i}/5)" for i in range(1, 5 + 1)}] for id in range(len(buses_descr)) +}) + + +class WEISurveyForm2024(forms.Form): + """ + Survey form for the year 2024. + Members score the different buses, from which we calculate the best associated bus. + """ + def set_registration(self, registration): + """ + Filter the bus selector with the buses of the current WEI. + """ + information = WEISurveyInformation2024(registration) + + question = information.questions[information.step] + self.fields[question] = forms.ChoiceField( + label=mark_safe(WORDS[question][0]), + widget=forms.RadioSelect(), + ) + answers = [(answer, WORDS[question][1][answer]) for answer in WORDS[question][1]] + self.fields[question].choices = answers + + +class WEIBusInformation2024(WEIBusInformation): + """ + For each question, the bus has ordered answers + """ + scores: dict + + def __init__(self, bus): + self.scores = {} + for question in WORDS: + self.scores[question] = [] + super().__init__(bus) + + +class WEISurveyInformation2024(WEISurveyInformation): + """ + We store the id of the selected bus. We store only the name, but is not used in the selection: + that's only for humans that try to read data. + """ + + step = 0 + questions = list(WORDS.keys()) + + def __init__(self, registration): + for question in WORDS: + setattr(self, str(question), None) + super().__init__(registration) + + +class WEISurvey2024(WEISurvey): + """ + Survey for the year 2024. + """ + + @classmethod + def get_year(cls): + return 2024 + + @classmethod + def get_survey_information_class(cls): + return WEISurveyInformation2024 + + def get_form_class(self): + return WEISurveyForm2024 + + def update_form(self, form): + """ + Filter the bus selector with the buses of the WEI. + """ + form.set_registration(self.registration) + + @transaction.atomic + def form_valid(self, form): + self.information.step += 1 + for question in WORDS: + if question in form.cleaned_data: + answer = form.cleaned_data[question] + setattr(self.information, question, answer) + self.save() + + @classmethod + def get_algorithm_class(cls): + return WEISurveyAlgorithm2024 + + def is_complete(self) -> bool: + """ + The survey is complete once the bus is chosen. + """ + for question in WORDS: + if not getattr(self.information, question): + return False + return True + + @lru_cache() + def score(self, bus): + if not self.is_complete(): + raise ValueError("Survey is not ended, can't calculate score") + + bus_info = self.get_algorithm_class().get_bus_information(bus) + # Score is the given score by the bus subtracted to the mid-score of the buses. + s = 0 + for question in WORDS: + s += bus_info.scores[question][str(getattr(self.information, question))] + return s + + @lru_cache() + def scores_per_bus(self): + return {bus: self.score(bus) for bus in self.get_algorithm_class().get_buses()} + + @lru_cache() + def ordered_buses(self): + values = list(self.scores_per_bus().items()) + values.sort(key=lambda item: -item[1]) + return values + + @classmethod + def clear_cache(cls): + return super().clear_cache() + + +class WEISurveyAlgorithm2024(WEISurveyAlgorithm): + """ + The algorithm class for the year 2024. + We use Gale-Shapley algorithm to attribute 1y students into buses. + """ + + @classmethod + def get_survey_class(cls): + return WEISurvey2024 + + @classmethod + def get_bus_information_class(cls): + return WEIBusInformation2024 + + def run_algorithm(self, display_tqdm=False): + """ + Gale-Shapley algorithm implementation. + We modify it to allow buses to have multiple "weddings". + """ + surveys = list(self.get_survey_class()(r) for r in self.get_registrations()) # All surveys + surveys = [s for s in surveys if s.is_complete()] # Don't consider invalid surveys + # Don't manage hardcoded people + surveys = [s for s in surveys if not hasattr(s.information, 'hardcoded') or not s.information.hardcoded] + + # Reset previous algorithm run + for survey in surveys: + survey.free() + survey.save() + + non_men = [s for s in surveys if s.registration.gender != 'male'] + men = [s for s in surveys if s.registration.gender == 'male'] + + quotas = {} + registrations = self.get_registrations() + non_men_total = registrations.filter(~Q(gender='male')).count() + for bus in self.get_buses(): + free_seats = bus.size - WEIMembership.objects.filter(bus=bus, registration__first_year=False).count() + # Remove hardcoded people + free_seats -= WEIMembership.objects.filter(bus=bus, registration__first_year=True, + registration__information_json__icontains="hardcoded").count() + quotas[bus] = 4 + int(non_men_total / registrations.count() * free_seats) + + tqdm_obj = None + if display_tqdm: + from tqdm import tqdm + tqdm_obj = tqdm(total=len(non_men), desc="Non-hommes") + + # Repartition for non men people first + self.make_repartition(non_men, quotas, tqdm_obj=tqdm_obj) + + quotas = {} + for bus in self.get_buses(): + free_seats = bus.size - WEIMembership.objects.filter(bus=bus, registration__first_year=False).count() + free_seats -= sum(1 for s in non_men if s.information.selected_bus_pk == bus.pk) + # Remove hardcoded people + free_seats -= WEIMembership.objects.filter(bus=bus, registration__first_year=True, + registration__information_json__icontains="hardcoded").count() + quotas[bus] = free_seats + + if display_tqdm: + tqdm_obj.close() + + from tqdm import tqdm + tqdm_obj = tqdm(total=len(men), desc="Hommes") + + self.make_repartition(men, quotas, tqdm_obj=tqdm_obj) + + if display_tqdm: + tqdm_obj.close() + + # Clear cache information after running algorithm + WEISurvey2024.clear_cache() + + def make_repartition(self, surveys, quotas=None, tqdm_obj=None): + free_surveys = surveys.copy() # Remaining surveys + while free_surveys: # Some students are not affected + survey = free_surveys[0] + buses = survey.ordered_buses() # Preferences of the student + for bus, current_score in buses: + if self.get_bus_information(bus).has_free_seats(surveys, quotas): + # Selected bus has free places. Put student in the bus + survey.select_bus(bus) + survey.save() + free_surveys.remove(survey) + break + else: + # Current bus has not enough places. Remove the least preferred student from the bus if existing + least_preferred_survey = None + least_score = -1 + # Find the least student in the bus that has a lower score than the current student + for survey2 in surveys: + if not survey2.information.valid or survey2.information.get_selected_bus() != bus: + continue + score2 = survey2.score(bus) + if current_score <= score2: # Ignore better students + continue + if least_preferred_survey is None or score2 < least_score: + least_preferred_survey = survey2 + least_score = score2 + + if least_preferred_survey is not None: + # Remove the least student from the bus and put the current student in. + # If it does not exist, choose the next bus. + least_preferred_survey.free() + least_preferred_survey.save() + free_surveys.append(least_preferred_survey) + survey.select_bus(bus) + survey.save() + free_surveys.remove(survey) + break + else: + raise ValueError(f"User {survey.registration.user} has no free seat") + + if tqdm_obj is not None: + tqdm_obj.n = len(surveys) - len(free_surveys) + tqdm_obj.refresh() diff --git a/apps/wei/migrations/0009_weiregistration_specific_diet.py b/apps/wei/migrations/0009_weiregistration_specific_diet.py new file mode 100644 index 00000000..8fa0f82a --- /dev/null +++ b/apps/wei/migrations/0009_weiregistration_specific_diet.py @@ -0,0 +1,18 @@ +# Generated by Django 4.2.15 on 2024-08-28 20:47 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('wei', '0008_auto_20240111_1545'), + ] + + operations = [ + migrations.AddField( + model_name='weiregistration', + name='specific_diet', + field=models.TextField(blank=True, default='', verbose_name='specific diet'), + ), + ] diff --git a/apps/wei/models.py b/apps/wei/models.py index 76fd465d..80584268 100644 --- a/apps/wei/models.py +++ b/apps/wei/models.py @@ -232,6 +232,12 @@ class WEIRegistration(models.Model): verbose_name=_("health issues"), ) + specific_diet = models.TextField( + blank=True, + default="", + verbose_name=_("specific diet"), + ) + emergency_contact_name = models.CharField( max_length=255, verbose_name=_("emergency contact name"), diff --git a/apps/wei/templates/wei/1A_list.html b/apps/wei/templates/wei/1A_list.html index d9b82937..9229fa22 100644 --- a/apps/wei/templates/wei/1A_list.html +++ b/apps/wei/templates/wei/1A_list.html @@ -12,7 +12,7 @@
{% render_table bus_repartition_table %}
- {% trans "Start attribution!" %} + {% trans "Start attribution !" %}
{% render_table table %}
diff --git a/apps/wei/templates/wei/attribute_bus_1A.html b/apps/wei/templates/wei/attribute_bus_1A.html index 3305981b..89dc692b 100644 --- a/apps/wei/templates/wei/attribute_bus_1A.html +++ b/apps/wei/templates/wei/attribute_bus_1A.html @@ -28,6 +28,9 @@
{% trans 'health issues'|capfirst %}
{{ object.health_issues|default:"—" }}
+
{% trans 'specific diet'|capfirst %}
+
{{ object.specific_diet|default:"—" }}
+
{% trans 'suggested bus'|capfirst %}
{{ survey.information.selected_bus_name }}
diff --git a/apps/wei/templates/wei/weimembership_form.html b/apps/wei/templates/wei/weimembership_form.html index 017b0dd7..ec6ebed4 100644 --- a/apps/wei/templates/wei/weimembership_form.html +++ b/apps/wei/templates/wei/weimembership_form.html @@ -67,6 +67,9 @@ SPDX-License-Identifier: GPL-3.0-or-later
{% trans 'health issues'|capfirst %}
{{ registration.health_issues }}
+
{% trans 'specific diet'|capfirst %}
+
{{ registration.specific_diet }}
+
{% trans 'emergency contact name'|capfirst %}
{{ registration.emergency_contact_name }}
diff --git a/apps/wei/tests/test_wei_algorithm_2023.py b/apps/wei/tests/test_wei_algorithm_2023.py index c94cbb0c..55c4a6d7 100644 --- a/apps/wei/tests/test_wei_algorithm_2023.py +++ b/apps/wei/tests/test_wei_algorithm_2023.py @@ -6,8 +6,6 @@ from datetime import date, timedelta from django.contrib.auth.models import User from django.test import TestCase -from django.urls import reverse -from note.models import NoteUser from ..forms.surveys.wei2023 import WEIBusInformation2023, WEISurvey2023, WORDS, WEISurveyInformation2023 from ..models import Bus, WEIClub, WEIRegistration @@ -127,44 +125,3 @@ class TestWEIAlgorithm(TestCase): self.assertLessEqual(max_score - score, 25) # Always less than 25 % of tolerance self.assertLessEqual(penalty / 100, 25) # Tolerance of 5 % - - def test_register_1a(self): - """ - Test register a first year member to the WEI and complete the survey - """ - response = self.client.get(reverse("wei:wei_register_1A", kwargs=dict(wei_pk=self.wei.pk))) - self.assertEqual(response.status_code, 200) - - user = User.objects.create(username="toto", email="toto@example.com") - NoteUser.objects.create(user=user) - response = self.client.post(reverse("wei:wei_register_1A", kwargs=dict(wei_pk=self.wei.pk)), dict( - user=user.id, - soge_credit=True, - birth_date=date(2000, 1, 1), - gender='nonbinary', - clothing_cut='female', - clothing_size='XS', - health_issues='I am a bot', - emergency_contact_name='NoteKfet2020', - emergency_contact_phone='+33123456789', - )) - qs = WEIRegistration.objects.filter(user_id=user.id) - self.assertTrue(qs.exists()) - registration = qs.get() - self.assertRedirects(response, reverse("wei:wei_survey", kwargs=dict(pk=registration.pk)), 302, 200) - for question in WORDS: - # Fill 1A Survey, 20 pages - # be careful if questionnary form change (number of page, type of answer...) - response = self.client.post(reverse("wei:wei_survey", kwargs=dict(pk=registration.pk)), { - question: "1" - }) - registration.refresh_from_db() - survey = WEISurvey2023(registration) - self.assertRedirects(response, reverse("wei:wei_survey", kwargs=dict(pk=registration.pk)), 302, - 302 if survey.is_complete() else 200) - self.assertIsNotNone(getattr(survey.information, question), "Survey page " + question + " failed") - survey = WEISurvey2023(registration) - self.assertTrue(survey.is_complete()) - survey.select_bus(self.buses[0]) - survey.save() - self.assertIsNotNone(survey.information.get_selected_bus()) diff --git a/apps/wei/tests/test_wei_algorithm_2024.py b/apps/wei/tests/test_wei_algorithm_2024.py new file mode 100644 index 00000000..a22b0a3a --- /dev/null +++ b/apps/wei/tests/test_wei_algorithm_2024.py @@ -0,0 +1,172 @@ +# Copyright (C) 2018-2024 by BDE ENS Paris-Saclay +# SPDX-License-Identifier: GPL-3.0-or-later + +import random +from datetime import date, timedelta + +from django.contrib.auth.models import User +from django.test import TestCase +from django.urls import reverse +from note.models import NoteUser + +from ..forms.surveys.wei2024 import WEIBusInformation2024, WEISurvey2024, WORDS, WEISurveyInformation2024 +from ..models import Bus, WEIClub, WEIRegistration + + +class TestWEIAlgorithm(TestCase): + """ + Run some tests to ensure that the WEI algorithm is working well. + """ + fixtures = ('initial',) + + def setUp(self): + """ + Create some test data, with one WEI and 10 buses with random score attributions. + """ + self.user = User.objects.create_superuser( + username="weiadmin", + password="admin", + email="admin@example.com", + ) + self.user.save() + self.client.force_login(self.user) + sess = self.client.session + sess["permission_mask"] = 42 + sess.save() + + self.wei = WEIClub.objects.create( + name="WEI 2024", + email="wei2024@example.com", + parent_club_id=2, + membership_fee_paid=12500, + membership_fee_unpaid=5500, + membership_start='2024-01-01', + membership_end='2024-12-31', + date_start=date.today() + timedelta(days=2), + date_end='2024-12-31', + year=2024, + ) + + self.buses = [] + for i in range(10): + bus = Bus.objects.create(wei=self.wei, name=f"Bus {i}", size=10) + self.buses.append(bus) + information = WEIBusInformation2024(bus) + for question in WORDS: + information.scores[question] = {answer: random.randint(1, 5) for answer in WORDS[question][1]} + information.save() + bus.save() + + def test_survey_algorithm_small(self): + """ + There are only a few people in each bus, ensure that each person has its best bus + """ + # Add a few users + for i in range(10): + user = User.objects.create(username=f"user{i}") + registration = WEIRegistration.objects.create( + user=user, + wei=self.wei, + first_year=True, + birth_date='2000-01-01', + ) + information = WEISurveyInformation2024(registration) + for question in WORDS: + options = list(WORDS[question][1].keys()) + setattr(information, question, random.choice(options)) + information.step = 20 + information.save(registration) + registration.save() + + # Run algorithm + WEISurvey2024.get_algorithm_class()().run_algorithm() + + # Ensure that everyone has its first choice + for r in WEIRegistration.objects.filter(wei=self.wei).all(): + survey = WEISurvey2024(r) + preferred_bus = survey.ordered_buses()[0][0] + chosen_bus = survey.information.get_selected_bus() + self.assertEqual(preferred_bus, chosen_bus) + + def test_survey_algorithm_full(self): + """ + Buses are full of first year people, ensure that they are happy + """ + # Add a lot of users + for i in range(95): + user = User.objects.create(username=f"user{i}") + registration = WEIRegistration.objects.create( + user=user, + wei=self.wei, + first_year=True, + birth_date='2000-01-01', + ) + information = WEISurveyInformation2024(registration) + for question in WORDS: + options = list(WORDS[question][1].keys()) + setattr(information, question, random.choice(options)) + information.step = 20 + information.save(registration) + registration.save() + + # Run algorithm + WEISurvey2024.get_algorithm_class()().run_algorithm() + + penalty = 0 + # Ensure that everyone seems to be happy + # We attribute a penalty for each user that didn't have its first choice + # The penalty is the square of the distance between the score of the preferred bus + # and the score of the attributed bus + # We consider it acceptable if the mean of this distance is lower than 5 % + for r in WEIRegistration.objects.filter(wei=self.wei).all(): + survey = WEISurvey2024(r) + chosen_bus = survey.information.get_selected_bus() + buses = survey.ordered_buses() + score = min(v for bus, v in buses if bus == chosen_bus) + max_score = buses[0][1] + penalty += (max_score - score) ** 2 + + self.assertLessEqual(max_score - score, 25) # Always less than 25 % of tolerance + + self.assertLessEqual(penalty / 100, 25) # Tolerance of 5 % + + def test_register_1a(self): + """ + Test register a first year member to the WEI and complete the survey + """ + response = self.client.get(reverse("wei:wei_register_1A", kwargs=dict(wei_pk=self.wei.pk))) + self.assertEqual(response.status_code, 200) + + user = User.objects.create(username="toto", email="toto@example.com") + NoteUser.objects.create(user=user) + response = self.client.post(reverse("wei:wei_register_1A", kwargs=dict(wei_pk=self.wei.pk)), dict( + user=user.id, + soge_credit=True, + birth_date=date(2000, 1, 1), + gender='nonbinary', + clothing_cut='female', + clothing_size='XS', + health_issues='I am a bot', + emergency_contact_name='NoteKfet2020', + emergency_contact_phone='+33123456789', + )) + qs = WEIRegistration.objects.filter(user_id=user.id) + self.assertTrue(qs.exists()) + registration = qs.get() + self.assertRedirects(response, reverse("wei:wei_survey", kwargs=dict(pk=registration.pk)), 302, 200) + for question in WORDS: + # Fill 1A Survey, 10 pages + # be careful if questionnary form change (number of page, type of answer...) + response = self.client.post(reverse("wei:wei_survey", kwargs=dict(pk=registration.pk)), { + question: "1" + }) + registration.refresh_from_db() + survey = WEISurvey2024(registration) + self.assertRedirects(response, reverse("wei:wei_survey", kwargs=dict(pk=registration.pk)), 302, + 302 if survey.is_complete() else 200) + self.assertIsNotNone(getattr(survey.information, question), "Survey page " + question + " failed") + survey = WEISurvey2024(registration) + self.assertTrue(survey.is_complete()) + survey.select_bus(self.buses[0]) + survey.save() + self.assertIsNotNone(survey.information.get_selected_bus()) diff --git a/apps/wei/tests/test_wei_registration.py b/apps/wei/tests/test_wei_registration.py index 2470b48d..21da07f8 100644 --- a/apps/wei/tests/test_wei_registration.py +++ b/apps/wei/tests/test_wei_registration.py @@ -767,7 +767,7 @@ class TestDefaultWEISurvey(TestCase): WEISurvey.update_form(None, None) self.assertEqual(CurrentSurvey.get_algorithm_class().get_survey_class(), CurrentSurvey) - self.assertEqual(CurrentSurvey.get_year(), 2023) + self.assertEqual(CurrentSurvey.get_year(), 2024) class TestWeiAPI(TestAPI): diff --git a/apps/wei/views.py b/apps/wei/views.py index 7a0e32fe..76943198 100644 --- a/apps/wei/views.py +++ b/apps/wei/views.py @@ -900,6 +900,9 @@ class WEIValidateRegistrationView(ProtectQuerysetMixin, ProtectedCreateView): form.fields["last_name"].initial = registration.user.last_name form.fields["first_name"].initial = registration.user.first_name + if "caution_check" in form.fields: + form.fields["caution_check"].initial = registration.caution_check + if registration.soge_credit: form.fields["credit_type"].disabled = True form.fields["credit_type"].initial = NoteSpecial.objects.get(special_type="Virement bancaire") @@ -941,6 +944,9 @@ class WEIValidateRegistrationView(ProtectQuerysetMixin, ProtectedCreateView): club = registration.wei user = registration.user + if "caution_check" in form.data: + registration.caution_check = form.data["caution_check"] == "on" + registration.save() membership = form.instance membership.user = user membership.club = club diff --git a/locale/fr/LC_MESSAGES/django.po b/locale/fr/LC_MESSAGES/django.po index a73797e4..b02ca86d 100644 --- a/locale/fr/LC_MESSAGES/django.po +++ b/locale/fr/LC_MESSAGES/django.po @@ -3001,6 +3001,7 @@ msgid "Credit from Société générale" msgstr "Crédit de la Société générale" #: apps/wei/models.py:188 +#: apps/wei/forms/registration.py:84 msgid "Caution check given" msgstr "ChÚque de caution donné"