Basic UX for adding requirements to training levels

training/views.py

@@ -25,6 +25,7 @@ class TraineeDetail(views.ProfileDetail):
     def get_context_data(self, **kwargs):
         context = super(TraineeDetail, self).get_context_data(**kwargs)
         context["page_title"] = "{}'s Training Record".format(self.object)
+        context["levels"] = models.TrainingLevel.objects.all()
         context["categories"] = models.TrainingCategory.objects.all() 
         choices = models.TrainingItemQualification.CHOICES
         context["depths"] = choices
@@ -66,3 +67,22 @@ class AddQualification(generic.CreateView):
         kwargs = super(AddQualification, self).get_form_kwargs()
         kwargs['pk'] = self.kwargs['pk']
         return kwargs
+
+
+class AddLevelRequirement(generic.CreateView): 
+    template_name = "edit_training_level.html"
+    model = models.TrainingLevelRequirement
+    form_class = forms.RequirementForm
+
+    def get_context_data(self, **kwargs):
+        context = super(AddLevelRequirement, self).get_context_data(**kwargs)
+        context["page_title"] = "Add Requirements to Training Level {}".format(models.TrainingLevel.objects.get(pk=self.kwargs['pk']))
+        return context
+
+    def get_form_kwargs(self):
+        kwargs = super(AddLevelRequirement, self).get_form_kwargs()
+        kwargs['pk'] = self.kwargs['pk']
+        return kwargs
+
+    def get_success_url(self):
+        return reverse_lazy('trainee_detail')