Merge pull request #189 from CDC-IITDH/karthikmurakonda-patch-3

Update companyViews.py
CDC-IITDH · Nov 9, 2023 · 6c484c1 · 6c484c1
2 parents f6d27a9 + 72c6187
commit 6c484c1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/CDC_Backend/APIs/companyViews.py b/CDC_Backend/APIs/companyViews.py
@@ -418,13 +418,14 @@ def addInternship(request):
             internship.is_work_from_home = True
         else:
             internship.is_work_from_home = False
-        
-        if ALLOWED_BATCH in data[ALLOWED_BATCH] and data[ALLOWED_BATCH] is None or json.loads(data[ALLOWED_BATCH]) == "":
+
+        if ALLOWED_BATCH in data and (data[ALLOWED_BATCH] is None or json.loads(data[ALLOWED_BATCH]) == ""):
             raise ValueError('Allowed Batches cannot be empty')
-        elif ALLOWED_BATCH in data[ALLOWED_BATCH] and set(json.loads(data[ALLOWED_BATCH])).issubset(BATCHES):
+        elif ALLOWED_BATCH in data and set(json.loads(data[ALLOWED_BATCH])).issubset(BATCHES):
             internship.allowed_batch = json.loads(data[ALLOWED_BATCH])
         else:
             internship.allowed_batch = ['2021']
+
 
         if data[ALLOWED_BRANCH] is None or json.loads(data[ALLOWED_BRANCH]) == "":
             raise ValueError('Allowed Branch cannot be empty')