Minor tweaks to templates, permissions fix

2019-11-27 12:44:54 +01:00
parent a54ec52f52
commit c61adb84e2
4 changed files with 12 additions and 10 deletions
--- a/geogame/main/views.py
+++ b/geogame/main/views.py
@@ -162,11 +162,14 @@ class NewGameView(views.LoginRequiredMixin, View):
            )


-class RoundView(views.LoginRequiredMixin, UpdateView):
+class RoundView(views.UserPassesTestMixin, UpdateView):
    model = GameRound
    form_class = GuessForm
    template_name = 'main/round.html'

+    def test_func(self, *args, **kwargs):
+        return self.request.user == get_object_or_404(Game, pk=self.kwargs.get('game_pk', 0)).user
+
    def get_object(self):
        round_id = self.kwargs.get('round_pk', 0)
        return get_object_or_404(GameRound, pk=round_id)