1 files changed, 44 insertions, 0 deletions
diff --git a/grades/views.py b/grades/views.py
new file mode 100644
index 0000000..86803c9
--- /dev/null
+++ b/grades/views.py
@@ -0,0 +1,44 @@
+from django.shortcuts import render
+from django.contrib.auth.decorators import login_required
+from django.forms import inlineformset_factory
+from grades.forms import GradingSystemForm
+from grades.models import GradingSystem, GradeRange
+
+# Create your views here.
+@login_required
+def grading_systems(request):
+    user = request.user
+    default_grading_system = GradingSystem.objects.get(name='default')
+    grading_systems = GradingSystem.objects.filter(creator=user).exclude(
+            name='default')
+    return render(request, 'grading_systems.html', {'default_grading_system':
+                  default_grading_system, 'grading_systems': grading_systems})
+
+
+@login_required
+def add_grading_system(request, system_id=None):
+    user = request.user
+    grading_system = None
+    if system_id is not None:
+        grading_system = GradingSystem.objects.get(id=system_id)
+    GradeRangeFormSet = inlineformset_factory(GradingSystem, GradeRange,
+                                              fields='__all__', extra=0)
+    grade_form = GradingSystemForm(instance=grading_system)
+
+    if request.method == 'POST':
+        formset = GradeRangeFormSet(request.POST, instance=grading_system)
+        grade_form = GradingSystemForm(request.POST, instance=grading_system)
+        if grade_form.is_valid():
+            system = grade_form.save(commit=False)
+            system.creator = user
+            system.save()
+            system_id = system.id
+        if formset.is_valid():
+            formset.save()
+        if 'add' in request.POST:
+            GradeRangeFormSet = inlineformset_factory(GradingSystem, GradeRange,
+                                                      fields='__all__', extra=1)
+    formset = GradeRangeFormSet(instance=grading_system)
+
+    return render(request, 'add_grades.html', {'formset': formset,
+                  'grade_form': grade_form, "system_id": system_id})