Merge pull request #7 from teamhashed:whoismaruf/issue6

Set cookies when refresh token.

credentials/api/views.py

@@ -2,16 +2,28 @@
 from django.http import JsonResponse
 from credentials.api.serializers import CustomUserDetailSerializer
 from django.contrib.auth import get_user_model
+from dj_rest_auth.jwt_auth import set_jwt_cookies
 
 
 def get_tokens_for_user(request):
-    refresh = RefreshToken.for_user(request.user)
-    print(f'token for {request.user}',refresh.access_token)
-    q__set = get_user_model().objects.filter(email=request.user.email).first()
-    user_detil = CustomUserDetailSerializer(q__set)
-    return JsonResponse({
-        'refresh_token': str(refresh),
-        'access_token': str(refresh.access_token),
-        'user': user_detil.data
-    })
-
+    try:
+        refresh = RefreshToken.for_user(request.user)
+        print(request.user)
+        q__set = get_user_model().objects.filter(email=request.user.email).first()
+        user_detail = CustomUserDetailSerializer(q__set)
+        data = {
+            'refresh_token': str(refresh),
+            'access_token': str(refresh.access_token),
+            'user': user_detail.data
+        }
+        res = JsonResponse(data)
+        set_jwt_cookies(response=res,
+                        access_token=str(refresh.access_token),
+                        refresh_token=str(refresh)
+                        )
+        return res
+    except Exception as e:
+        return JsonResponse({
+            'status': 404,
+            'message': str(e)
+        })