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