check for user attr on g before attempting to use it

This commit is contained in:
jasquat 2023-09-19 07:36:10 -04:00
parent ed4eb4cd28
commit 1b18cb88c1
1 changed files with 1 additions and 2 deletions

View File

@ -62,7 +62,6 @@ def verify_token(token: str | None = None, force_run: bool | None = False) -> No
user_model = None user_model = None
if token_info["token"] is not None: if token_info["token"] is not None:
# import pdb; pdb.set_trace()
user_model = _get_user_model_from_token(token_info["token"]) user_model = _get_user_model_from_token(token_info["token"])
elif token_info["api_key"] is not None: elif token_info["api_key"] is not None:
user_model = _get_user_model_from_api_key(token_info["api_key"]) user_model = _get_user_model_from_api_key(token_info["api_key"])
@ -71,7 +70,7 @@ def verify_token(token: str | None = None, force_run: bool | None = False) -> No
g.user = user_model g.user = user_model
# If the user is valid, store the token for this session # If the user is valid, store the token for this session
if g.user: if hasattr(g, "user") and g.user:
if token_info["token"]: if token_info["token"]:
# This is an id token, so we don't have a refresh token yet # This is an id token, so we don't have a refresh token yet
g.token = token_info["token"] g.token = token_info["token"]