【Flask项目2】生成token和验证token(11)

导读:本篇文章讲解 【Flask项目2】生成token和验证token(11),希望对大家有帮助,欢迎收藏,转发!站点地址:www.bmabk.com

用户登录

用户登录的功能我们使用 Json Web Token(JWT) 来为登录用户生成token令牌,并返回给客户端,客户端有了令牌之后,每次访问服务器,都携带Token,后台经过验证就能知道当前登录的用户是谁,以实现鉴权,权限控制。
在这里插入图片描述

comment——utils——token_pyjwt.py
生成token和验证token

import jwt
from comment.utils import const
from datetime import datetime,timedelta
from jwt import PyJWTError
from flask import current_app
from comment.models.user import User

def generate_tokens(uid):
	'''
	一个用户在一次会话生成一个token
	:param uid: 用户id
    :return:
	'''
	#params:是生成token的参数
	params={
		'id':uid,
		#exp:代表token的有效时间,datetime.utcnow():代表当前时间
        #timedelta:表示转化为毫秒
		'exp':datetime.utcnow()+timedelta(seconds=const.JWT_EXPIRY_SECOND)
	}
	#key:密钥,
    #algorithm:算法,算法是SHA-256
    #SHA-256:密码散列函数算法.256字节长的哈希值(32个长度的数组)---》16进制字符串表示,长度为64。信息摘要,不可以逆
	return jwt.encode(payload=params,key=const.SECRET_KEY,algorithm='HS256')


def verify_tokens(token_str):
	'''
	验证token
	:param token_str:如果验证成功返回用户id
    :return:
	'''
	try:
		#返回之前生成token的时候的字典,字典种包含id和exp
		data=jwt.decode(token_str,key=const.SECRET_KEY,algorithms='HS256')
        current_app.logger.info(data)
        user=User.query.filter(User.id==data['id']).first()
        if user and user.onlock==0:		#如果用户存在,并且没有锁定
        	return {'id':user.id}
        else:
        	return {"message":"数据库中不存在当前用户,或者用户已经过期"}
 	    except PyJWTError as e:
        current_app.logger.error(e)
        return {"message":"token验证失败"}

comment——utils——const.py
定义常量

SECRET_KEY=os.urandom(16)   #生成一个随机数作为密钥
JWT_EXPIRY_SECOND=60*60 #TOKENDE 有效时间,一个小时

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/74117.html

(0)
小半的头像小半

相关推荐

极客之音——专业性很强的中文编程技术网站,欢迎收藏到浏览器,订阅我们!