From b8329fdd99a361079c5b0d9f49ceecbf9fa40957 Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Wed, 10 Feb 2021 15:06:08 +0100 Subject: [PATCH] fix records filtering --- teleforma/webclass/models.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/teleforma/webclass/models.py b/teleforma/webclass/models.py index b971968a..8ba6006d 100644 --- a/teleforma/webclass/models.py +++ b/teleforma/webclass/models.py @@ -394,9 +394,14 @@ class WebclassRecord(models.Model): return "Enregistrement webclass %d" % self.id @staticmethod - def get_records(period, course, course_type): + def get_records(period, course, course_type=None): record_ids = set() - for record in WebclassRecord.objects.filter(period=period, course=course, course_type=course_type): + if not course_type: + records = WebclassRecord.objects.filter(period=period, course=course, course_type=course_type) + else: + records = WebclassRecord.objects.filter(period=period, course=course).exclude(course_type=None) + + for record in records: record_ids.add(record.record_id) if not record_ids: return [] -- 2.39.5