Use get_by_natural_key in jwt_get_secret_key

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