Profile list

apps/tournament/views.py

@@ -74,6 +74,32 @@ class TournamentDetailView(DetailView):
 class TeamDetailView(LoginRequiredMixin, DetailView):
     model = Team
 
+    def dispatch(self, request, *args, **kwargs):
+        if not request.user.admin and self.request.user not in self.get_object().tournament.organizers:
+            raise PermissionDenied
+        return super().dispatch(request, *args, **kwargs)
+
+    def post(self, request, *args, **kwargs):
+        team = self.get_object()
+        if "zip" in request.POST:
+            solutions = team.solutions.all()
+
+            out = BytesIO()
+            zf = zipfile.ZipFile(out, "w")
+
+            for solution in solutions:
+                zf.write(solution.file.path, str(solution) + ".pdf")
+
+            zf.close()
+
+            resp = HttpResponse(out.getvalue(), content_type="application/x-zip-compressed")
+            resp['Content-Disposition'] = 'attachment; filename={}'\
+                .format(_("Solutions for team {team}.zip")
+                        .format(team=str(team)).replace(" ", "%20"))
+            return resp
+
+        return self.get(request, *args, **kwargs)
+
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)