gestiojeux/gestiojeux_auth/views.py

from django.views.generic import TemplateView, RedirectView
from django.shortcuts import redirect
from django.urls import reverse
from django.dispatch import receiver
from django.contrib.auth import logout as auth_logout
from django.contrib.auth import user_logged_in, user_logged_out, user_login_failed
from django.contrib import messages

from urllib.parse import quote as urlquote


class LoginView(TemplateView):
    template_name = "registration/login_switch.html"

    def dispatch(self, request, *args, **kwargs):
        if request.user.is_authenticated:
            messages.warning(
                request,
                "Vous êtes déjà connecté·e en tant que {}.".format(
                    request.user.username
                ),
            )
            return redirect(self.get_next_url() or "/")

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

    def get_next_url(self):
        if self.request.method == "GET":
            req_dict = self.request.GET
        elif self.request.method == "POST":
            req_dict = self.request.POST
        return req_dict.get("next")

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)

        next_url = self.get_next_url()
        if next_url:
            context["pass_url"] = "{}?next={}".format(
                reverse("gestiojeux_auth:password_login"), urlquote(next_url, safe="")
            )
            context["cas_url"] = "{}?next={}".format(
                reverse("gestiojeux_auth:cas_ng_login"), urlquote(next_url, safe="")
            )
        else:
            context["pass_url"] = reverse("gestiojeux_auth:password_login")
            context["cas_url"] = reverse("gestiojeux_auth:cas_ng_login")

        return context


class LogoutView(RedirectView):
    permanent = False

    def get_redirect_url(self, *args, **kwargs):
        CAS_BACKEND_NAME = "django_cas_ng.backends.CASBackend"
        if self.request.session["_auth_user_backend"] != CAS_BACKEND_NAME:
            auth_logout(self.request)
            if "next" in self.request.GET:
                return self.request.GET["next"]
            return reverse("mainsite:home")

        if "next" in self.request.GET:
            return "{}?next={}".format(
                reverse("gestiojeux_auth:cas_ng_logout"),
                urlquote(self.request.GET["next"], safe=""),
            )
        return reverse("gestiojeux_auth:cas_ng_logout")


@receiver(user_logged_in)
def on_login(request, user, **kwargs):
    messages.success(request, "Connexion réussie. Bienvenue, {}.".format(user))


@receiver(user_logged_out)
def on_logout(request, **kwargs):
    messages.info(request, "Vous avez bien été déconnecté·e.")


@receiver(user_login_failed)
def on_login_failed(request, **kwargs):
    messages.error(request, "Connexion échouée.")