flask自己写登录验证和角色验证-Toy模板网

这篇具有很好参考价值的文章主要介绍了flask自己写登录验证和角色验证。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

python flask程序需要用户登录也需要角色验证，可以采用如下方法

def is_login(func):
    @wraps(func)
    def inner(*args,**kwargs):
        user = session.get('user')
        if not user:
            return redirect(url_for('login'))
        return func(*args,**kwargs)
    return inner

def roles_required(*args1):
    def outer(func):
        @wraps(*args1)
        def inner(*args,**kwargs):
            user = session.get('user')
            if not user:
                return redirect(url_for('login'))
            else:
                sql="select role from users where username='{}'".format(user)
                flag=len(mydb.get_all(sql))
                if flag:
                    result=mydb.get_all(sql)
                    if result[0]['role'] in args1:
                        print(f"{result[0]['role']}在{args1}中")
                        return func(*args, **kwargs)
                    else:
                        print('用户权限不足')
                        return redirect(url_for('index'))
                else:
                    print('用户没有登录')
                    return redirect(url_for('login'))
        return inner
    return outer

函数代码

@app.route('/userjson',methods=['GET','POST'],endpoint='userjson')
@roles_required('admin')
def userjson():
    sql='select * from users'
    data=mydb.get_all(sql)
    info = request.values
    print(info)
    limit = info.get('limit', 10)  # 每页显示的条数
    offset = info.get('offset', 0)  # 分片数，(页码-1)*limit，它表示一段数据的起点
    username = info.get('uname','')
    if  username:
        sql="select * from users where username='{}'".format(username)
        print(sql)
        data=mydb.get_all(sql)
    return jsonify({'total': len(data), 'rows': data[int(offset):(int(offset) + int(limit))]})