Django学习笔记第八篇--实战练习四--为你的视图函数自定义装饰器

零、背景：

对于登录后面所有视图函数，都需要验证登录信息，一般而言就是验证cookie里面的一些信息。所以你可以这么写函数：

 def personinfo(request):
     if request.COOKIES.get("login_flag") == "":
         return HttpResponse("Success!")
     else:
         return HttpResponse("Failed!")

这样就可以验证登录与否然后对应返回了。但是有没有觉得每个视图函数都这么写，类似了。那么我们可以自定义装饰器。

一、装饰器的写法：

logincheck/logincheck.py

 from django.http import HttpResponseRedirect
 def login_need(func):
     def in_func(request):
         if request.COOKIES.get("login_name") not in ["",None," "]:
             if request.COOKIES.get("login_flag") == "":
                 return func(request)
             else:
                 print request.COOKIES
                 return HttpResponseRedirect("/myapp1/login")
         else:
             print request.COOKIES
             return HttpResponseRedirect("/myapp1/login")
     return in_func

然后在myapp1/views.py中使用

 @logincheck.login_need
 def main(request):
     return render(request,"welcome.html")

就可以了，后面的视图都是这么写会省很多事情。

二、突然发现我这个视图这么写，会导致一个问题越权：

[01/Feb/2018 17:08:54]"GET /myapp1/mainpage?name=chenran HTTP/1.1" 301 0
[01/Feb/2018 17:08:54]"GET /myapp1/mainpage/?name=chenran HTTP/1.1" 200 130
[01/Feb/2018 17:09:06]"GET /myapp1/mainpage/?name=chenran01 HTTP/1.1" 200 130
[01/Feb/2018 17:09:13]"GET /myapp1/mainpage/?name=chenranxss HTTP/1.1" 200 130

可想而知，我登录的时候都是一个账号，我返回的信息是确实跟随着那么参数走的，这样是有问题的。

视图函数应该这么写才对：

 @logincheck.login_need
 def main(request):
     if request.COOKIES["login_name"] == request.GET["name"]:
         return render(request,"welcome.html")
     else:
         return render(request,"error.html")

这次没想到自己还写了个越权，真的是一步不小心都不行。