flask 实现token机制的示例代码
token的生成
用token校验身份，是前后端交互的常用方式。
它有以下特性：
- 会失效
- 加密
- 可以根据它拿到用户的信息
生成方式（内部配置的私钥+有效期+用户的id）
#导入依赖包
fromflaskimportrequest,jsonify,current_app
fromitsdangerousimportTimedJSONWebSignatureSerializerasSerializer
defcreate_token(api_user):
'''
生成token
:paramapi_user:用户id
:return:token
'''
#第一个参数是内部的私钥，这里写在共用的配置信息里了，如果只是测试可以写死
#第二个参数是有效期(秒)
s=Serializer(current_app.config["SECRET_KEY"],expires_in=3600)
#接收用户id转换与编码
token=s.dumps({"id":api_user}).decode("ascii")
returntoken
token的校验
校验接收到的token，如果成功返回用户信息，否则返回None
#基于上面的基础再导入用户的模型类 fromapp.modelimportUser defverify_token(token): ''' 校验token :paramtoken: :return:用户信息orNone ''' #参数为私有秘钥，跟上面方法的秘钥保持一致 s=Serializer(current_app.config["SECRET_KEY"]) try: #转换为字典 data=s.loads(token) exceptException: returnNone #拿到转换后的数据，根据模型类去数据库查询用户信息 user=User.query.get(data["id"]) returnuser
用装饰器写一个必须携带token的校验
有很多接口是必须登录才能操作的，最好的方式就是在写一个装饰器，添加在需要的api上
#在上面的基础上导入 importfunctools deflogin_required(view_func): @functools.wraps(view_func) defverify_token(*args,**kwargs): try: #在请求头上拿到token token=request.headers["z-token"] exceptException: #没接收的到token,给前端抛出错误 #这里的code推荐写一个文件统一管理。这里为了看着直观就先写死了。 returnjsonify(code=4103,msg='缺少参数token') s=Serializer(current_app.config["SECRET_KEY"]) try: s.loads(token) exceptException: returnjsonify(code=4101,msg="登录已过期") returnview_func(*args,**kwargs) returnverify_token
使用案例
生成token(案例)
#此处的api是蓝图的对象
from.importapi
fromapp.modelimportUser,db
fromflaskimportrequest,jsonify
#导入刚刚写的文件方法
fromapp.utils.commonimportcreate_token,login_required,verify_token
@api.route("/login",methods=["POST"])
deflogin():
'''
用户登录
:return:token
'''
res_dir=request.get_json()
ifres_dirisNone:
#这里的code，依然推荐用一个文件管理状态
returnjsonify(code=4103,msg="未接收到参数")
#获取前端传过来的参数
phone=res_dir.get("phone")
password=res_dir.get("password")
#校验参数
ifnotall([phone,password]):
returnjsonify(code=4103,msg="请填写手机号或密码")
ifnotre.match(r"1[23456789]\d{9}",phone):
returnjsonify(code=4103,msg="手机号有误")
try:
user=User.query.filter_by(phone=phone).first()
exceptException:
returnjsonify(code=4004,msg="获取信息失败")
ifuserisNoneornotuser.check_password(password):
returnjsonify(code=4103,msg="手机号或密码错误")
#获取用户id，传入生成token的方法，并接收返回的token
token=create_token(user.id)
#把token返回给前端
returnjsonify(code=0,msg="成功",data=token)
必须登录的校验与根据token拿到用户信息(案例)
@api.route("/user/detail")
@login_required#必须登录的装饰器校验
defuserInfo():
'''
用户信息
: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
}
returnjsonify(code=0,msg="成功",data=data)
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。