django @login_required trang trí cho một superuser

Question

Có một trang trí trong django tương tự như @login_required cũng kiểm tra xem người dùng có phải là siêu người dùng không?

Cảm ơn

Answer 1

Sử dụng user_passes_test trang trí:

from django.contrib.auth.decorators import user_passes_test 

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

Answer 2

Trong trường hợp thành viên đội ngũ nhân viên là đủ, bạn có thể sử dụng @staff_member_required trang trí:

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

@staff_member_required 
def my_view(request): 
    ... 

Answer 3

Nếu bạn muốn có chức năng tương tự như @staff_member_required bạn có thể dễ dàng viết trang trí của riêng bạn. Lấy @staff_member làm ví dụ chúng ta có thể làm sth như thế này.

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 

Ví dụ này là sửa đổi staff_member_required, vừa thay đổi một lần kiểm tra trong lambda.