TriHard
/
rDrama
forked from rDrama/rDrama
1
0
Fork 0
Code Pull Requests Activity

dfs

master
Aevann1 2021-12-24 05:05:29 +02:00
parent 4e98eb888e
commit 4a0f2177fe
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
files/helpers/wrappers.py
View File

@ -7,7 +7,7 @@ def get_logged_in_user():
token = request.headers.get("Authorization","").strip()
if token:
if g.db: client = g.db.query(ClientAuth).filter(ClientAuth.access_token == token).one_or_none()
if hasattr(g, 'db') and g.db: client = g.db.query(ClientAuth).filter(ClientAuth.access_token == token).one_or_none()
else: return None
if not client: return None
@ -21,7 +21,7 @@ def get_logged_in_user():
nonce = session.get("login_nonce", 0)
try:
if g.db: v = g.db.query(User).filter_by(id=lo_user).one_or_none()
if hasattr(g, 'db') and g.db: v = g.db.query(User).filter_by(id=lo_user).one_or_none()
else: return None
except: return None