django user_passes_test декоратор

Question

Как мне реализовать декоратор @user_passes_test(lambda u: u.is_superuser) для представлений на основе классов?Я использовал это раньше для представлений, основанных на функциях, и у меня есть обходной путь, но он кажется неестественным.

Разве это не должно быть покрыто методом отправки?

Answer 1

Вы используете @method_decorator для dispatch метода класса:

from django.views.generic import View
from django.utils.decorators import method_decorator
from django.contrib.auth.decorators import user_passes_test

class MyView(View):
    @method_decorator(user_passes_test(lambda u: u.is_superuser))
    def dispatch(self, *args, **kwargs):
        return super(MyView, self).dispatch(*args, **kwargs)

Answer 2

Опираясь на ответ @Chris Pratt, вы, вероятно, захотите сделать это в нескольких классах представлений, поэтому имеет смысл превратить его в миксин.

class SuperuserRequiredMixin(object):
    @method_decorator(user_passes_test(lambda u: u.is_superuser))
    def dispatch(self, *args, **kwargs):
        return super(SuperuserRequiredMixin, self).dispatch(*args, **kwargs)

Использование

class MyView(SuperuserRequiredMixin, View):
    def get(self, request):
        ...

Чтобы предотвратить непредвиденные ошибки MRO , убедитесь, что mixin является первым унаследованным классом.

Таким же образом вы можете реализовать LoginRequiredMixin или любые другие общие тесты, которые вы используете в своем приложении.

Редактировать: Django 1.9 добавляет AccessMixin, LoginRequiredMixin, PermissionRequiredMixin и UserPassesTestMixin

Answer 3

Вы должны взглянуть на django-фигурные скобки и его UserPassesTestMixin .

Answer 4

Я использовал @view_decorator из этого фрагмента: http://djangosnippets.org/snippets/2505/, чтобы обернуть мои обычные функциональные декораторы.