Pool support

apps/tournament/forms.py

@@ -3,14 +3,20 @@ import re
 from datetime import datetime
 
 from django import forms
+from django.db.models import Q
 from django.utils.translation import gettext_lazy as _
 
 from member.models import TFJMUser, Solution, Synthesis
 from tfjm.inputs import DatePickerInput, DateTimePickerInput, AmountInput
-from tournament.models import Tournament, Team
+from tournament.models import Tournament, Team, Pool
 
 
 class TournamentForm(forms.ModelForm):
+    organizers = forms.ModelMultipleChoiceField(
+        TFJMUser.objects.filter(Q(role="0admin") | Q(role="1volunteer")).order_by('role'),
+        label=_("Organizers"),
+    )
+
     def clean(self):
         cleaned_data = super().clean()
 
@@ -122,3 +128,74 @@ class SynthesisForm(forms.ModelForm):
     class Meta:
         model = Synthesis
         fields = ('file', 'source', 'round',)
+
+
+class PoolForm(forms.ModelForm):
+    team1 = forms.ModelChoiceField(
+        Team.objects.filter(validation_status="2valid").all(),
+        empty_label=_("Choose a team..."),
+        label=_("Team 1"),
+    )
+
+    problem1 = forms.IntegerField(
+        min_value=1,
+        max_value=8,
+        initial=1,
+        label=_("Problem defended by team 1"),
+    )
+
+    team2 = forms.ModelChoiceField(
+        Team.objects.filter(validation_status="2valid").all(),
+        empty_label=_("Choose a team..."),
+        label=_("Team 2"),
+    )
+
+    problem2 = forms.IntegerField(
+        min_value=1,
+        max_value=8,
+        initial=2,
+        label=_("Problem defended by team 2"),
+    )
+
+    team3 = forms.ModelChoiceField(
+        Team.objects.filter(validation_status="2valid").all(),
+        empty_label=_("Choose a team..."),
+        label=_("Team 3"),
+    )
+
+    problem3 = forms.IntegerField(
+        min_value=1,
+        max_value=8,
+        initial=3,
+        label=_("Problem defended by team 3"),
+    )
+
+    def clean(self):
+        cleaned_data = super().clean()
+
+        team1, pb1 = cleaned_data["team1"], cleaned_data["problem1"]
+        team2, pb2 = cleaned_data["team2"], cleaned_data["problem2"]
+        team3, pb3 = cleaned_data["team3"], cleaned_data["problem3"]
+
+        sol1 = Solution.objects.get(team=team1, problem=pb1, final=team1.selected_for_final)
+        sol2 = Solution.objects.get(team=team2, problem=pb2, final=team2.selected_for_final)
+        sol3 = Solution.objects.get(team=team3, problem=pb3, final=team3.selected_for_final)
+
+        cleaned_data["teams"] = [team1, team2, team3]
+        cleaned_data["solutions"] = [sol1, sol2, sol3]
+
+        return cleaned_data
+
+    def save(self, commit=True):
+        pool = super().save(commit)
+
+        pool.refresh_from_db()
+        pool.teams.set(self.cleaned_data["teams"])
+        pool.solutions.set(self.cleaned_data["solutions"])
+        pool.save()
+
+        return pool
+
+    class Meta:
+        model = Pool
+        fields = ('round', 'juries',)