flask 实现token机制的示例代码

#导入依赖包from flask import request,jsonify,current_appfrom itsdangerous import TimedJSONWebSignatureSerializer as Serializerdef create_token(api_user):  '''  生成token  :param api_user:用户id  :return: token  '''    #第一个参数是内部的私钥，这里写在共用的配置信息里了，如果只是测试可以写死  #第二个参数是有效期(秒)  s = Serializer(current_app.config["SECRET_KEY"],expires_in=3600)  #接收用户id转换与编码  token = s.dumps({"id":api_user}).decode("ascii")  return token

#基于上面的基础再导入用户的模型类from app.model import Userdef verify_token(token):  '''  校验token  :param token:   :return: 用户信息 or None  '''    #参数为私有秘钥，跟上面方法的秘钥保持一致  s = Serializer(current_app.config["SECRET_KEY"])  try:    #转换为字典    data = s.loads(token)  except Exception:    return None  #拿到转换后的数据，根据模型类去数据库查询用户信息  user = User.query.get(data["id"])  return user

#在上面的基础上导入import functoolsdef login_required(view_func):  @functools.wraps(view_func)  def verify_token(*args,**kwargs):    try:      #在请求头上拿到token      token = request.headers["z-token"]    except Exception:      #没接收的到token,给前端抛出错误      #这里的code推荐写一个文件统一管理。这里为了看着直观就先写死了。      return jsonify(code = 4103,msg = '缺少参数token')        s = Serializer(current_app.config["SECRET_KEY"])    try:      s.loads(token)    except Exception:      return jsonify(code = 4101,msg = "登录已过期")    return view_func(*args,**kwargs)  return verify_token

#此处的api是蓝图的对象from . import apifrom app.model import User,dbfrom flask import request,jsonify#导入刚刚写的文件方法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，依然推荐用一个文件管理状态    return jsonify(code = 4103,msg = "未接收到参数")    #获取前端传过来的参数  phone = res_dir.get("phone")  password = res_dir.get("password")    #校验参数  if not all([phone,password]):    return jsonify(code=4103, msg="请填写手机号或密码")  if not re.match(r"1[23456789]/d{9}",phone):    return jsonify(code=4103,msg="手机号有误")  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="手机号或密码错误")    #获取用户id，传入生成token的方法，并接收返回的token  token = create_token(user.id)    #把token返回给前端  return jsonify(code=0,msg="成功",data=token)

@api.route("/user/detail")@login_required #必须登录的装饰器校验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)