From e18feb2f92e4c6e13d0814dafde19347d745ecbb Mon Sep 17 00:00:00 2001 From: ChamRun Date: Wed, 26 Oct 2022 14:56:07 +0330 Subject: [PATCH] check type of `identity` --- flask_jwt/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/flask_jwt/__init__.py b/flask_jwt/__init__.py index f864b78..b343c4d 100644 --- a/flask_jwt/__init__.py +++ b/flask_jwt/__init__.py @@ -50,7 +50,11 @@ def _default_jwt_payload_handler(identity): iat = datetime.utcnow() exp = iat + current_app.config.get('JWT_EXPIRATION_DELTA') nbf = iat + current_app.config.get('JWT_NOT_BEFORE_DELTA') - identity = getattr(identity, 'id') or identity['id'] + if type(identity) is dict: + identity = identity['id'] + else: + identity = getattr(identity, 'id') + return {'exp': exp, 'iat': iat, 'nbf': nbf, 'identity': identity}