from django.contrib.auth.mixins import LoginRequiredMixin from django.core.exceptions import PermissionDenied from django.db import transaction from django.urls import reverse_lazy from django.utils.translation import gettext_lazy as _ from django.views.generic import CreateView, FormView, DetailView, RedirectView, UpdateView from .forms import TeamForm, JoinTeamForm, ParticipationForm from .models import Team class CreateTeamView(LoginRequiredMixin, CreateView): model = Team form_class = TeamForm extra_context = dict(title=_("Create team")) template_name = "participation/create_team.html" @transaction.atomic def form_valid(self, form): user = self.request.user registration = user.registration if not registration.participates: form.add_error(None, _("You don't participate, so you can't create a team.")) return self.form_invalid(form) elif registration.team: form.add_error(None, _("You are already in a team.")) return self.form_invalid(form) ret = super().form_valid(form) registration.team = form.instance registration.save() return ret def get_success_url(self): return reverse_lazy("index") class JoinTeamView(LoginRequiredMixin, FormView): model = Team form_class = JoinTeamForm extra_context = dict(title=_("Join team")) template_name = "participation/create_team.html" @transaction.atomic def form_valid(self, form): user = self.request.user registration = user.registration if not registration.participates: form.add_error(None, _("You don't participate, so you can't create a team.")) return self.form_invalid(form) elif registration.team: form.add_error(None, _("You are already in a team.")) return self.form_invalid(form) ret = super().form_valid(form) registration.team = form.instance registration.save() return ret def get_success_url(self): return reverse_lazy("index") class MyTeamDetailView(LoginRequiredMixin, RedirectView): def get_redirect_url(self, *args, **kwargs): user = self.request.user registration = user.registration if registration.participates: if registration.team: return reverse_lazy("participation:team_detail", args=(registration.team_id,)) raise PermissionDenied(_("You are not in a team.")) raise PermissionDenied(_("You don't participate, so you don't have any team.")) class TeamDetailView(LoginRequiredMixin, DetailView): model = Team class TeamUpdateView(LoginRequiredMixin, UpdateView): model = Team form_class = TeamForm template_name = "participation/update_team.html" def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context["participation_form"] = ParticipationForm(data=self.request.POST or None, instance=self.object.participation) return context @transaction.atomic def form_valid(self, form): participation_form = ParticipationForm(data=self.request.POST or None, instance=self.object.participation) if not participation_form.is_valid(): return self.form_invalid(form) participation_form.save() return super().form_valid(form) def get_success_url(self): return reverse_lazy("participation:team_detail", args=(self.object.pk,))