django @login требуется декоратор для суперпользователя


есть ли декоратор в django, похожий на @login_required, который также проверяет, является ли пользователь суперпользователем?

спасибо

3   51   2012-08-17 14:05:39

3 ответа:

использовать user_passes_test оформитель:

from django.contrib.auth.decorators import user_passes_test

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

в случае, если штатное расписание является достаточным, вы можете использовать @staff_member_required оформитель:

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

@staff_member_required
def my_view(request):
    ...

Если вы хотите иметь аналогичную функциональность для @staff_member_required вы можете легко написать свой собственный декоратор. Взяв @staff_member в качестве примера, мы можем сделать sth так.

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

этот пример является модифицированным staff_member_required, просто изменен один чек в лямбда.