django 自定义登录验证装饰器,使用as_view()进行验证

废话不多说直接上代码

from django.views.generic import View # 导入视图类


# 自定义的登录验证装饰器
def login(func):
    def warpper(request, *args, **kwargs):
        if request.session.get('is_login'):
            return func(request, *args, **kwargs)
        else:
            return redirect(reverse('user:login')) # 没有通过验证返回登录页
    return warpper

# django使用类进行路由时,路由最终都会走自带的as_view()方法
# 所以重写 as_view方法,后续视图类需要登录验证的我们就继承LoginView
class LoginView(View):
    @classmethod
    def as_view(cls, **kwargs):
    	return login(super().as_view())  # 走我们自定义的登录验证函数



自定义装饰器,如果每个视图都要进行装饰台麻烦了,直接使用这种方法,需要验证的,我们继承下就行了,简单方便

你可能感兴趣的:(python)