plateforme-tfjm2/apps/tournament/forms.py

import os
import re

from django import forms
from django.db.models import Q
from django.utils import timezone
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, Pool


class TournamentForm(forms.ModelForm):
    """
    Create and update tournaments.
    """

    # Only organizers can organize tournaments. Well, that's pretty normal...
    organizers = forms.ModelMultipleChoiceField(
        TFJMUser.objects.filter(Q(role="0admin") | Q(role="1volunteer")).order_by('role'),
        label=_("Organizers"),
    )

    def clean(self):
        cleaned_data = super().clean()

        if not self.instance.pk:
            if Tournament.objects.filter(name=cleaned_data["data"], year=os.getenv("TFJM_YEAR")):
                self.add_error("name", _("This tournament already exists."))
            if cleaned_data["final"] and Tournament.objects.filter(final=True, year=os.getenv("TFJM_YEAR")):
                self.add_error("name", _("The final tournament was already defined."))

        return cleaned_data

    class Meta:
        model = Tournament
        exclude = ('year',)
        widgets = {
            "price": AmountInput(),
            "date_start": DatePickerInput(),
            "date_end": DatePickerInput(),
            "date_inscription": DateTimePickerInput(),
            "date_solutions": DateTimePickerInput(),
            "date_syntheses": DateTimePickerInput(),
            "date_solutions_2": DateTimePickerInput(),
            "date_syntheses_2": DateTimePickerInput(),
        }


class OrganizerForm(forms.ModelForm):
    """
    Register an organizer in the website.
    """

    class Meta:
        model = TFJMUser
        fields = ('last_name', 'first_name', 'email', 'is_superuser',)

    def clean(self):
        cleaned_data = super().clean()

        if TFJMUser.objects.filter(email=cleaned_data["email"], year=os.getenv("TFJM_YEAR")).exists():
            self.add_error("email", _("This organizer already exist."))

        return cleaned_data

    def save(self, commit=True):
        user = self.instance
        user.role = '0admin' if user.is_superuser else '1volunteer'
        user.save()
        super().save(commit)


class TeamForm(forms.ModelForm):
    """
    Add and update a team.
    """
    tournament = forms.ModelChoiceField(
        Tournament.objects.filter(date_inscription__gte=timezone.now(), final=False),
    )

    class Meta:
        model = Team
        fields = ('name', 'trigram', 'tournament',)

    def clean(self):
        cleaned_data = super().clean()

        cleaned_data["trigram"] = cleaned_data["trigram"].upper()

        if not re.match("[A-Z]{3}", cleaned_data["trigram"]):
            self.add_error("trigram", _("The trigram must be composed of three upcase letters."))

        if not self.instance.pk:
            if Team.objects.filter(trigram=cleaned_data["trigram"], year=os.getenv("TFJM_YEAR")).exists():
                self.add_error("trigram", _("This trigram is already used."))

            if Team.objects.filter(name=cleaned_data["name"], year=os.getenv("TFJM_YEAR")).exists():
                self.add_error("name", _("This name is already used."))

            if cleaned_data["tournament"].date_inscription < timezone.now:
                self.add_error("tournament", _("This tournament is already closed."))

        return cleaned_data


class JoinTeam(forms.Form):
    """
    Form to join a team with an access code.
    """

    access_code = forms.CharField(
        label=_("Access code"),
        max_length=6,
    )

    def clean(self):
        cleaned_data = super().clean()

        if not re.match("[a-z0-9]{6}", cleaned_data["access_code"]):
            self.add_error('access_code', _("The access code must be composed of 6 alphanumeric characters."))

        team = Team.objects.filter(access_code=cleaned_data["access_code"])
        if not team.exists():
            self.add_error('access_code', _("This access code is invalid."))
        team = team.get()
        if not team.invalid:
            self.add_error('access_code', _("The team is already validated."))
        cleaned_data["team"] = team

        return cleaned_data


class SolutionForm(forms.ModelForm):
    """
    Form to upload a solution.
    """

    problem = forms.ChoiceField(
        label=_("Problem"),
        choices=[(str(i), _("Problem #{problem:d}").format(problem=i)) for i in range(1, 9)],
    )

    class Meta:
        model = Solution
        fields = ('file', 'problem',)


class SynthesisForm(forms.ModelForm):
    """
    Form to upload a synthesis.
    """

    class Meta:
        model = Synthesis
        fields = ('file', 'source', 'round',)


class PoolForm(forms.ModelForm):
    """
    Form to add a pool.
    Should not be used: prefer to pass by API and auto-add pools with the results of the draw.
    """

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