From: Guillaume Pellerin <yomguy@parisson.com>
Date: Wed, 10 Feb 2021 14:06:08 +0000 (+0100)
Subject: fix records filtering
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=b8329fdd99a361079c5b0d9f49ceecbf9fa40957;p=teleforma.git

fix records filtering
---

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 []