From 2a61ee289639d5623c1a0279b5705995ef7c62a0 Mon Sep 17 00:00:00 2001 From: FreneticScribbler Date: Tue, 4 Jan 2022 18:36:27 +0000 Subject: [PATCH] Hmmm --- .../commands/generateSampleTrainingData.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/training/management/commands/generateSampleTrainingData.py b/training/management/commands/generateSampleTrainingData.py index ae37635b..6dbc0daf 100644 --- a/training/management/commands/generateSampleTrainingData.py +++ b/training/management/commands/generateSampleTrainingData.py @@ -193,9 +193,13 @@ class Command(BaseCommand): break item = random.choice(items) items.remove(item) - if i % 3 == 0: - models.TrainingLevelRequirement.objects.create(level=technician, item=item, depth=random.choice(models.TrainingItemQualification.CHOICES)[0]) - else: - models.TrainingLevelRequirement.objects.create(level=supervisor, item=item, depth=random.choice(models.TrainingItemQualification.CHOICES)[0]) + try: + if i % 3 == 0: + models.TrainingLevelRequirement.objects.create(level=technician, item=item, depth=random.choice(models.TrainingItemQualification.CHOICES)[0]) + else: + models.TrainingLevelRequirement.objects.create(level=supervisor, item=item, depth=random.choice(models.TrainingItemQualification.CHOICES)[0]) + except: # noqa + print("Failed create for {}. Weird.".format(item)) + self.levels.append(technician) self.levels.append(supervisor)