17 lines
351 B
Python
17 lines
351 B
Python
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_by(id=identity).one_or_none()
|