Solutions

apps/tournament/views.py

@@ -1,12 +1,31 @@
+import zipfile
+from io import BytesIO
+
 from django.contrib.auth.mixins import LoginRequiredMixin
+from django.core.exceptions import PermissionDenied
 from django.db.models import Q
+from django.http import HttpResponse
 from django.utils.translation import gettext_lazy as _
 from django.views.generic import DetailView
 from django_tables2.views import SingleTableView
 
-from member.models import TFJMUser
+from member.models import TFJMUser, Solution
 from .models import Tournament, Team
-from .tables import TournamentTable, TeamTable
+from .tables import TournamentTable, TeamTable, SolutionTable
+
+
+class AdminMixin(object):
+    def dispatch(self, request, *args, **kwargs):
+        if not request.user.admin:
+            raise PermissionDenied
+        return super().dispatch(request, *args, **kwargs)
+
+
+class TeamMixin(object):
+    def dispatch(self, request, *args, **kwargs):
+        if not request.user.team:
+            raise PermissionDenied
+        return super().dispatch(request, *args, **kwargs)
 
 
 class TournamentListView(SingleTableView):
@@ -61,3 +80,88 @@ class TeamDetailView(LoginRequiredMixin, DetailView):
         context["title"] = _("Information about team")
 
         return context
+
+
+class SolutionsView(LoginRequiredMixin, TeamMixin, SingleTableView):
+    model = Solution
+    table_class = SolutionTable
+    template_name = "tournament/solutions_list.html"
+    extra_context = dict(title=_("Solutions"))
+
+    def post(self, request, *args, **kwargs):
+        if "zip" in request.POST:
+            solutions = request.user.team.solutions
+
+            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(request.user.team)).replace(" ", "%20"))
+            return resp
+
+        return self.get(request, *args, **kwargs)
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+
+        context["tournaments"] = \
+            Tournament.objects if self.request.user.admin else self.request.user.organized_tournaments
+
+        return context
+
+    def get_queryset(self):
+        qs = super().get_queryset()
+        if not self.request.user.admin:
+            qs = qs.filter(team__tournament__organizers=self.request.user)
+        return qs.order_by('team__tournament__date_start', 'team__tournament__name', 'team__trigram', 'problem',)
+
+
+class SolutionsOrgaListView(LoginRequiredMixin, AdminMixin, SingleTableView):
+    model = Solution
+    table_class = SolutionTable
+    template_name = "tournament/solutions_orga_list.html"
+    extra_context = dict(title=_("All solutions"))
+
+    def post(self, request, *args, **kwargs):
+        if "tournament_zip" in request.POST:
+            tournament = Tournament.objects.get(pk=request.POST["tournament_zip"][0])
+            solutions = tournament.solutions
+            if not request.user.admin and request.user not in tournament.organizers:
+                raise PermissionDenied
+
+            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 tournament {tournament}.zip")
+                        .format(tournament=str(tournament)).replace(" ", "%20"))
+            return resp
+
+        return self.get(request, *args, **kwargs)
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+
+        context["tournaments"] = \
+            Tournament.objects if self.request.user.admin else self.request.user.organized_tournaments
+
+        return context
+
+    def get_queryset(self):
+        qs = super().get_queryset()
+        if not self.request.user.admin:
+            qs = qs.filter(team__tournament__organizers=self.request.user)
+        return qs.order_by('team__tournament__date_start', 'team__tournament__name', 'team__trigram', 'problem',)