LIL-705

apps/notification/tasks.py

@@ -33,7 +33,7 @@ def send_certificates(email=None, date=None, dry_run=False):
             file.close()
             return
 
-    date = datetime.strptime(date, '%d-%m-%Y') if date else now().date()
+    date = datetime.strptime(date, '%d-%m-%Y').date() if date else now().date()
     today = now().date()
     users = set(list(SchoolPayment.objects.filter(date_end=date, add_days=False).values_list('user_id', flat=True)))
     user_notifications_qs = UserNotification.objects.filter(user_id__in=users)
@@ -42,11 +42,9 @@ def send_certificates(email=None, date=None, dry_run=False):
         'user_id', flat=True).distinct()
     for user_id in users:
         if user_id in notified_users:
+            print('skip', user_id)
             continue
-        if not user_id in user_notifications:
-            un = UserNotification.objects.create(user_id=user_id)
-        else:
-            un = user_notifications.get(user_id)
+        un = user_notifications.get(user_id, UserNotification(user_id=user_id))
         print(un.user.email)
         if dry_run:
             continue