Forgot that needed migrations generating

training/views.py

@@ -1,10 +1,12 @@
-from django.shortcuts import render
+import reversion
 
+from django.shortcuts import render
 from django.urls import reverse_lazy
 from django.views import generic
 from PyRIGS.views import OEmbedView, is_ajax
 from training import models, forms
 from django.utils import timezone
+from django.db import transaction
 
 from users import views
 
@@ -98,6 +100,13 @@ class AddLevelRequirement(generic.CreateView):
     def get_success_url(self):
         return reverse_lazy('trainee_detail')
 
+    @transaction.atomic()
+    @reversion.create_revision()
+    def form_valid(self, form, *args, **kwargs):
+        reversion.add_to_revision(form.cleaned_data['level'])
+        reversion.set_comment("Level requirement added")
+        return super().form_valid(form, *args, **kwargs)
+
 
 class LevelDetail(generic.DetailView):
     template_name = "level_detail.html"
@@ -129,4 +138,3 @@ class ConfirmLevel(generic.RedirectView):
         level_qualification.confirmed_on = timezone.now()
         level_qualification.save()
         return reverse_lazy('trainee_detail', kwargs={'pk': kwargs['pk']})
-