28 lines
652 B
Python
28 lines
652 B
Python
from flask import redirect
|
|
from flask_jwt_extended import JWTManager
|
|
|
|
from pear_admin.orms.user import UserORM
|
|
|
|
jwt = JWTManager()
|
|
|
|
|
|
@jwt.user_identity_loader
|
|
def user_identity_lookup(user):
|
|
return user.id
|
|
|
|
|
|
@jwt.user_lookup_loader
|
|
def user_lookup_callback(_jwt_header, jwt_data):
|
|
identity = jwt_data["sub"]
|
|
return UserORM.query.filter(UserORM.id == identity).one_or_none()
|
|
|
|
|
|
@jwt.expired_token_loader
|
|
def expired_token_callback():
|
|
return {"msg": "token 已过期，请重新登录", "code": -1}, 403
|
|
|
|
|
|
@jwt.unauthorized_loader
|
|
def missing_token_callback(error):
|
|
return {"msg": "操作未授权，请重新登录", "code": -1}, 403
|