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的回答,您可能希望在多个视图类中执行此操作,因此将其转换为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