aboutsummaryrefslogtreecommitdiff
path: root/webmentions_ssg/auth.py
blob: 7bd93a335a99663ebe61c29b8bcb135fa05c8562 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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: int) -> User | None:
    """
    Load a user from the database by identifier.

    :param id: User identifier stored in the login session.
    :return: Matching user, or ``None`` if no user exists with the identifier.
    """
    return db.session.get(User, int(id))