Improved try-excepts
This commit is contained in:
parent
cc131a894b
commit
c338e8f959
@ -17,5 +17,5 @@ def verify_token(token):
|
|||||||
try:
|
try:
|
||||||
jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])
|
jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])
|
||||||
return True
|
return True
|
||||||
except:
|
except jwt.PyJWTError:
|
||||||
return False
|
return False
|
||||||
|
@ -32,7 +32,7 @@ def extract_token_data(token):
|
|||||||
if token is not None:
|
if token is not None:
|
||||||
try:
|
try:
|
||||||
return jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])
|
return jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])
|
||||||
except:
|
except jwt.PyJWTError:
|
||||||
return None
|
return None
|
||||||
else:
|
else:
|
||||||
return None
|
return None
|
||||||
@ -52,13 +52,13 @@ def get_email_from_token_data():
|
|||||||
return email
|
return email
|
||||||
else:
|
else:
|
||||||
return None
|
return None
|
||||||
except:
|
except jwt.PyJWTError:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
else: # "Normal" token, extract username from token
|
else: # "Normal" token, extract username from token
|
||||||
try:
|
try:
|
||||||
return jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])['email']
|
return jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])['email']
|
||||||
except:
|
except jwt.PyJWTError:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
return None
|
return None
|
||||||
|
Loading…
Reference in New Issue
Block a user