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