django如何通过类视图使用装饰器
需求:当我们想禁止ip黑名单访问我们的某些页面时,例如注册页面。应该怎么操作呢?
解决方案:这时候我们可以设计一个装饰器,过滤ip黑名单。
装饰器的写法如下:
fromfunctoolsimportwraps defban_ip(func): @wraps(func) defwrapper(request,*args,**kwargs): IP=request.META.get("REMOTE_ADDR") ifIPin["127.0.0.1"]:#可以添加ip黑名单到列表里 returnHttpResponse("禁止访问") returnfunc returnwrapper
如果不是类视图,可以这样使用:
@ban_ip defget_register(request): returnrender(request,"注册页面.html") defpost_resiger(request): name=request.POST.get("name") password=request.POST.get("password") print(name) print(password) returnHttpResponse("注册成功")
这样就可以达到一个效果。如果ip在黑名单里,那么访问注册页面时会弹出“禁止访问”。
需求:但是,问题来了,当我们使用类视图时,定义视图函数时还有一个self参数。因为装饰器没有传这个参,所以会报错。应该怎么办呢?
解决办法:
(1)给类视图中特定的方法
classregister(View): @method_decorator(ban_ip) defget(self,request): returnrender(request,"注册页面.html") defpost(self,request): name=request.POST.get("name") password=request.POST.get("password") print(name) print(password) returnHttpResponse("注册成功")
解读:
- 注意:针对函数定义的装饰器,不能直接应用到类视图的方法中,因为少了一个self参数
- 解决:可以使用method_decorator装饰器,为函数装饰器补充第一个self参数,使它可以应用到类的方法中。
(2)给所有类视图的所有方法应用装饰器
类视图的实现是通过as_view()这个方法来分发不同的请求来实现了。我们可以利用这点来给类视图的所有方法都应用装饰器。
一个比较简单但是不实用的办法就是在项目的urls.py上面来实现。在分发请求的前面加上装饰器。
url(r'^register$',ban_ip(views.register.as_view()))
虽然这样比较方便,但是太不美观了。装饰器的优雅都被它破坏了,那我们应该怎么办呢?
解决办法:把as_view()的方法重写一下,给它加入装饰器的功能。
classregister(View): #重写请求分发方法,as_view()的核心语句 @method_decorator(ban_ip) defdispatch(self,request,*args,**kwargs): returnsuper().dispatch(request,*args,**kwargs) defget(self,request): returnrender(request,"注册页面.html") defpost(self,request): name=request.POST.get("name") password=request.POST.get("password") print(name) print(password) returnHttpResponse("注册成功")
(3)对类视图的某个方法应用装饰器
@method_decorator(ban_ip,name='get') classregister(View): defget(self,request): returnrender(request,"post.html") defpost(self,request): title=request.POST.get("title") content=request.POST.get("content") print(title) print(content) returnHttpResponse("发布成功")
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。