From a8ddaf16fce0d13cea366e6282393c1375a53b17 Mon Sep 17 00:00:00 2001 From: Ravera Thibault Date: Mon, 22 Jan 2018 13:30:09 -0800 Subject: [PATCH] Use get_by_natural_key in jwt_get_secret_key --- rest_framework_jwt/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/rest_framework_jwt/utils.py b/rest_framework_jwt/utils.py index c72197bc..813c597f 100644 --- a/rest_framework_jwt/utils.py +++ b/rest_framework_jwt/utils.py @@ -23,7 +23,9 @@ def jwt_get_secret_key(payload=None): """ if api_settings.JWT_GET_USER_SECRET_KEY: User = get_user_model() # noqa: N806 - user = User.objects.get(pk=payload.get('user_id')) + username_field = get_username_field() + username = payload.get(username_field) + user = User.objects.get_by_natural_key(username) key = str(api_settings.JWT_GET_USER_SECRET_KEY(user)) return key return api_settings.JWT_SECRET_KEY