From 85607b39dd108487a438ea5497ba129f0d4726a2 Mon Sep 17 00:00:00 2001 From: Mahaffey Date: Fri, 1 Jun 2018 12:02:09 -0400 Subject: [PATCH] added context to each serializer in get_response_data in registration/views so HyperLinkedIdentityFields for user return would not throw error --- rest_auth/registration/views.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/rest_auth/registration/views.py b/rest_auth/registration/views.py index 0e0ab0d..7522dcd 100644 --- a/rest_auth/registration/views.py +++ b/rest_auth/registration/views.py @@ -46,6 +46,10 @@ class RegisterView(CreateAPIView): return super(RegisterView, self).dispatch(*args, **kwargs) def get_response_data(self, user): + """ + Here I added context to the return of each serializer so + HyperlinkedIdentityFields would work for the user return + """ if allauth_settings.EMAIL_VERIFICATION == \ allauth_settings.EmailVerificationMethod.MANDATORY: return {"detail": _("Verification e-mail sent.")} @@ -55,9 +59,13 @@ class RegisterView(CreateAPIView): 'user': user, 'token': self.token } - return JWTSerializer(data).data + return JWTSerializer(data, + context={'request': self.request} + ).data else: - return TokenSerializer(user.auth_token).data + return TokenSerializer(user.auth_token, + context={'request': self.request} + ).data def create(self, request, *args, **kwargs): serializer = self.get_serializer(data=request.data)