统一身份认证系统




pip install pyjwt
]]>
import jwt
import datetime
def generate_token(user_id):
payload = {
'user_id': user_id,
'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1)
}
token = jwt.encode(payload, 'SECRET_KEY', algorithm='HS256')
return token
]]>
def verify_token(token):
try:
payload = jwt.decode(token, 'SECRET_KEY', algorithms=['HS256'])
return payload
except jwt.ExpiredSignatureError:
return "Token expired"
except jwt.InvalidTokenError:
return "Invalid token"
]]>