webmentions_ssg/auth.py


1
2
3
4
5
6
7
8
9
10
11
12
13

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))