场景:
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