diff --git a/training/templates/trainee_detail.html b/training/templates/trainee_detail.html
index 9ea9ee3b..0dd0c3e7 100644
--- a/training/templates/trainee_detail.html
+++ b/training/templates/trainee_detail.html
@@ -57,7 +57,7 @@
{% endif %}
{% else %}
-
+
{% endif %}
{% else %}
diff --git a/training/views.py b/training/views.py
index 39c7114a..ff763e50 100644
--- a/training/views.py
+++ b/training/views.py
@@ -27,7 +27,7 @@ class TraineeDetail(views.ProfileDetail):
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()
+ context["categories"] = models.TrainingCategory.objects.all().prefetch_related('items')
choices = models.TrainingItemQualification.CHOICES
context["depths"] = choices
for i in [x for x,_ in choices]:
@@ -124,11 +124,9 @@ class RemoveRequirement(generic.DeleteView):
class ConfirmLevel(generic.RedirectView):
def get_redirect_url(self, *args, **kwargs):
- # TODO Prevent duplicate training level qualifications existing
- level_qualification = models.TrainingLevelQualification.objects.filter(trainee=kwargs['pk'], level=kwargs['level_pk'])
- print(level_qualification)
- #level_qualification.confirmed_by = self.request.user
- #level_qualification.confirmed_on = timezone.now()
- #level_qualification.save()
+ level_qualification = models.TrainingLevelQualification.objects.get(trainee=kwargs['pk'], level=kwargs['level_pk'])
+ level_qualification.confirmed_by = self.request.user
+ level_qualification.confirmed_on = timezone.now()
+ level_qualification.save()
return reverse_lazy('trainee_detail', kwargs={'pk': kwargs['pk']})