Make TrainingLevelRequirement the correct level of unique

Also updates generateSampleData to match

training/management/commands/generateSampleTrainingData.py

@@ -35,15 +35,15 @@ class Command(BaseCommand):
 
     def setup_categories(self):
         names = [(1, "Basic"), (2, "Sound"), (3, "Lighting"), (4, "Rigging"), (5, "Power"), (6, "Haulage")]
-        
+
         for i, name in names:
             category = models.TrainingCategory.objects.create(reference_number=i, name=name)
             category.save()
             self.categories.append(category)
 
     def setup_items(self):
-        names = ["Motorised Power Towers", "Catering", "Forgetting Cables", "Gazebo Construction", "Balanced Audio", "Unbalanced Audio", "BBQ/Bin Interactions", "Pushing Boxes", "How Not To Die", "Setting up projectors", "Basketing truss", "First Aid", "Digging Trenches", "Avoiding Bin Lorries", "Getting cherry pickers stuck in mud", "Crashing the Van"]
-        
+        names = ["Motorised Power Towers", "Catering", "Forgetting Cables", "Gazebo Construction", "Balanced Audio", "Unbalanced Audio", "BBQ/Bin Interactions", "Pushing Boxes", "How Not To Die", "Setting up projectors", "Basketing truss", "First Aid", "Digging Trenches", "Avoiding Bin Lorries", "Getting cherry pickers stuck in mud", "Crashing the Van", "Getting pigs to fly", "Basketing picnics", "Python programming", "Building Cables", "Unbuilding Cables", "Cat Herding", "Pancake making", "Tidying up", "Reading Manuals", "Bikeshedding"]
+
         for i,name in enumerate(names):
             item = models.TrainingItem.objects.create(category=random.choice(self.categories), reference_number=random.randint(0, 100), name=name)
             self.items.append(item)
@@ -54,10 +54,15 @@ class Command(BaseCommand):
             technician = models.TrainingLevel.objects.create(level=models.TrainingLevel.TECHNICIAN, department=i, description="Moral pinnacle derive ultimate war dead. Strong fearful joy contradict battle christian faithful enlightenment prejudice zarathustra moral.")
             supervisor = models.TrainingLevel.objects.create(level=models.TrainingLevel.SUPERVISOR, department=i, description="Spirit holiest merciful mountains inexpedient reason value. Suicide ultimate hope.")
             supervisor.prerequisite_levels.add(technician)
+            items = self.items.copy()
             for i in range(0, 30):
+                if len(items) == 0:
+                    break
+                item = random.choice(items)
+                items.remove(item)
                 if i % 3 == 0:
-                    models.TrainingLevelRequirement.objects.create(level=technician, item=random.choice(self.items), depth=random.choice(models.TrainingItemQualification.CHOICES)[0])
+                    models.TrainingLevelRequirement.objects.create(level=technician, item=item, depth=random.choice(models.TrainingItemQualification.CHOICES)[0])
                 else:
-                    models.TrainingLevelRequirement.objects.create(level=supervisor, item=random.choice(self.items), depth=random.choice(models.TrainingItemQualification.CHOICES)[0])
+                    models.TrainingLevelRequirement.objects.create(level=supervisor, item=item, depth=random.choice(models.TrainingItemQualification.CHOICES)[0])
             self.levels.append(technician)
             self.levels.append(supervisor)

training/migrations/0004_auto_20210819_1808.py

@@ -0,0 +1,21 @@
+# Generated by Django 3.1.7 on 2021-08-19 17:08
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('training', '0003_auto_20210716_0150'),
+    ]
+
+    operations = [
+        migrations.AlterUniqueTogether(
+            name='trainingitemqualification',
+            unique_together={('trainee', 'item', 'depth')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='traininglevelqualification',
+            unique_together={('trainee', 'level')},
+        ),
+    ]

training/migrations/0005_auto_20210819_1833.py

@@ -0,0 +1,17 @@
+# Generated by Django 3.1.7 on 2021-08-19 17:33
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('training', '0004_auto_20210819_1808'),
+    ]
+
+    operations = [
+        migrations.AlterUniqueTogether(
+            name='traininglevelrequirement',
+            unique_together={('level', 'item')},
+        ),
+    ]

training/models.py

@@ -147,6 +147,9 @@ class TrainingLevelRequirement(models.Model):
     def __str__(self):
         return "{} in {}".format(TrainingItemQualification.CHOICES[self.depth][1], self.item)
 
+    class Meta:
+        unique_together = ["level", "item"]
+
 
 class TrainingLevelQualification(models.Model):
     trainee = models.ForeignKey('Trainee', related_name='levels', on_delete=models.RESTRICT)

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
 
