Filtering search results.

[wolnelektury.git] / src / pz / bank.py

diff --git a/src/pz/bank.py b/src/pz/bank.py
index e1e91b7..0c75c7b 100644 (file)
--- a/src/pz/bank.py
+++ b/src/pz/bank.py
@@ -1,5 +1,8 @@
 import csv
+from io import StringIO
+from django.conf import settings
 from django.http import HttpResponse
+from django.utils.safestring import mark_safe
 from django.utils.translation import ugettext_lazy as _
 
 
@@ -7,34 +10,108 @@ def bank_export(modeladmin, request, queryset):
     response = HttpResponse(content_type='text/csv; charset=cp1250')
     response['Content-Disposition'] = 'attachment; filename=export.csv'
     writer = csv.writer(response)
-    writer.writerow([
-        'Identyfikator płatności (IDP)',
-        'Nazwa Płatnika',
-        'Adres Płatnika Ulica + numer domu',
-        'Adres Płatnika kod+miejscowość',
-        'Numer kierunkowy banku Płatnika',
-        'Numer rachunku bankowego Płatnika',
-        'Identyfikator Odbiorcy (NIP Odbiorcy)',
-        'Osobowość prawna Płatnika (Osoba fizyczna)'
-    ])
-
-    # TODO: ansi encoding
 
     for obj in queryset:
-        street_addr = obj.street
-        if obj.building:
-            street_addr += ' ' + obj.building
-        if obj.flat:
-            street_addr += ' m. ' + obj.flat
-        street_addr = street_addr.strip()
         writer.writerow([
             obj.payment_id,
-            ' '.join([obj.first_name, obj.last_name]).strip(),
-            street_addr,
+            obj.full_name,
+            '',
+            obj.street_address,
             ' '.join([obj.postal_code, obj.town]).strip(),
             obj.iban[2:10],
             obj.iban,
-            '9521877087',
+            settings.PZ_NIP,
             'OF'
         ])
     return response
+
+
+def parse_export_feedback(f):
+    lines = csv.reader(StringIO(f.read().decode('cp1250')))
+    for line in lines:
+        payment_id = line[0]
+        status = int(line[8])
+        comment = line[9]
+        yield payment_id, status, comment
+
+
+def bank_order(date, sent_at, queryset):
+    response = HttpResponse(content_type='application/octet-stream')
+    response['Content-Disposition'] = 'attachment; filename=order.PLD'
+    rows = []
+
+    no_dates = []
+    no_amounts = []
+    cancelled = []
+
+    if date is None:
+        raise ValueError('Payment date not set yet.')
+
+    for debit in queryset:
+        if debit.bank_acceptance_date is None:
+            no_dates.append(debit)
+        if debit.amount is None:
+            no_amounts.append(debit)
+        if debit.cancelled_at and debit.cancelled_at.date() <= date and (sent_at is None or debit.cancelled_at < sent_at):
+            cancelled.append(debit)
+
+    if no_dates or no_amounts or cancelled:
+        t = ''
+        if no_dates:
+            t += 'Bank acceptance not received for: '
+            t += ', '.join(
+                '<a href="/admin/pz/directdebit/{}/change">{}</a>'.format(
+                    debit.pk, debit
+                )
+                for debit in no_dates
+            )
+            t += '. '
+        if no_amounts:
+            t += 'Amount not set for: '
+            t += ', '.join(
+                '<a href="/admin/pz/directdebit/{}/change">{}</a>'.format(
+                    debit.pk, debit
+                )
+                for debit in no_amounts
+            )
+            t += '. '
+        if cancelled:
+            t += 'Debits have been cancelled: '
+            t += ', '.join(
+                '<a href="/admin/pz/directdebit/{}/change">{}</a>'.format(
+                    debit.pk, debit
+                )
+                for debit in cancelled
+            )
+            t += '. '
+        raise ValueError(mark_safe(t))
+
+    for debit in queryset:
+        rows.append(
+            '{order_code},{date},{amount},{dest_bank_id},0,"{dest_iban}","{user_iban}",'
+            '"{dest_addr}","{user_addr}",0,{user_bank_id},'
+            '"/NIP/{dest_nip}/IDP/{payment_id}|/TXT/{payment_desc}||",'
+            '"","","01"'.format(
+                order_code=210,
+                date=date.strftime('%Y%m%d'),
+
+                amount=debit.amount * 100,
+                dest_bank_id=settings.PZ_IBAN[2:10],
+                dest_iban=settings.PZ_IBAN,
+                user_iban=debit.iban,
+                dest_addr=settings.PZ_ADDRESS_STRING,
+                user_bank_id=debit.iban[2:10],
+                dest_nip=settings.PZ_NIP,
+                payment_id=debit.payment_id,
+                payment_desc=settings.PZ_PAYMENT_DESCRIPTION,
+                user_addr = '|'.join((
+                    debit.full_name,
+                    '',
+                    debit.street_address,
+                    ' '.join((debit.postal_code, debit.town))
+                ))
+            )
+        )
+    response.write('\r\n'.join(rows).encode('cp1250'))
+
+    return response