added an option to sign up for grade 12

praekeltfoundation · Jan 13, 2016 · eab1cab · eab1cab
1 parent 5bd99dc
commit eab1cab
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/oneplus/auth_views.py b/oneplus/auth_views.py
@@ -238,7 +238,7 @@ def post():
                         except Course.DoesNotExist:
                             course = Course.objects.create(name=settings.GRADE_10_COURSE_NAME)
                         classs = Class.objects.create(name=settings.GRADE_10_OPEN_CLASS_NAME, course=course)
-                else:
+                elif data["grade"] == "Grade 11":
                     try:
                         classs = Class.objects.get(name=settings.GRADE_11_OPEN_CLASS_NAME)
                     except Class.DoesNotExist:
@@ -247,6 +247,15 @@ def post():
                         except Course.DoesNotExist:
                             course = Course.objects.create(name=settings.GRADE_11_COURSE_NAME)
                         classs = Class.objects.create(name=settings.GRADE_11_OPEN_CLASS_NAME, course=course)
+                else:
+                    try:
+                        classs = Class.objects.get(name=settings.GRADE_12_OPEN_CLASS_NAME)
+                    except Class.DoesNotExist:
+                        try:
+                            course = Course.objects.get(name=settings.GRADE_12_COURSE_NAME)
+                        except Course.DoesNotExist:
+                            course = Course.objects.create(name=settings.GRADE_12_COURSE_NAME)
+                        classs = Class.objects.create(name=settings.GRADE_12_OPEN_CLASS_NAME, course=course)
 
                 create_participant(new_learner, classs)