@@ -15,7 +17,7 @@ class ItemList(generic.ListView):
     def get_context_data(self, **kwargs):
         context = super(ItemList, self).get_context_data(**kwargs)
         context["page_title"] = "Training Items"
-        context["categories"] = models.TrainingCategory.objects.all()    
+        context["categories"] = models.TrainingCategory.objects.all()
         return context
 
 
@@ -80,7 +82,7 @@ class AddQualification(generic.CreateView):
         return kwargs
 
 
-class AddLevelRequirement(generic.CreateView): 
+class AddLevelRequirement(generic.CreateView):
     template_name = "edit_training_level.html"
     model = models.TrainingLevelRequirement
     form_class = forms.RequirementForm
@@ -96,7 +98,14 @@ class AddLevelRequirement(generic.CreateView):
         return kwargs
 
     def get_success_url(self):
-        return reverse_lazy('trainee_detail')
+        return reverse_lazy('level_detail', kwargs={"pk": self.kwargs['pk']})
+
+    @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):
@@ -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']})
-

users/templates/profile_detail.html

@@ -56,43 +56,37 @@
         </div>
     {% endif %}
     <div class="row">
-        <div class="col-12">
+        <div class="col-lg-4 col-12 mb-2">
             <div class="card">
-                <div class="row no-gutters">
-                    <div class="col-md-3">
-                        <img src="{{object.profile_picture}}" class="card-img img-fluid" />
-                    </div>
-                    <div class="col-md-9">
-                        <div class="card-body">
-                            <dl class="row">
-                                <dt class="col-5">First Name</dt>
-                                <dd class="col-7">{{object.first_name}}</dd>
+                <img src="{{object.profile_picture}}" class="card-img img-fluid" />
+                <div class="card-body">
+                    <dl class="row">
+                        <dt class="col-5">First Name</dt>
+                        <dd class="col-7">{{object.first_name}}</dd>
 
-                                <dt class="col-5">Last Name</dt>
-                                <dd class="col-7">{{object.last_name}}</dd>
+                        <dt class="col-5">Last Name</dt>
+                        <dd class="col-7">{{object.last_name}}</dd>
 
-                                <dt class="col-5">Email</dt>
-                                <dd class="col-7">{{object.email}}</dd>
+                        <dt class="col-5">Email</dt>
+                        <dd class="col-7">{{object.email}}</dd>
 
-                                <dt class="col-5">Last Login</dt>
-                                <dd class="col-7">{{object.last_login|date:"d/m/Y H:i"}}</dd>
+                        <dt class="col-5">Last Login</dt>
+                        <dd class="col-7">{{object.last_login|date:"d/m/Y H:i"}}</dd>
 
-                                <dt class="col-5">Date Joined</dt>
-                                <dd class="col-7">{{object.date_joined|date:"d/m/Y H:i"}}</dd>
+                        <dt class="col-5">Date Joined</dt>
+                        <dd class="col-7">{{object.date_joined|date:"d/m/Y H:i"}}</dd>
 
-                                <dt class="col-5">Initials</dt>
-                                <dd class="col-7">{{object.initials}}</dd>
+                        <dt class="col-5">Initials</dt>
+                        <dd class="col-7">{{object.initials}}</dd>
 
-                                <dt class="col-5">Phone</dt>
-                                <dd class="col-7">{{object.phone|linkornone:'tel'}}</dd>
-                            </dl>
-                        </div>
-                    </div>
+                        <dt class="col-5">Phone</dt>
+                        <dd class="col-7">{{object.phone|linkornone:'tel'}}</dd>
+                    </dl>
                 </div>
-            </div>
+             </div>
         </div>
         {% if not request.is_ajax and object.pk == user.pk %}
-            <div class="col-12 my-2">
+            <div class="col-lg-8 col-12">
                 <div class="card">
                     <div class="card-header">Personal iCal Details</div>
                     <div class="card-body">
@@ -152,9 +146,19 @@
                 </div>
             </div>
         {% endif %}
+        <div class="col mb-2">
+            <div class="card">
+                <div class="card-header">Training Record</div>
+                <div class="card-body">
+                <a href="{% url 'trainee_detail' object.pk %}" class="btn btn-primary"><span class="fas fa-eye"></span> View Training Record</a>
+                </div>
+            </div>
+        </div>
+    </div>
+    <div class="card">
+        <div class="card-header">Events</div>
+        {% with object.latest_events as events %}
+            {% include 'partials/event_table.html' %}
+        {% endwith %}
     </div>
-    <h4>Events</h4>
-    {% with object.latest_events as events %}
-        {% include 'partials/event_table.html' %}
-    {% endwith %}
 {% endblock %}