Django: Staff Dekorator

Question

Próbuję napisać „personel tylko” dekorator dla Django, ale nie wydaje się uzyskać go do pracy:

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 

Próbując follow lead from here. Dostaję:

'WSGIRequest' object has no attribute '__name__'

Ale jeśli wziąć te 3 linie, ja po prostu bezużyteczne „wewnętrzny błąd serwera”. Co ja tu robię źle?

Answer 1

Ten styl funkcji dekorator jest używany z parametryzowane dekoratora - np kiedy zrobić:

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

Jeśli nie masz rzeczywiście wywołując funkcję zewnętrzną, czyli używasz go tak:

@staffonly 
def function(request): 

Otrzymasz dziwne wyniki, ponieważ obiekt funkcji zostanie przekazany do niewłaściwej funkcji zagnieżdżonej w dekoratorze.

Answer 2

ten dekorator już istnieje jako

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

@staff_member_required 

Trunk: http://code.djangoproject.com/browser/django/trunk/django/contrib/admin/views/decorators.py

Answer 3

Dla klasy opartych Zobacz s, można ozdobić metody wysyłają klasy widoku, podobnie jak to:

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