這篇文章給大家介紹怎么在flask中實(shí)現(xiàn)一個(gè)token機(jī)制，內(nèi)容非常詳細(xì)，感興趣的小伙伴們可以參考借鑒，希望對(duì)大家能有所幫助。

歷下網(wǎng)站制作公司哪家好，找創(chuàng)新互聯(lián)公司！從網(wǎng)頁(yè)設(shè)計(jì)、網(wǎng)站建設(shè)、微信開(kāi)發(fā)、APP開(kāi)發(fā)、響應(yīng)式網(wǎng)站設(shè)計(jì)等網(wǎng)站項(xiàng)目制作，到程序開(kāi)發(fā)，運(yùn)營(yíng)維護(hù)。創(chuàng)新互聯(lián)公司從2013年成立到現(xiàn)在10年的時(shí)間，我們擁有了豐富的建站經(jīng)驗(yàn)和運(yùn)維經(jīng)驗(yàn)，來(lái)保證我們的工作的順利進(jìn)行。專(zhuān)注于網(wǎng)站建設(shè)就選創(chuàng)新互聯(lián)公司。

token 的生成

用token校驗(yàn)身份，是前后端交互的常用方式。

它有以下特性：

• 會(huì)失效

• 加密

• 可以根據(jù)它拿到用戶的信息

生成方式（ 內(nèi)部配置的私鑰+有效期+用戶的id ）

#導(dǎo)入依賴包
from flask import request,jsonify,current_app
from itsdangerous import TimedJSONWebSignatureSerializer as Serializer

def create_token(api_user):
  '''
  生成token
  :param api_user:用戶id
  :return: token
  '''
  
  #第一個(gè)參數(shù)是內(nèi)部的私鑰，這里寫(xiě)在共用的配置信息里了，如果只是測(cè)試可以寫(xiě)死
  #第二個(gè)參數(shù)是有效期(秒)
  s = Serializer(current_app.config["SECRET_KEY"],expires_in=3600)
  #接收用戶id轉(zhuǎn)換與編碼
  token = s.dumps({"id":api_user}).decode("ascii")
  return token

token的校驗(yàn)

校驗(yàn)接收到的token，如果成功返回用戶信息，否則返回None

#基于上面的基礎(chǔ)再導(dǎo)入用戶的模型類(lèi)
from app.model import User

def verify_token(token):
  '''
  校驗(yàn)token
  :param token: 
  :return: 用戶信息 or None
  '''
  
  #參數(shù)為私有秘鑰，跟上面方法的秘鑰保持一致
  s = Serializer(current_app.config["SECRET_KEY"])
  try:
    #轉(zhuǎn)換為字典
    data = s.loads(token)
  except Exception:
    return None
  #拿到轉(zhuǎn)換后的數(shù)據(jù)，根據(jù)模型類(lèi)去數(shù)據(jù)庫(kù)查詢用戶信息
  user = User.query.get(data["id"])
  return user

用裝飾器寫(xiě)一個(gè)必須攜帶token的校驗(yàn)

有很多接口是必須登錄才能操作的，最好的方式就是在寫(xiě)一個(gè)裝飾器，添加在需要的api上

#在上面的基礎(chǔ)上導(dǎo)入
import functools

def login_required(view_func):
  @functools.wraps(view_func)
  def verify_token(*args,**kwargs):
    try:
      #在請(qǐng)求頭上拿到token
      token = request.headers["z-token"]
    except Exception:
      #沒(méi)接收的到token,給前端拋出錯(cuò)誤
      #這里的code推薦寫(xiě)一個(gè)文件統(tǒng)一管理。這里為了看著直觀就先寫(xiě)死了。
      return jsonify(code = 4103,msg = '缺少參數(shù)token')
    
    s = Serializer(current_app.config["SECRET_KEY"])
    try:
      s.loads(token)
    except Exception:
      return jsonify(code = 4101,msg = "登錄已過(guò)期")

    return view_func(*args,**kwargs)

  return verify_token

使用案例

生成token(案例)

#此處的api是藍(lán)圖的對(duì)象
from . import api
from app.model import User,db
from flask import request,jsonify
#導(dǎo)入剛剛寫(xiě)的文件方法
from app.utils.common import create_token,login_required,verify_token

@api.route("/login",methods=["POST"])
def login():
  '''
  用戶登錄
  :return:token
  '''
  res_dir = request.get_json()
  if res_dir is None:
    #這里的code，依然推薦用一個(gè)文件管理狀態(tài)
    return jsonify(code = 4103,msg = "未接收到參數(shù)")
  
  #獲取前端傳過(guò)來(lái)的參數(shù)
  phone = res_dir.get("phone")
  password = res_dir.get("password")
  
  #校驗(yàn)參數(shù)
  if not all([phone,password]):
    return jsonify(code=4103, msg="請(qǐng)?zhí)顚?xiě)手機(jī)號(hào)或密碼")

  if not re.match(r"1[23456789]\d{9}",phone):
    return jsonify(code=4103,msg="手機(jī)號(hào)有誤")

  try:
    user = User.query.filter_by(phone=phone).first()
  except Exception:
    return jsonify(code=4004,msg="獲取信息失敗")

  if user is None or not user.check_password(password):
    return jsonify(code=4103,msg="手機(jī)號(hào)或密碼錯(cuò)誤")
  
  #獲取用戶id，傳入生成token的方法，并接收返回的token
  token = create_token(user.id)
  
  #把token返回給前端
  return jsonify(code=0,msg="成功",data=token)

必須登錄的校驗(yàn)與根據(jù)token拿到用戶信息(案例)

@api.route("/user/detail")
@login_required #必須登錄的裝飾器校驗(yàn)
def userInfo():
  '''
  用戶信息
  :return:data
  '''
  token = request.headers["z-token"]
  #拿到token，去換取用戶信息
  user = verify_token(token)

  data = {
    "phone":user.phone,
    "name":user.name,
    "head_portrait":user.head_portrait,
    "intro":user.intro,
    "level":user.level
  }

  return jsonify(code=0,msg="成功",data=data)

關(guān)于怎么在flask中實(shí)現(xiàn)一個(gè)token機(jī)制就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，可以學(xué)到更多知識(shí)。如果覺(jué)得文章不錯(cuò)，可以把它分享出去讓更多的人看到。

名稱(chēng)欄目：怎么在flask中實(shí)現(xiàn)一個(gè)token機(jī)制-創(chuàng)新互聯(lián)
文章URL：http://weahome.cn/article/despeo.html