Add harmonization view

Signed-off-by: Emmy D'Anello <emmy.danello@animath.fr>

participation/views.py

@@ -9,6 +9,7 @@ from tempfile import mkdtemp
 from typing import Any, Dict
 from zipfile import ZipFile
 
+import gspread
 from django.conf import settings
 from django.contrib import messages
 from django.contrib.auth.mixins import LoginRequiredMixin
@@ -41,7 +42,7 @@ from tfjm.views import AdminMixin, VolunteerMixin
 from .forms import AddJuryForm, JoinTeamForm, MotivationLetterForm, NoteForm, ParticipationForm, PassageForm, \
     PoolForm, PoolTeamsForm, RequestValidationForm, SolutionForm, SynthesisForm, TeamForm, TournamentForm, \
     UploadNotesForm, ValidateParticipationForm
-from .models import Note, Participation, Passage, Pool, Solution, Synthesis, Team, Tournament
+from .models import Note, Participation, Passage, Pool, Solution, Synthesis, Team, Tournament, Tweak
 from .tables import NoteTable, ParticipationTable, PassageTable, PoolTable, TeamTable, TournamentTable
 
 
@@ -710,6 +711,81 @@ class TournamentPublishNotesView(VolunteerMixin, SingleObjectMixin, RedirectView
         return reverse_lazy("participation:tournament_detail", args=(kwargs['pk'],))
 
 
+class TournamentHarmonizeView(VolunteerMixin, DetailView):
+    """
+    Harmonize the notes of a tournament.
+    """
+    model = Tournament
+    template_name = "participation/tournament_harmonize.html"
+
+    def dispatch(self, request, *args, **kwargs):
+        if not request.user.is_authenticated:
+            return self.handle_no_permission()
+        tournament = self.get_object()
+        reg = request.user.registration
+        if not reg.is_admin and (not reg.is_volunteer or tournament not in reg.organized_tournaments.all()):
+            return self.handle_no_permission()
+        if self.kwargs['round'] not in (1, 2):
+            raise Http404
+        return super().dispatch(request, *args, **kwargs)
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+
+        tournament = self.get_object()
+        context['round'] = self.kwargs['round']
+        context['pools'] = tournament.pools.filter(round=context["round"]).all()
+        context['title'] = _("Harmonize notes of {tournament} - Day {round}") \
+            .format(tournament=tournament, round=context["round"])
+
+        notes = dict()
+        for participation in self.object.participations.all():
+            note = sum(pool.average(participation) for pool in context['pools'])
+            tweak = sum(tweak.diff for tweak in participation.tweaks.filter(pool__in=context['pools']).all())
+            notes[participation] = {'note': note, 'tweak': tweak}
+        context["notes"] = sorted(notes.items(), key=lambda x: x[1]['note'], reverse=True)
+
+        return context
+
+
+class TournamentHarmonizeNoteView(VolunteerMixin, DetailView):
+    model = Tournament
+
+    def dispatch(self, request, *args, **kwargs):
+        if not request.user.is_authenticated:
+            return self.handle_no_permission()
+        tournament = self.get_object()
+        reg = request.user.registration
+        if not reg.is_admin and (not reg.is_volunteer or tournament not in reg.organized_tournaments.all()):
+            return self.handle_no_permission()
+        if self.kwargs['round'] not in (1, 2) or self.kwargs['action'] not in ('add', 'remove') \
+                or self.kwargs['trigram'] not in [p.team.trigram for p in tournament.participations.all()]:
+            raise Http404
+        return super().dispatch(request, *args, **kwargs)
+
+    def get(self, request, *args, **kwargs):
+        tournament = self.get_object()
+        participation = tournament.participations.get(team__trigram=kwargs['trigram'])
+        pool = tournament.pools.get(round=kwargs['round'], participations=participation)
+        tweak_qs = Tweak.objects.filter(participation=participation, pool=pool)
+        old_diff = tweak_qs.first().diff if tweak_qs.exists() else 0
+        new_diff = old_diff + (1 if kwargs['action'] == 'add' else -1)
+        if new_diff == 0:
+            tweak_qs.delete()
+        else:
+            tweak_qs.update_or_create(defaults={'diff': new_diff},
+                                      create_defaults={'diff': new_diff, 'participation': participation, 'pool': pool})
+
+        gc = gspread.service_account_from_dict(settings.GOOGLE_SERVICE_CLIENT)
+        spreadsheet = gc.open_by_key(tournament.notes_sheet_id)
+        worksheet = spreadsheet.worksheet("Classement final")
+        column = 3 if kwargs['round'] == '1' else 5
+        row = worksheet.find(f"{participation.team.name} ({participation.team.trigram})", in_column=1).row
+        worksheet.update_cell(row, column, new_diff)
+
+        return redirect(reverse_lazy("participation:tournament_harmonize", args=(tournament.pk, kwargs['round'],)))
+
+
 class SolutionUploadView(LoginRequiredMixin, FormView):
     template_name = "participation/upload_solution.html"
     form_class = SolutionForm