61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
#coding=utf-8
|
||
from ast import If
|
||
from typing import Any
|
||
import jwt
|
||
from jwt.exceptions import InvalidTokenError
|
||
from passlib.context import CryptContext
|
||
from datetime import datetime, timedelta,timezone
|
||
from src.utils.config import read_config
|
||
|
||
# 密钥，用于签名和验证JWT
|
||
SECRET_KEY = 'fastapi_longyu12'
|
||
# 令牌过期时间 默认十天
|
||
EXP = datetime.now(timezone.utc)+timedelta(days=1)
|
||
project = read_config("project")
|
||
if project.get("tokenTIme"):
|
||
EXP = datetime.now(timezone.utc)+ timedelta(days=project.get("tokenTIme"))
|
||
|
||
"""
|
||
生成 token 令牌
|
||
@payload: Any 存在token里面的信息
|
||
"""
|
||
def createToken(payload: Any):
|
||
payload["exp"] = EXP
|
||
|
||
token = jwt.encode(payload, SECRET_KEY, algorithm='HS256' )
|
||
|
||
return token
|
||
|
||
def vertify_is_login(token: str):
|
||
try:
|
||
payload = jwt.decode(token, SECRET_KEY, algorithms=["HS256"], verify=False)
|
||
print(int(payload.get("exp")) , int((datetime.now(timezone.utc)).timestamp()))
|
||
if int(payload.get("exp")) < int((datetime.now(timezone.utc)).timestamp()):
|
||
print("登录已过期")
|
||
return False
|
||
else:
|
||
return True
|
||
except jwt.exceptions.ExpiredSignatureError:
|
||
print("登录已过期")
|
||
return False
|
||
except (jwt.exceptions.InvalidSignatureError, jwt.exceptions.DecodeError):
|
||
print("无效凭证")
|
||
return False
|
||
|
||
|
||
# def getTokenInfo():
|
||
|
||
|
||
|
||
|
||
# 载荷，即JWT中包含的信息
|
||
# payload = {
|
||
# 'user_id': 123,
|
||
# 'username': 'john_doe',
|
||
# 'exp': datetime.datetime.utcnow() + datetime.timedelta(days=1)
|
||
# }
|
||
|
||
# # 生成JWT
|
||
# token = jwt.encode(payload, secret_key, algorithm='HS256')
|
||
# print('Generated JWT:', token)
|
||
|