Add vaccine sheet field, closes #18

apps/registration/views.py

@@ -29,7 +29,7 @@ from tfjm.views import UserMixin, UserRegistrationMixin, VolunteerMixin
 
 from .forms import AddOrganizerForm, CoachRegistrationForm, HealthSheetForm, \
     ParentalAuthorizationForm, PaymentForm, PhotoAuthorizationForm, SignupForm, StudentRegistrationForm, UserForm, \
-    VolunteerRegistrationForm
+    VaccineSheetForm, VolunteerRegistrationForm
 from .models import ParticipantRegistration, Payment, Registration, StudentRegistration
 from .tables import RegistrationTable
 
@@ -344,6 +344,27 @@ class UserUploadHealthSheetView(UserRegistrationMixin, UpdateView):
         return reverse_lazy("registration:user_detail", args=(self.object.user.pk,))
 
 
+class UserUploadVaccineSheetView(UserRegistrationMixin, UpdateView):
+    """
+    A participant can send its vaccine sheet.
+    """
+    model = StudentRegistration
+    form_class = VaccineSheetForm
+    template_name = "registration/upload_vaccine_sheet.html"
+    extra_context = dict(title=_("Upload vaccine sheet"))
+
+    @transaction.atomic
+    def form_valid(self, form):
+        old_instance = StudentRegistration.objects.get(pk=self.object.pk)
+        if old_instance.vaccine_sheet:
+            old_instance.vaccine_sheet.delete()
+            old_instance.save()
+        return super().form_valid(form)
+
+    def get_success_url(self):
+        return reverse_lazy("registration:user_detail", args=(self.object.user.pk,))
+
+
 class UserUploadParentalAuthorizationView(UserRegistrationMixin, UpdateView):
     """
     A participant can send its parental authorization.
@@ -484,6 +505,29 @@ class HealthSheetView(LoginRequiredMixin, View):
         return FileResponse(open(path, "rb"), content_type=mime_type, filename=true_file_name)
 
 
+class VaccineSheetView(LoginRequiredMixin, View):
+    """
+    Display the sent health sheet.
+    """
+    def get(self, request, *args, **kwargs):
+        filename = kwargs["filename"]
+        path = f"media/authorization/vaccine/{filename}"
+        if not os.path.exists(path):
+            raise Http404
+        student = StudentRegistration.objects.get(vaccine_sheet__endswith=filename)
+        user = request.user
+        if not (student.user == user or user.registration.is_admin or user.registration.is_volunteer and student.team
+                and student.team.participation.tournament in user.registration.organized_tournaments.all()):
+            raise PermissionDenied
+        # Guess mime type of the file
+        mime = Magic(mime=True)
+        mime_type = mime.from_file(path)
+        ext = mime_type.split("/")[1].replace("jpeg", "jpg")
+        # Replace file name
+        true_file_name = _("Vaccine sheet of {student}.{ext}").format(student=str(student), ext=ext)
+        return FileResponse(open(path, "rb"), content_type=mime_type, filename=true_file_name)
+
+
 class ParentalAuthorizationView(LoginRequiredMixin, View):
     """
     Display the sent parental authorization.