Джанго: штатный декоратор - PullRequest
48 голосов
/ 22 апреля 2010

Я пытаюсь написать декоратор "только для персонала" для Django, но я не могу заставить его работать:

def staff_only(error='Only staff may view this page.'):
    def _dec(view_func):
        def _view(request, *args, **kwargs):
            u = request.user
            if u.is_authenticated() and u.is_staff:
                return view_func(request, *args, **kwargs)
            messages.error(request, error)
            return HttpResponseRedirect(request.META.get('HTTP_REFERER', reverse('home')))
        _view.__name__ = view_func.__name__
        _view.__dict__ = view_func.__dict__
        _view.__doc__ = view_func.__doc__
        return _view
    return _dec

Пытаясь следовать примеру отсюда . Я получаю:

'WSGIRequest' object has no attribute '__name__'

Но если я вычеркну эти 3 строки, я получу бесполезную «Внутреннюю ошибку сервера». Что я тут не так делаю?

Ответы [ 3 ]

144 голосов
/ 23 апреля 2010

Этот декоратор уже существует как

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

@staff_member_required

Ствол: http://code.djangoproject.com/browser/django/trunk/django/contrib/admin/views/decorators.py

11 голосов
/ 12 октября 2017

Для Представления на основе классов с, вы можете украсить метод отправки класса представления, например:

11 голосов
/ 22 апреля 2010

Этот стиль функции декоратора используется с параметризованным декоратором - например, когда вы делаете:

@staffonly(my_arguments)
def function(request):
    blah

Если вы на самом деле не вызываете внешнюю функцию, то есть используете ее так:

@staffonly
def function(request):

Вы получите странные результаты, так как объект функции будет передан не той одной из вложенных функций в декоратор.

...