Added new design

[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 9bc5e68..2705990 100644 (file)
--- a/src/club/management/commands/prolong.py
+++ b/src/club/management/commands/prolong.py
@@ -1,4 +1,8 @@
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#

 from datetime import timedelta
 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
 from django.core.management.base import BaseCommand, CommandError
 from django.utils.timezone import now
 from club.models import Schedule


@@ -6,7 +10,11 @@ from club.models import Schedule
 
 class Command(BaseCommand):
     def handle(self, *args, **options):
 
 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)
             print(s, s.email, s.expires_at)

-            s.pay(None)
+            try:
+                s.pay(None)
+            except:
+                traceback.print_exc()
+