django @login_required bir süper kullanıcı için dekoratör


90

Django'da @login_required'a benzer, kullanıcının bir süper kullanıcı olup olmadığını da test eden bir dekoratör var mı?

Teşekkürler

Yanıtlar:



76

Durumunda personel üyelik yeterlidir ve do not kullanabileceğiniz, kullanıcı bir süper olup olmadığını kontrol etmeniz gerekir @staff_member_requireddekoratör:

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

@staff_member_required
def my_view(request):
    ...

34
Buraya geldiğimde aradığım şey buydu, bu yüzden buraya koydum ve burada bırakacağım çünkü başka biri için faydalı olabileceğini düşünüyorum.
Bit68

3
@ Bit68. Onu arıyorum :)
Shiv Shankar

7

@Staff_member_required ile benzer işlevselliğe sahip olmak istiyorsanız, kolayca kendi dekoratörünüzü yazabilirsiniz. @Staff_member'ı örnek alarak şöyle bir şey yapabiliriz:

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

Bu örnek değiştirilmiş bir staff_member_required, lambda'da bir çek değiştirildi.


4
Bu ithalat bildirimini de isteyeceksiniz :) from django.contrib.auth import REDIRECT_FIELD_NAME
Bryan Tarpley

3

Sınıf tabanlı görünümler için, yeniden kullanılabilir bir dekoratör oluşturmak:

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):
        # ...

3

Mixins kullanmanızı tavsiye ederim, örneğin:

from django.contrib.auth.mixins import UserPassesTestMixin


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

Sonra ekleyebilir SuperUserCheckiçin Viewsınıfın:

class MyView(SuperUserCheck, View):

2

kullanıcı profiliniz varsa, bunu kolayca yapabilirsiniz

@login_required
@user_passes_test(lambda u: True if u.profile.role==2 else False )
def add_listing(request):
    #...

1

Yeni kod yazmadan sınıf tabanlı bir görünümde bir süper kullanıcıya ihtiyaç duymak için:

from django.utils.decorators import method_decorator
from django.contrib.auth.decorators import user_passes_test

@method_decorator(user_passes_test(lambda u: u.is_superuser), name='dispatch')
class AdminCreateUserView(LoginRequiredMixin, FormView):
    ...
    ...
    ...
Sitemizi kullandığınızda şunları okuyup anladığınızı kabul etmiş olursunuz: Çerez Politikası ve Gizlilik Politikası.
Licensed under cc by-sa 3.0 with attribution required.