Merge pull request #10 from JuanBenitez97/custom-login-response

Custom login response

Author: Juan Benitez

api_crud/urls.py

@@ -1,13 +1,14 @@
 
 from django.contrib import admin
 from django.conf.urls import include, url
-from .views import RegisterView
+from .views import RegisterView, CustomLoginView
 
 
 # urls
 urlpatterns = [
     url(r'^', include('movies.urls')),
-    url(r'^rest-auth/', include('rest_auth.urls')),
+    url(r'^rest-auth/login/', CustomLoginView.as_view()),
     url(r'^rest-auth/registration/', RegisterView.as_view()),
+    url(r'^rest-auth/', include('rest_auth.urls')),
     url(r'^admin/', admin.site.urls),
 ]

api_crud/views.py

@@ -6,7 +6,9 @@
 from allauth.account.utils import complete_signup
 from allauth.account import app_settings as allauth_settings
 from rest_auth.models import TokenModel
+from rest_auth.views import LoginView
 from rest_auth.registration.app_settings import RegisterSerializer, register_permission_classes
+from django.contrib.auth.models import User
 
 sensitive_post_parameters_m = method_decorator(
     sensitive_post_parameters('password1', 'password2')
@@ -39,3 +41,17 @@ def perform_create(self, serializer):
 
         complete_signup(self.request._request, user, None, None)
         return user
+
+
+class CustomLoginView(LoginView):
+    
+    def get_response(self):
+        orginal_response = super().get_response()
+
+        custom_response = {"user": {
+            "username": self.user.username,
+            "email": self.user.email
+        }}
+        
+        orginal_response.data.update(custom_response)
+        return orginal_response