JWT令牌的生成和解令

import base64 import copy import hmac import json import time class Jwt(): def __init__(self): pass @staticmethod def b64encode(j_s): #将加密后的等号用空字符进行替换 return base64.urlsafe_b64encode(j_s).replace(b'=',b'') @staticmethod def b64decode(b_s): #不全被替换掉的等号 为后面的提取时间做准备 rem=len(b_s)%4 if rem>0: b_s += b'='*(4-rem) #补全等号 #将中间的部分进行解码并返回 return base64.urlsafe_b64decode(b_s) @staticmethod #生成令牌 def encode(payload,key,exp=300): #对header进行base64转码 header ={'alg':'Hs256','ytp':'Jwt'} #给出header #separators：分隔符，实际上是(item_separator, dict_separator)的一个元组，默认的就是(‘,’,’:’)；这表示dictionary内keys之间用“,”隔开，而KEY和value之间用“：”隔开。sort_keys是告诉编码器按照字典key排序(a到z)输出。 #生成header的json串 header_json=json.dumps(header,sort_keys=True,separators=(',',':')) #生成base64的header header_bs=Jwt.b64encode(header_json.encode()) #对payload进行base64转码 my_payload=copy.deepcopy(payload) #给出payload my_payload['exp']=time.time()+int(exp) #生成payload 的json串 payload_json=json.dumps(my_payload,sort_keys=True,separators=(',',':')) #生成payload进行base64转码 payload_bs=Jwt.b64encode(payload_json.encode()) #对sign进行base64转码 sign=[header_bs+payload_bs] if isinstance(key,str): # 判断key类型,为str 则转化为byte key=key.encode() hm=hmac.new(key,header_bs + b'.' +payload_bs, digestmod='SHA256') # 获取签名结果 hm_bs=Jwt.b64encode(hm.digest()) return header_bs + b'.' + payload_bs + b'.' +hm_bs @staticmethod def decode(jwt_s,key): #解令 #校验token #1，检查签名【前两项bs 再做一次hmac签名，与第三部分进行比较。若两者相等，校验成功；否则，失败】 #2， 检查时间戳是否过期【过期则 raise（）】 #3， 返回payload铭文，也就是payload字典对象 header_bs,payload_bs,sign_bs=jwt_s.split(b'.') if isinstance(key,str): key=key.encode() hm=hmac.new(key,header_bs + b'.' + payload_bs, digestmod='SHA256') new_sign_bs=Jwt.b64encode(hm.digest()) if new_sign_bs != sign_bs: raise('不正确') #对payload解码（解令） payload_json = Jwt.b64decode(payload_bs) #json字符串 -> python对象 payload = json.loads(payload_json) #时间比较 exp = payload['exp'] now_t=time.time() if now_t > exp: raise ('已经过期') return payload if __name__ == '__main__': s=Jwt.encode({'username':'xiaoerlang'},'123456',100) print(s) d=Jwt.decode(s,'123456') print(d)

我自己使用时的结果：

b'eyJhbGciOiJIczI1NiIsInl0cCI6Ikp3dCJ9.eyJleHAiOjE1NzI1MjM2MDcuNDY5NjU5MywidXNlcm5hbWUiOiJ4aWFvZXJsYW5nIn0.0RDRL-Jgb7-q7UO46LT6W5DG3l-DfqSZ4k15hLqfdrM' {'exp': 1572523607.4696593, 'username': 'xiaoerlang'}

以上结果仅凭参考，由于时间的不同生成的结果也不同。为了进一步验证您的结果是否正确，可以使用JWT官网jtw官网进行结果验证。