login and auth bug fixed

apps/jwt_auth/views.py

@@ -50,12 +50,8 @@ class VerifyMobileAndGetTokenView(APIView):
         serializer = self.InputSerializer(data=request.data)
         serializer.is_valid(raise_exception=True)
 
-        print("ok")
-
         login_opts = verify_mobile_and_get_token(**serializer.validated_data)
 
-        print("ok2")
-
         response_data = self.OutputSerializer(login_opts, many=True).data
 
         return Response(response_data)
@@ -140,9 +136,12 @@ class CustomTokenObtainPairView(APIView):
 
         try:
             user_login_opt = CUserRoleLoginOpt.objects.get(uuid=uuid_opt)
-            user = user_login_opt.cuser_role.cuser
+            user = user_login_opt.cuser
             cuser_role = user_login_opt.cuser_role
 
+
+            workspace = cuser_role.sys_ws
+
             if user_login_opt:
                 refresh = RefreshToken.for_user(user)
 
@@ -151,15 +150,15 @@ class CustomTokenObtainPairView(APIView):
                 #print(cuser_role.uuid)
 
 
-                try:
-                    customerObj = CustomerModel.objects.get(cuser_role__uuid=str(cuser_role.uuid))
+                '''try:
+                    #customerObj = CustomerModel.objects.get(cuser_role__uuid=str(cuser_role.uuid))
+                    # workspace = customerObj.sys_ws
 
-                    #workspace = WorkspaceModel.objects.get(cuser_role_owner__uuid=str(cuser_role.uuid))
+                    workspace = cuser_role.sys_ws
 
-                    workspace = customerObj.sys_ws
                 except:
                     operator = CustomerResponsibleModel.objects.get(cuser_role=cuser_role)
-                    workspace = operator.sys_ws
+                    workspace = operator.sys_ws'''
 
                 refresh['workspace_uuid'] = str(workspace.uuid)
 
@@ -174,7 +173,10 @@ class CustomTokenObtainPairView(APIView):
                                            cuser=user, cuser_role=cuser_role)
                     jwt_db.save()
 
-                    CUserRoleLoginOpt.objects.filter(cuser_role__cuser=user).delete()
+                    try:
+                        CUserRoleLoginOpt.objects.filter(cuser=user).delete()
+                    except Exception as exp:
+                        pass
 
                 except Exception as exp:
                     return Response({'error': str(exp)})