aboutsummaryrefslogtreecommitdiffstats
path: root/src/expense/auth/blueprint.py
blob: 4f36c1205556a98f495cb843ec35baceb78a7fb5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from typing import Optional
from flask import Blueprint

from ..model import User
from .. import loginManager, db

auth = Blueprint("auth", __name__, url_prefix="/auth")


@loginManager.user_loader
def load_user(id: str) -> Optional[User]:
    """
    Callback to load user from id which is a number
    """
    if id is not None:
        return db.session.scalars(db.select(User).where(User.id == id)).one_or_none()
    return None


from . import login, logout, register, delete