Skip to content
Snippets Groups Projects
Commit dd6a5643 authored by Bjarke Madsen's avatar Bjarke Madsen
Browse files

use new-style query

parent 1c6f0b67
Branches
Tags
1 merge request!44Feature/comp 208 google o auth poc
from sqlalchemy import select
from flask import jsonify from flask import jsonify
from datetime import datetime from datetime import datetime
from compendium_v2.db import session_scope from compendium_v2.db import session_scope
...@@ -31,7 +32,7 @@ def fetch_user(profile: dict): ...@@ -31,7 +32,7 @@ def fetch_user(profile: dict):
with session_scope() as session: with session_scope() as session:
sub_id = profile['sub'] sub_id = profile['sub']
user = session.query(User).filter(User.oidc_sub == sub_id).first() user = session.scalar(select(User).where(User.oidc_sub == sub_id))
if user is None: if user is None:
return None return None
user.last_login = datetime.utcnow() user.last_login = datetime.utcnow()
...@@ -47,7 +48,7 @@ def user_loader(user_id: str): ...@@ -47,7 +48,7 @@ def user_loader(user_id: str):
""" """
with session_scope() as session: with session_scope() as session:
user = session.query(User).filter(User.id == user_id).first() user = session.scalar(select(User).where(User.id == user_id))
if user is None: if user is None:
return None return None
user.last_login = datetime.utcnow() user.last_login = datetime.utcnow()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment