django user_passes_test 装饰器

Question

我该如何在基于类的视图中使用 @user_passes_test(lambda u: u.is_superuser) 这个装饰器呢？我之前在基于函数的视图中用过这个，虽然我有个变通的方法，但感觉不太自然。

难道这不应该在调度方法中处理吗？

Answer 1

我使用了这个代码片段中的 @view_decorator：http://djangosnippets.org/snippets/2505/，来包装我普通的函数装饰器。

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

为了避免出现意外的方法解析顺序错误，确保这个混入类是你继承的第一个类。

你可以用同样的方法实现一个LoginRequiredMixin，或者你在应用中使用的其他常见测试。

编辑： Django 1.9增加了AccessMixin、LoginRequiredMixin、PermissionRequiredMixin和UserPassesTestMixin。

Answer 3

你可以在这个类的 dispatch 方法上使用 @method_decorator 装饰器：

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)