Django user_passes_test декоратор

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

Не следует ли покрывать метод отправки?

Ответы

Ответ 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)

Ответ 2

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

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

Ответ 3

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