N'envoie pas de mail si jamais il n'y a pas de cours pour une semainev0.6.0

1 files changed, 11 insertions, 8 deletions

diff --git a/management/commands/sendmails.py b/management/commands/sendmails.py
index 3126243..5a2054f 100644
--- a/management/commands/sendmails.py
+++ b/management/commands/sendmails.py
@@ -38,22 +38,25 @@ class Command(BaseCommand):
         subscriptions = Subscription.objects.filter(active=True)
         content = {}
         mails = []
+        footer = loader.get_template("mail/mail_footer.txt")
 
         print("Generating messages...")
         for subscription in subscriptions:
             if subscription.group.id not in content:
                 courses = Course.objects.get_courses_for_group(subscription.group, begin__gte=start, begin__lt=end)
-                grouped_courses = group_courses(courses)
 
-                template = loader.get_template("mail/mail_timetable.txt")
-                context = Context({"subscription": subscription, "courses": grouped_courses, "week": week})
-                content[subscription.group.id] = template.render(context)
+                if len(courses) > 0:
+                    grouped_courses = group_courses(courses)
 
-            footer = loader.get_template("mail/mail_footer.txt")
-            context = Context({"admins": settings.ADMINS, "token": subscription.token, "domain": settings.DEFAULT_DOMAIN})
-            mail_content = content[subscription.group.id] + footer.render(context)
+                    template = loader.get_template("mail/mail_timetable.txt")
+                    context = Context({"subscription": subscription, "courses": grouped_courses, "week": week})
+                    content[subscription.group.id] = template.render(context)
 
-            mails.append(("{0} - {1} - Semaine {2}".format(subscription.group.timetable.name, subscription.group.name, week), mail_content, settings.DEFAULT_FROM_EMAIL, [subscription.email],))
+            if subscription.group.id in content:
+                context = Context({"admins": settings.ADMINS, "token": subscription.token, "domain": settings.DEFAULT_DOMAIN})
+                mail_content = content[subscription.group.id] + footer.render(context)
+
+                mails.append(("{0} - {1} - Semaine {2}".format(subscription.group.timetable.name, subscription.group.name, week), mail_content, settings.DEFAULT_FROM_EMAIL, [subscription.email],))
 
         print("Sending mails...")
         send_mass_mail(mails)