Add 'is van driver' to trainee list, haulage super doesn't count as a supervisor

2026-01-27 02:12:18 +00:00 · 2021-10-09 10:29:03 +01:00
parent 55558d1a4a
commit dbb9e3e530
5 changed files with 36 additions and 27 deletions
--- a/training/management/commands/generateSampleTrainingData.py
+++ b/training/management/commands/generateSampleTrainingData.py
@@ -90,4 +90,4 @@ class Command(BaseCommand):
                                                   is_staff=True)
        supervisor.set_password('supervisor')
        supervisor.save()
-        models.TrainingLevelQualification.objects.create(trainee=supervisor, level=self.levels[-1], confirmed_on=timezone.now())
+        models.TrainingLevelQualification.objects.create(trainee=supervisor, level=models.TrainingLevel.objects.filter(level__gte=models.TrainingLevel.SUPERVISOR).exclude(department=models.TrainingLevel.HAULAGE).first(), confirmed_on=timezone.now())