Check for None instead

Author: stevebrownlee

LearningAPI/views/foundations.py

@@ -39,6 +39,15 @@ def create_entry(pk, exercise, request, user_id):
     """
     cohort = request.data.get('cohort', "Unassigned")
 
+    if cohort != None:
+        # Possible values are a string that starts with 'd' for day cohorts or 'e' for evening cohorts
+        # followed by a number representing the cohort number
+        # Convert 'd' into 'Day ' and 'e' into 'Evening '
+        if cohort.startswith('d'):
+            cohort = 'Day ' + cohort[1:]
+        elif cohort.startswith('e'):
+            cohort = 'Evening ' + cohort[1:]
+
     exercise.cohort = cohort
     exercise.slug = pk
     exercise.attempts = request.data.get('attempts', 0)