Collect from 网页模板
Modified by 追梦人物

Django ListView DetailView等基于类的视图如何添加装饰器?

场景:

Django开发中,如果我们使用了类视图,如:ListView、DetailView、UpdateView等,这时我们又想要对这个视图添加一个装饰器,来实现某种功能,这时候该怎么处理呢?

环境:

python 3.6

Django 1.11

错误用法

错误实现方式:

def is_login(func): 
    def wrapper(request,*args,**kwargs): 
    # 若检测不到用户就跳转登录页面 
        if not request.session.get("user"): 
            return redirect(reverse('login')) 
        return  func(request,*args, **kwargs) 
    return wrapper 

@is_login 
class myinfor(generic.ListView): 
    pass

报错信息:

AttributeError: 'function' object has no attribute 'as_view'

正确用法

实现方式一:

from django.utils.decorators import method_decorator 


def is_login(func): 
    def wrapper(request,*args,**kwargs): 
    # 若检测不到用户就跳转登录页面 
        if not request.session.get("user"): 
            return redirect(reverse('login')) 
        return  func(request,*args, **kwargs) 
    return wrapper 

# 使用method_decorator将装饰器包裹起来,同时,name参数是必须的,dispatch支持所有请求类型,包含get、post等,如果指定某种请求方式改为:name='get'. 教程源站(bigyoung.cn)

@method_decorator(is_login, name='dispatch') 
class myinfor(generic.ListView): 
    pass

实现方式二:

通过路由配置实现:(不推荐)

'''教程源站:BigYoung.cn'''

from django.utils.decorators import never_cache

urlpatterns += [
    path('myinfo/', never_cache(myinfor.as_view()), name='myinfo'),
]

进阶用法:

如果你有多个装饰器需要装饰,如下多个装饰器:

def is_login(func): 
    def wrapper(request,*args,**kwargs): 
    # 若检测不到用户就跳转登录页面 
        if not request.session.get("user"): 
            return redirect(reverse('login')) 
        return  func(request,*args, **kwargs) 
    return wrapper 

def is_admin(func): 
    # 检测是否为admin管理员。教程源站(bigyoung.cn)
    def wrapper(request,*args,**kwargs): 
        pass
    return wrapper 

实现方式一:

from django.utils.decorators import method_decorator 

decorators = [is_login, is_admin]  # 将需要的装饰器放在一个列表中,装饰器会按照列表中的顺序装饰目标对象

@method_decorator(decorators, name='dispatch')
class myinfor(generic.ListView): 
    pass                            

实现方式二:

'''教程源站(bigyoung.cn)'''

from django.utils.decorators import method_decorator 

@method_decorator(is_admin, name='dispatch') 
@method_decorator(is_login, name='dispatch') 
class myinfor(generic.ListView): 
    pass

更多用法,可以看看官方文档:https://docs.djangoproject.com/en/2.2/topics/class-based-views/intro/#id1