學習筆記：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的函數