From c20771da8aa58f2183803d8d423e0fffa375f457 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?V=C3=AD=C3=B0ir=20Valberg=20Gu=C3=B0mundsson?=
 <valberg@orn.li>
Date: Fri, 24 Nov 2017 22:19:17 +0100
Subject: [PATCH] Fixed some stuff for ya' @tykling

---
 src/teams/templates/task_create.html |  2 +-
 src/teams/views.py                   | 17 ++++++++++++-----
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/src/teams/templates/task_create.html b/src/teams/templates/task_create.html
index 7fde553d..536eec60 100644
--- a/src/teams/templates/task_create.html
+++ b/src/teams/templates/task_create.html
@@ -14,7 +14,7 @@ Create Task for {{ task.team.name }} Team
             <button type="submit" class="btn btn-primary">
         </form>
     </div>
-    <div class="panel-footer"><i>This task belongs to the <a href="{% url 'teams:detail' slug=team.slug camp_slug=team.camp.slug %}">{{ team.name }} Team</a></i></div>
+    <div class="panel-footer"><i>This task belongs to the <a href="{% url 'teams:detail' team_slug=team.slug camp_slug=team.camp.slug %}">{{ team.name }} Team</a></i></div>
 </div>
 {% endblock %}
 
diff --git a/src/teams/views.py b/src/teams/views.py
index a856d71c..17152a2d 100644
--- a/src/teams/views.py
+++ b/src/teams/views.py
@@ -17,11 +17,12 @@ import logging
 logger = logging.getLogger("bornhack.%s" % __name__)
 
 
-class EnsureTeamResponsibleMixin(SingleObjectMixin):
+class EnsureTeamResponsibleMixin(object):
     def dispatch(self, request, *args, **kwargs):
-        if request.user not in self.get_object().responsible.all():
+        self.team = Team.objects.get(slug=kwargs['team_slug'], camp=self.camp)
+        if request.user not in self.team.responsible.all():
             messages.error(request, 'No thanks')
-            return redirect('teams:detail', camp_slug=self.camp.slug, team_slug=self.get_object().slug)
+            return redirect('teams:detail', camp_slug=self.camp.slug, team_slug=self.team.slug)
 
         return super().dispatch(
             request, *args, **kwargs
@@ -161,10 +162,16 @@ class TaskCreateView(LoginRequiredMixin, CampViewMixin, EnsureTeamResponsibleMix
 
     def get_context_data(self, *args, **kwargs):
         context = super().get_context_data(**kwargs)
-        self.team = Team.objects.get(slug=kwargs['team_slug'], camp=self.camp)
         context['team'] = self.team
         return context
 
     def get_success_url(self):
-        return reverse_lazy('task_detail', kwargs={'camp_slug': self.camp.slug, 'team_slug': self.team.slug, 'slug': self.get_object().slug})
+        return reverse_lazy(
+            'task_detail',
+            kwargs={
+                'camp_slug': self.camp.slug,
+                'team_slug': self.team.slug,
+                'slug': self.get_object().slug
+            }
+        )