Style fix

[wolnelektury.git] / src / club / management / commands / prolong.py

diff --git a/src/club/management/commands/prolong.py b/src/club/management/commands/prolong.py
index 3c8b36c..2705990 100644 (file)
--- a/src/club/management/commands/prolong.py
+++ b/src/club/management/commands/prolong.py
@@ -2,6 +2,7 @@
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from datetime import timedelta
+import traceback
 from django.core.management.base import BaseCommand, CommandError
 from django.utils.timezone import now
 from club.models import Schedule
@@ -9,7 +10,11 @@ from club.models import Schedule
 
 class Command(BaseCommand):
     def handle(self, *args, **options):
-        for s in Schedule.objects.filter(is_cancelled=False, expires_at__lt=now() + timedelta(1)):
+        for s in Schedule.objects.exclude(monthly=False, yearly=False).filter(is_cancelled=False, expires_at__lt=now() + timedelta(1)):
             print(s, s.email, s.expires_at)
-            s.pay(None)
+            try:
+                s.pay(None)
+            except:
+                traceback.print_exc()
+