用户登录
用户登录的功能我们使用 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