Merge pull request #1948 from jelanmathewjames/dev

fix: key error
gtech-mulearn · May 14, 2024 · 78d44f5 · 78d44f5
2 parents f24b2c1 + 398313e
commit 78d44f5
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/api/integrations/wadhwani/wadhwani_views.py b/api/integrations/wadhwani/wadhwani_views.py
@@ -19,7 +19,7 @@ def post(self, request):
         headers = {'Content-Type': 'application/x-www-form-urlencoded'}
         response = requests.post(url, data=data, headers=headers)
 
-        if response.json()["status"] == "ERROR":
+        if response.json().get("error", None):
             return CustomResponse(general_message="Invalid credentials").get_failure_response()
         return CustomResponse(response=response.json()).get_success_response()
 
@@ -47,7 +47,7 @@ def post(self, request):
         }
         response = requests.post(url, data=data)
 
-        if response.json()["status"] == "ERROR":
+        if response.json().get("status", None) == "ERROR":
             return CustomResponse(general_message="Invalid Input").get_failure_response()
         return CustomResponse(response=response.json()).get_success_response()
 
@@ -61,7 +61,7 @@ def post(self, request):
         headers = {'Authorization': token}
         response = requests.get(url, headers=headers)
 
-        if response.json()["status"] == "ERROR":
+        if response.json().get("status", None) == "ERROR":
             return CustomResponse(general_message="No courses available").get_failure_response()
         return CustomResponse(response=response.json()).get_success_response()
 
@@ -77,7 +77,7 @@ def post(self, request):
         headers = {'Authorization': token}
         response = requests.get(url, params={"username": user.email}, headers=headers)
 
-        if response.json()["status"] == "ERROR":
+        if response.json().get("status", None) == "ERROR":
             return CustomResponse(general_message="User doesn't have any enrolled courses").get_failure_response()
         return CustomResponse(response=response.json()).get_success_response()
 
@@ -95,6 +95,6 @@ def post(self, request):
         url = settings.WADHWANI_BASE_URL + f"/api/v1/courseservice/oauth/course/{course_id}/reports/quiz/student/{user.email}"
         response = requests.get(url, headers=headers)
 
-        if response.json()["status"] == "ERROR":
+        if response.json().get("status", None) == "ERROR":
             return CustomResponse(general_message="No quiz data available").get_failure_response()
         return CustomResponse(response=response.json()).get_success_response()