From 3ae507b469f978cc19104d0afaf5d6164199878e Mon Sep 17 00:00:00 2001 From: FreneticScribbler Date: Tue, 25 Jan 2022 13:08:33 +0000 Subject: [PATCH] Filter trainees for active approved users Closes #477 --- training/models.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/training/models.py b/training/models.py index b579e24f..e28087ca 100644 --- a/training/models.py +++ b/training/models.py @@ -6,11 +6,18 @@ from django.utils.safestring import mark_safe from versioning.versioning import RevisionMixin +class TraineeManager(models.Manager): + def get_queryset(self): + return super().get_queryset().filter(is_active=True, is_approved=True) + + @reversion.register(for_concrete_model=False, fields=[]) class Trainee(Profile, RevisionMixin): class Meta: proxy = True + objects = TraineeManager() + # FIXME use queryset def started_levels(self): return [level for level in TrainingLevel.objects.all() if level.percentage_complete(self) > 0 and level.pk not in self.level_qualifications.values_list('level', flat=True)]