Upload notes from a CSV sheet

apps/participation/views.py

@@ -6,6 +6,7 @@ import os
 from zipfile import ZipFile
 
 from django.conf import settings
+from django.contrib import messages
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.contrib.sites.models import Site
 from django.core.exceptions import PermissionDenied
@@ -18,6 +19,7 @@ from django.urls import reverse_lazy
 from django.utils import timezone
 from django.utils.translation import gettext_lazy as _
 from django.views.generic import CreateView, DetailView, FormView, RedirectView, TemplateView, UpdateView, View
+from django.views.generic.detail import SingleObjectMixin
 from django.views.generic.edit import FormMixin, ProcessFormView
 from django_tables2 import SingleTableView
 from magic import Magic
@@ -28,7 +30,7 @@ from tfjm.views import AdminMixin, VolunteerMixin
 
 from .forms import JoinTeamForm, MotivationLetterForm, NoteForm, ParticipationForm, PassageForm, PoolForm, \
     PoolTeamsForm, RequestValidationForm, SolutionForm, SynthesisForm, TeamForm, TournamentForm, \
-    ValidateParticipationForm
+    UploadNotesForm, ValidateParticipationForm
 from .models import Note, Participation, Passage, Pool, Solution, Synthesis, Team, Tournament
 from .tables import NoteTable, ParticipationTable, PassageTable, PoolTable, TeamTable, TournamentTable
 
@@ -703,6 +705,43 @@ class PoolUpdateTeamsView(VolunteerMixin, UpdateView):
         return self.handle_no_permission()
 
 
+class PoolUploadNotesView(VolunteerMixin, FormView, DetailView):
+    model = Pool
+    form_class = UploadNotesForm
+    template_name = 'participation/upload_notes.html'
+
+    def dispatch(self, request, *args, **kwargs):
+        self.object = self.get_object()
+
+        if request.user.registration.is_admin or request.user.registration.is_volunteer \
+                and (self.object.tournament in request.user.registration.organized_tournaments.all()
+                     or request.user.registration in self.object.juries.all()):
+            return super().dispatch(request, *args, **kwargs)
+
+        return self.handle_no_permission()
+
+    @transaction.atomic
+    def form_valid(self, form):
+        pool = self.get_object()
+        parsed_notes = form.cleaned_data['parsed_notes']
+
+        for vr, notes in parsed_notes.items():
+            if vr not in pool.juries.all():
+                form.add_error('file', _("The following user is not registered as a jury:") + " " + str(vr))
+
+            for i, passage in enumerate(pool.passages.all()):
+                note = Note.objects.get(jury=vr, passage=passage)
+                passage_notes = notes[6 * i:6 * (i + 1)]
+                note.set_all(*passage_notes)
+                note.save()
+
+        messages.success(self.request, _("Notes were successfully uploaded."))
+        return super().form_valid(form)
+
+    def get_success_url(self):
+        return reverse_lazy('participation:pool_detail', args=(self.kwargs['pk'],))
+
+
 class PassageCreateView(VolunteerMixin, CreateView):
     model = Passage
     form_class = PassageForm