Django: Оформитель персонала

Я пытаюсь написать декоратор "только для сотрудников" для 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 строки, я просто получаю бесполезную "Внутреннюю ошибку сервера". Что я здесь делаю неправильно?

Ответы

Ответ 2

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

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

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

@staffonly
def function(request):

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

Ответ 3

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

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


@method_decorator(staff_member_required, name='dispatch')
class ExampleTemplateView(TemplateView):
    ...