décorateur django @login_required pour un superutilisateur

Question

Existe-t-il un décorateur dans django similaire à @login_required qui teste également si l'utilisateur est un superutilisateur ?

Merci

Answer 1

Utilisez le user_passes_test décorateur :

from django.contrib.auth.decorators import user_passes_test

@user_passes_test(lambda u: u.is_superuser)
def my_view(request):
    ...

Answer 2

En cas adhésion du personnel est suffisante et vous faites pas vous devez vérifier si l'utilisateur est un superutilisateur, vous pouvez utiliser la fonction @staff_member_required décorateur :

from django.contrib.admin.views.decorators import staff_member_required

@staff_member_required
def my_view(request):
    ...

Answer 3

Si vous voulez avoir une fonctionnalité similaire à celle de @staff_member_required, vous pouvez facilement écrire votre propre décorateur. En prenant @staff_member comme exemple, nous pouvons faire quelque chose comme ceci :

from django.contrib.auth import REDIRECT_FIELD_NAME
from django.contrib.admin.views.decorators import user_passes_test

def superuser_required(view_func=None, redirect_field_name=REDIRECT_FIELD_NAME,
                   login_url='account_login_url'):
    """
    Decorator for views that checks that the user is logged in and is a
    superuser, redirecting to the login page if necessary.
    """
    actual_decorator = user_passes_test(
        lambda u: u.is_active and u.is_superuser,
        login_url=login_url,
        redirect_field_name=redirect_field_name
    )
    if view_func:
        return actual_decorator(view_func)
    return actual_decorator

Cet exemple est une version modifiée de staff_member_required, nous avons juste changé une vérification dans le lambda.

Answer 4

Je recommande d'utiliser des Mixins, par exemple :

from django.contrib.auth.mixins import UserPassesTestMixin

class SuperUserCheck(UserPassesTestMixin, View):
    def test_func(self):
        return self.request.user.is_superuser

Vous pouvez alors ajouter SuperUserCheck à View classe :

class MyView(SuperUserCheck, View):

Answer 5

Pour les vues basées sur des classes, création d'un décorateur réutilisable :

from django.contrib.auth.mixins import UserPassesTestMixin
from django.views.generic import View

def superuser_required():
    def wrapper(wrapped):
        class WrappedClass(UserPassesTestMixin, wrapped):
            def test_func(self):
                return self.request.user.is_superuser

        return WrappedClass
    return wrapper

@superuser_required()
class MyClassBasedView(View):
    def get(self, request):
        # ...