diff --git a/rest_framework_jwt/utils.py b/rest_framework_jwt/utils.py index c800ba66..72687a7f 100644 --- a/rest_framework_jwt/utils.py +++ b/rest_framework_jwt/utils.py @@ -56,7 +56,7 @@ def jwt_decode_handler(token): ) -def jwt_response_payload_handler(token, user=None): +def jwt_response_payload_handler(token, user=None, request=None): """ Returns the response data for both the login and refresh views. Override to return a custom response such as including the diff --git a/rest_framework_jwt/views.py b/rest_framework_jwt/views.py index 66ea4dca..358143a9 100644 --- a/rest_framework_jwt/views.py +++ b/rest_framework_jwt/views.py @@ -30,7 +30,7 @@ def post(self, request): if serializer.is_valid(): user = serializer.object.get('user') or request.user token = serializer.object.get('token') - response_data = jwt_response_payload_handler(token, user) + response_data = jwt_response_payload_handler(token, user, request) return Response(response_data) @@ -58,7 +58,7 @@ def post(self, request): if serializer.is_valid(): user = serializer.object.get('user') or request.user token = serializer.object.get('token') - response_data = jwt_response_payload_handler(token, user) + response_data = jwt_response_payload_handler(token, user, request) return Response(response_data) diff --git a/tests/utils.py b/tests/utils.py index 6950c80d..4f6c3dba 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,4 +1,4 @@ -def jwt_response_payload_handler(token, user=None): +def jwt_response_payload_handler(token, user=None, request=None): """ Returns the response data for both the login and refresh views. Override to return a custom response such as including the