FitTrackee/fittrackee/users/utils_token.py

34 lines
895 B
Python
Raw Normal View History

from datetime import datetime, timedelta
import jwt
from flask import current_app
def get_user_token(user_id, password_reset=False):
expiration_days = (
0
if password_reset
else current_app.config.get('TOKEN_EXPIRATION_DAYS')
)
expiration_seconds = (
current_app.config.get('PASSWORD_TOKEN_EXPIRATION_SECONDS')
if password_reset
else current_app.config.get('TOKEN_EXPIRATION_SECONDS')
)
payload = {
'exp': datetime.utcnow()
+ timedelta(days=expiration_days, seconds=expiration_seconds),
'iat': datetime.utcnow(),
'sub': user_id,
}
return jwt.encode(
2020-09-16 11:09:32 +02:00
payload,
current_app.config.get('SECRET_KEY'),
algorithm='HS256',
)
def decode_user_token(auth_token):
payload = jwt.decode(auth_token, current_app.config.get('SECRET_KEY'))
return payload['sub']