YellWOWPages/yellow/auth.py

import re

import peewee
from quart import session, redirect, url_for
from quart_keycloak import Keycloak, KeycloakAuthToken, KeycloakLogoutRequest
from yellow.factory import keycloak
from yellow.models import User


@keycloak.after_login()
async def handle_user_login(auth_token: KeycloakAuthToken):
    username = auth_token.username
    if not re.match(r"^[a-zA-Z0-9_\.-]+$", username):
        raise Exception("bad username")

    try:
        user = User.select().where(User.id == uid).get()
    except peewee.DoesNotExist:
        user = None

    if not user:
        # create new user if it does not exist yet
        user = User.create(id=uid, username=username)

    # user is now logged in
    session['user'] = user.to_json()
    return redirect(url_for('bp_routes.dashboard'))