27 lines
521 B
Python
27 lines
521 B
Python
from flask_jwt_extended import decode_token
|
|
from user_managment.models import User
|
|
from auth.jwt_controller import jwt
|
|
from auth import login
|
|
|
|
|
|
@login.request_loader
|
|
def load_person_from_request(request):
|
|
try:
|
|
token = request.json['access_token']
|
|
except Exception:
|
|
return None
|
|
|
|
data = decode_token(token)
|
|
user = User.objects(user_name=data['sub']).first()
|
|
|
|
if user:
|
|
return user
|
|
return None
|
|
|
|
|
|
def init_app(app):
|
|
jwt.init_app(app)
|
|
login.init_app(app)
|
|
return app
|
|
|