Layout fixes, cover update for files.

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

diff --git a/src/club/management/commands/send_receipts.py b/src/club/management/commands/send_receipts.py
index e461b68..92b5dc2 100644 (file)
--- a/src/club/management/commands/send_receipts.py
+++ b/src/club/management/commands/send_receipts.py
@@ -6,11 +6,62 @@ import traceback
 from django.core.management.base import BaseCommand, CommandError
 from django.utils.timezone import now
 from club.models import PayUOrder
+from funding.models import Funding
+from paypal.models import BillingAgreement
+from pz.models import Payment
 
 
 class Command(BaseCommand):
+    def add_arguments(self, parser):
+        parser.add_argument(
+            'year', type=int, metavar='YEAR',
+            help='Send receipts for the year.')
+        parser.add_argument(
+            '--emails',
+            help='Send only to these emails.')
+        parser.add_argument(
+            '--resend', action='store_true',
+            help='This is a re-send.')
+
     def handle(self, *args, **options):
-        year = 2019
-        for email in PayUOrder.objects.filter(completed_at__year=2019).order_by('schedule__email').values_list('schedule__email', flat=True).distinct():
+        year = options['year']
+        emails = set(
+            PayUOrder.objects.filter(
+                completed_at__year=year
+            ).order_by('schedule__email').values_list(
+                'schedule__email', flat=True
+            ).distinct()
+        )
+        emails.update(
+            BillingAgreement.objects.all().order_by(
+                'schedule__email').values_list(
+                'schedule__email', flat=True
+            ).distinct()
+        )
+        emails.update(
+            Funding.objects.exclude(email='').filter(
+                completed_at__year=year
+            ).order_by('email').values_list(
+                'email', flat=True
+            ).distinct()
+        )
+        emails.update(
+            Payment.objects.exclude(debit__email='').filter(
+                realised=True,
+                is_dd=True,
+                booking_date__year=year,
+            ).order_by('debit__email').values_list(
+                'debit__email', flat=True
+            ).distinct()
+        )
+
+        if options['emails']:
+            emails = options['emails'].split(',')
+
+        for email in emails:
             print(email)
-            PayUOrder.send_receipt(email, year)
+            try:
+                PayUOrder.send_receipt(email, year, resend=options['resend'])
+            except:
+                print('ERROR')
+                raise