You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
736 B
29 lines
736 B
import peewee
|
|
from quart import session, redirect, url_for
|
|
|
|
from yellow.factory import openid
|
|
from yellow.models import User
|
|
|
|
|
|
@openid.after_token()
|
|
async def handle_user_login(resp: dict):
|
|
access_token = resp["access_token"]
|
|
openid.verify_token(access_token)
|
|
|
|
user = await openid.user_info(access_token)
|
|
username = user['preferred_username']
|
|
uid = user['sub']
|
|
|
|
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.root'))
|