django学习 笔记---添加登录装饰器(FBV和CBV模式)

接上一篇博文的例子

dispatch添加装饰器原理

注:该装饰器对ajax的post提交没用,ajax直接在view中判断有无登录,让js去登录

FBV模式

def cookie(func):
    def deco(request,*args,**kwargs):
        # FBV 不太清楚为什么要写两次u,可能是内部吧?如果是正常的装饰器应该一次就够了
        u = request.get_signed_cookie('username', salt='user', default=None)
        if not u:
            return render(request,'login.html')
        return func(request,*args,**kwargs)
    return deco

@cookie
def houtai(request):
    u = request.get_signed_cookie('username',salt='user',default=None)
    return render(request,'houtai.html',{'user':u})
CBV模式
from django.utils.decorators import method_decorator
from django import views

# @method_decorator(cookie,name='dispatch')    # dispatch的便捷写法
class CBVtest(views.View):

    @method_decorator(cookie)   # 给dispatch方法添加装饰器,那么下面所有的get,post都会添加
    def dispatch(self, request, *args, **kwargs):
        return super(CBVtest,self).dispatch(request,*args,**kwargs)

    # @method_decorator(cookie)  # 单独添加
    def get(self,request):
        u = request.get_signed_cookie('username',salt='user',default=None)
        return render(request,'houtai.html',{'user':u})

    def post(self,request):
        return HttpResponse('post ok')


你可能感兴趣的:(Django学习笔记)