ConsulManager/flask-consul/units/token_auth.py

20 lines
616 B
Python
Raw Normal View History

2022-01-28 18:45:54 +00:00
from flask_httpauth import HTTPTokenAuth
2022-04-09 01:35:50 +00:00
from itsdangerous import TimedJSONWebSignatureSerializer
from units import consul_kv
secret_key = consul_kv.get_value('ConsulManager/assets/secret/skey')['sk']
2022-06-04 09:36:18 +00:00
s = TimedJSONWebSignatureSerializer(secret_key,expires_in=28800)
2022-01-28 18:45:54 +00:00
auth = HTTPTokenAuth()
@auth.verify_token
def verify_token(token):
try:
data = s.loads(token)
except Exception as e:
print("【login】认证异常",e,flush=True)
2022-01-28 18:45:54 +00:00
return False
return True
@auth.error_handler
def unauthorized():
return {"code": 50000, "data": f"403认证异常请重新登录"}, 200