flask 实现token机制的示例代码

时间:2021-05-22

token 的生成

用token校验身份,是前后端交互的常用方式。

它有以下特性:

  • 会失效
  • 加密
  • 可以根据它拿到用户的信息

生成方式( 内部配置的私钥+有效期+用户的id )

#导入依赖包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

token的校验

校验接收到的token,如果成功返回用户信息,否则返回None

#基于上面的基础再导入用户的模型类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

用装饰器写一个必须携带token的校验

有很多接口是必须登录才能操作的,最好的方式就是在写一个装饰器,添加在需要的api上

#在上面的基础上导入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

使用案例

生成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)

必须登录的校验与根据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)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章