python，自定义token生成

1、使用的包PyJWT来实现token生成

安装：pip install PyJWT==2.8.0

2、使用例子：

import jwt
import time'''
pip install pyJWT==2.8.0
'''SECRET_KEY = '%^ES*E&Ryurehuie9*7^%$#$EDFGHUYTRE#$%^&%$#@#$RTYGHIK'
DEFAULT_EXP = 7 * 24 * 60def create_token(user_id: int, timeout=None):''':param user_id: 传递用户的id:param timeout: token有效时间，默认是一天:return:'''if timeout == None:# 没有指定过期时间，就使用配置的时间戳timeout = DEFAULT_EXPelif type(timeout) != int:# 传递的类型有误，使用配置中的时间戳timeout = DEFAULT_EXP# 构造headerheaders = {'type': 'jwt','alg': 'HS256'}# 缓存数据payload = {'user_id': user_id,'exp': time.time() + timeout}token = jwt.encode(payload=payload, key=SECRET_KEY, algorithm="HS256", headers=headers)if isinstance(token, bytes):token = token.decode('utf-8')return token# 给websocket发起连接时，验证token
def decode_token(token: str):# 加密用的盐try:payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])return (payload.get('user_id'), payload.get('exp'))except jwt.ExpiredSignatureError:return Noneif __name__ == '__main__':token = create_token(1)print(token)ret = decode_token(token)print(ret)

3、在脚本中也可以使用