La commande cleancourses efface les différences de cours

1 files changed, 3 insertions, 1 deletions

diff --git a/management/commands/cleancourses.py b/management/commands/cleancourses.py
index 5f71861..42f8f8b 100644
--- a/management/commands/cleancourses.py
+++ b/management/commands/cleancourses.py
@@ -16,7 +16,7 @@
 
 from django.core.management.base import BaseCommand
 from django.db import transaction
-from edt.models import Course, Group, LastUpdate
+from edt.models import Course, CourseDelta, Group, LastUpdate
 
 
 class Command(BaseCommand):
@@ -28,10 +28,12 @@ class Command(BaseCommand):
     def handle(self, *args, **options):
         with transaction.atomic():
             if options["timetable"] is None:
+                CourseDelta.objects.all().delete()
                 Course.objects.all().delete()
                 Group.objects.all().delete()
                 LastUpdate.objects.all().delete()
             else:
+                CourseDelta.objects.filter(course__timetable__id__in=options["timetable"]).delete()
                 Course.objects.filter(timetable__id__in=options["timetable"]).delete()
                 Group.objects.filter(timetable__id__in=options["timetable"]).delete()
                 LastUpdate.objects.filter(timetable__id__in=options["timetable"]).delete()