from flask_login import LoginManager

from . import DATABASE as db
from .models import User

AUTH = LoginManager()

AUTH.login_view = "root.login"


@AUTH.user_loader
def load_user(id):
    return db.session.get(User, int(id))