From c46f75e5150f735240e888748f79995ed8e285ae Mon Sep 17 00:00:00 2001 From: Herald Yu Date: Wed, 15 Aug 2018 09:20:41 +0800 Subject: [PATCH] Fixed AttributeError: 'bytes' object has no attribute 'encode' --- flask_user/token_manager.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/flask_user/token_manager.py b/flask_user/token_manager.py index fee77e55..2d162b60 100644 --- a/flask_user/token_manager.py +++ b/flask_user/token_manager.py @@ -7,6 +7,7 @@ import base64 import string +import sys # Non-system imports are moved into the methods to make them an optional requirement @@ -43,7 +44,10 @@ def __init__(self, app): raise ConfigError('Config setting SECRET_KEY is missing.') # Print a warning if SECRET_KEY is too short - key = flask_secret_key.encode() + if sys.version_info[0] < 3: + key = flask_secret_key.encode() + else: + key = flask_secret_key if len(key)<32: print('WARNING: Flask-User TokenManager: SECRET_KEY is shorter than 32 bytes.') key = key + b' '*32 # Make sure the key is at least 32 bytes long