fix records filtering

diff --git a/teleforma/webclass/models.py b/teleforma/webclass/models.py
index b971968a8b0823b0184f419f24cb190ccf0c15ba..8ba6006d05cf551277a8ec7574e161f904b9b827 100644 (file)
--- 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 []