学习笔记：Flask token验证代码

本次文章为大家介绍基于token验证的注册及登录功能的实现！闲言少叙直接正题

1.首先先看一下数据库设计如下：

#pip install passlib 先安装模块
from passlib.apps import custom_app_context as pwd_context

class User(db.Model):
    __tablename__ = 'shop_list'
    userName = db.Column(db.BigInteger,primary_key = True)
    passWord = db.Column(db.Text,nullable=False)
    def hash_password(self, password): #密码加密方法
        self.passWord = pwd_context.encrypt(password)

    def verify_password(self, password): #验证密码方法
        return pwd_context.verify(password, self.passWord)

我们看到有两个方法 hash_password 和 verify_password；分别是密码加密方法和验证密码的方法。

 2.注册功能实现

@app.route('/api/v1/admin/register',methods=['POST'])
def register():
    username = request.form.get('username') 
    password = request.form.get('password')
    save = User(userName=username)
    save.hash_password(password)
    db.session.add(save)
    db.session.commit()
    return jsonify({'code':200,'msg':'注册成功'})

这个很简单了，首先接收用户名密码 然后保存到数据库即可，save = Shop_list(userName=username)是调用密码加密方法，最后一起commit保存到数据库

3.登录功能实现

@app.route('/api/v1/admin/login',methods=['POST'])
def login():
    username = request.form.get('username')
    password = request.form.get('password')
    obj = User.query.filter_by(userName=username).first()
    if not obj:
        return res_json(201,'','未找到该用户')
    if obj.verify_password(password):
        token = generate_token(username)
        return res_json(200,{'token':token},'登录成功')
    else:
        return res_json(201,'','密码错误')

 解释一下：res_json是我自己封装的返回json的函数