bornhack-website/src/teams/views/mixins.py

from django.contrib import messages
from django.shortcuts import redirect
from django.views.generic.detail import SingleObjectMixin

from teams.models import Team, TeamMember


class EnsureTeamResponsibleMixin(object):
    """
    Use to make sure request.user is responsible for the team specified by kwargs['team_slug']
    """

    def dispatch(self, request, *args, **kwargs):
        self.team = Team.objects.get(slug=kwargs["team_slug"], camp=self.camp)
        if request.user not in self.team.responsible_members.all():
            messages.error(request, "No thanks")
            return redirect(
                "teams:general", camp_slug=self.camp.slug, team_slug=self.team.slug
            )

        return super().dispatch(request, *args, **kwargs)


class EnsureTeamMemberResponsibleMixin(SingleObjectMixin):
    """
    Use to make sure request.user is responsible for the team which TeamMember belongs to
    """

    model = TeamMember

    def dispatch(self, request, *args, **kwargs):
        if request.user not in self.get_object().team.responsible_members.all():
            messages.error(request, "No thanks")
            return redirect(
                "teams:general",
                camp_slug=self.get_object().team.camp.slug,
                team_slug=self.get_object().team.slug,
            )

        return super().dispatch(request, *args, **kwargs)


class TeamViewMixin:
    def get_team(self):
        return self.get_object().team

    def get_context_data(self, *args, **kwargs):
        context = super().get_context_data(**kwargs)
        context["team"] = self.get_team()
        return context