Add & join teams

apps/member/views.py

@@ -1,14 +1,18 @@
+import random
+
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.contrib.auth.models import AnonymousUser
 from django.core.exceptions import PermissionDenied
 from django.db.models import Q
 from django.http import FileResponse
 from django.shortcuts import redirect
+from django.urls import reverse_lazy
 from django.utils.translation import gettext_lazy as _
 from django.views import View
-from django.views.generic import CreateView, UpdateView, DetailView
+from django.views.generic import CreateView, UpdateView, DetailView, FormView
 from django_tables2 import SingleTableView
 
+from tournament.forms import TeamForm, JoinTeam
 from tournament.models import Team
 from tournament.views import AdminMixin, TeamMixin
 from .forms import SignUpForm, TFJMUserForm
@@ -65,28 +69,49 @@ class UserDetailView(LoginRequiredMixin, DetailView):
         return context
 
 
-class MyTeamView(TeamMixin, DetailView):
+class AddTeamView(LoginRequiredMixin, CreateView):
     model = Team
+    form_class = TeamForm
 
-    def get_object(self, queryset=None):
-        return self.request.user.team
+    def form_valid(self, form):
+        team = form.instance
+        alphabet = "0123456789abcdefghijklmnopqrstuvwxyz0123456789"
+        code = ""
+        for _ in range(6):
+            code += random.choice(alphabet)
+        team.access_code = code
+        team.validation_status = "0invalid"
 
-    def dispatch(self, request, *args, **kwargs):
-        if isinstance(request.user, AnonymousUser):
-            raise PermissionDenied
+        team.save()
+        team.refresh_from_db()
 
-        team = self.get_object()
+        self.request.user.team = team
+        self.request.user.save()
 
-        if not request.user.participates or team is None:
-            raise PermissionDenied
-        return super().dispatch(request, *args, **kwargs)
+        return super().form_valid(form)
 
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
+    def get_success_url(self):
+        return reverse_lazy("member:my_team")
 
-        context["title"] = str(self.object)
 
-        return context
+class JoinTeamView(LoginRequiredMixin, FormView):
+    model = Team
+    form_class = JoinTeam
+    template_name = "tournament/team_form.html"
+
+    def form_valid(self, form):
+        team = form.cleaned_data["team"]
+        self.request.user.team = team
+        self.request.user.save()
+        return super().form_valid(form)
+
+    def get_success_url(self):
+        return reverse_lazy("member:my_team")
+
+
+class MyTeamView(TeamMixin, View):
+    def get(self, request, *args, **kwargs):
+        return redirect("tournament:team_detail", pk=request.user.team.pk)
 
 
 class DocumentView(LoginRequiredMixin, View):