flask 登录装饰器

 

登录装饰器   首先要自己设置cookies 

# 登录装饰器
def login_required(fn):
    def inner(*args, **kwargs):

        user_id = request.cookies.get('user_id', None)
        if user_id:
            # 查询用户
            user = User.query.get(int(user_id))
            fn(*args, **kwargs)

        else:
            # 没有登录
            return redirect('/login')

    return inner