From 4d0d4f02aa99fc9cc1b8be2ec2dc96db31665dc9 Mon Sep 17 00:00:00 2001 From: FreneticScribbler Date: Fri, 20 Aug 2021 12:38:30 +0100 Subject: [PATCH] Generate a sample supervisor --- .../management/commands/generateSampleTrainingData.py | 10 ++++++++++ training/models.py | 2 +- users/management/commands/generateSampleUserData.py | 2 ++ 3 files changed, 13 insertions(+), 1 deletion(-) diff --git a/training/management/commands/generateSampleTrainingData.py b/training/management/commands/generateSampleTrainingData.py index a95240cf..d29666d8 100644 --- a/training/management/commands/generateSampleTrainingData.py +++ b/training/management/commands/generateSampleTrainingData.py @@ -31,6 +31,7 @@ class Command(BaseCommand): self.setup_categories() self.setup_items() self.setup_levels() + self.setup_supervisor() print("Done generating training data") def setup_categories(self): @@ -66,3 +67,12 @@ class Command(BaseCommand): models.TrainingLevelRequirement.objects.create(level=supervisor, item=item, depth=random.choice(models.TrainingItemQualification.CHOICES)[0]) self.levels.append(technician) self.levels.append(supervisor) + + def setup_supervisor(self): + supervisor = models.Profile.objects.create(username="supervisor", first_name="Super", last_name="Visor", + initials="SV", + email="supervisor@example.com", is_active=True, + is_staff=True) + supervisor.set_password('supervisor') + supervisor.save() + models.TrainingLevelQualification.objects.create(trainee=supervisor, level=self.levels[-1], confirmed_on=timezone.now()) diff --git a/training/models.py b/training/models.py index 9d87419e..a543a9fc 100644 --- a/training/models.py +++ b/training/models.py @@ -12,7 +12,7 @@ class Trainee(Profile): def is_supervisor(self): # FIXME Efficiency for level_qualification in self.levels.select_related('level').all(): - if confirmed_on is not None and level_qualification.level.level >= TrainingLevel.SUPERVISOR: + if level_qualification.confirmed_on is not None and level_qualification.level.level >= TrainingLevel.SUPERVISOR: return True def get_records_of_depth(self, depth): diff --git a/users/management/commands/generateSampleUserData.py b/users/management/commands/generateSampleUserData.py index 63e75519..e6ac831a 100644 --- a/users/management/commands/generateSampleUserData.py +++ b/users/management/commands/generateSampleUserData.py @@ -20,6 +20,7 @@ class Command(BaseCommand): hs_group = None def handle(self, *args, **options): + print("Generating sample user data") from django.conf import settings if not (settings.DEBUG or settings.STAGING): @@ -32,6 +33,7 @@ class Command(BaseCommand): self.setup_groups() self.setup_useful_profiles() self.setup_generic_profiles() + print("Done generating sample user data") def setup_groups(self): self.keyholder_group = Group.objects.create(name='Keyholders')