django-orchestra-test/orchestra/contrib/payments/methods/sepadirectdebit.py

import datetime
import lxml.builder
import os
from lxml import etree
from lxml.builder import E
from io import StringIO

from django import forms
from django.utils import timezone
from django.utils.translation import ugettext_lazy as _
from django_iban.validators import IBANValidator, IBAN_COUNTRY_CODE_LENGTH
from rest_framework import serializers

from orchestra.plugins.forms import PluginDataForm

from .. import settings
from .options import PaymentMethod


class SEPADirectDebitForm(PluginDataForm):
    iban = forms.CharField(label='IBAN',
            widget=forms.TextInput(attrs={'size': '50'}))
    name = forms.CharField(max_length=128, label=_("Name"),
            widget=forms.TextInput(attrs={'size': '50'}))


class SEPADirectDebitSerializer(serializers.Serializer):
    iban = serializers.CharField(label='IBAN', validators=[IBANValidator()],
            min_length=min(IBAN_COUNTRY_CODE_LENGTH.values()), max_length=34)
    name = serializers.CharField(label=_("Name"), max_length=128)
    
    def validate(self, data):
        data['iban'] = data['iban'].strip()
        data['name'] = data['name'].strip()
        return data


class SEPADirectDebit(PaymentMethod):
    verbose_name = _("SEPA Direct Debit")
    label_field = 'name'
    number_field = 'iban'
    process_credit = True
    form = SEPADirectDebitForm
    serializer = SEPADirectDebitSerializer
    due_delta = datetime.timedelta(days=5)
    
    def get_bill_message(self):
        return _("This bill will be automatically charged to your bank account "
                 " with IBAN number<br><strong>%s</strong>.") % self.instance.number
    
    @classmethod
    def process(cls, transactions):
        debts = []
        credits = []
        for transaction in transactions:
            if transaction.amount < 0:
                credits.append(transaction)
            else:
                debts.append(transaction)
        processes = []
        if debts:
            proc = cls.process_debts(debts)
            processes.append(proc)
        if credits:
            proc = cls.process_credits(credits)
            processes.append(proc)
        return processes
    
    @classmethod
    def process_credits(cls, transactions):
        from ..models import TransactionProcess
        process = TransactionProcess.objects.create()
        context = cls.get_context(transactions)
        # http://businessbanking.bankofireland.com/fs/doc/wysiwyg/b22440-mss130725-pain001-xml-file-structure-dec13.pdf
        sepa = lxml.builder.ElementMaker(
             nsmap = {
                 'xsi': 'http://www.w3.org/2001/XMLSchema-instance',
                 None: 'urn:iso:std:iso:20022:tech:xsd:pain.001.001.03',
             }
        )
        sepa = sepa.Document(
            E.CstmrCdtTrfInitn(
                cls.get_header(context),
                E.PmtInf(                                   # Payment Info
                    E.PmtInfId(str(process.id)),            # Payment Id
                    E.PmtMtd("TRF"),                        # Payment Method
                    E.NbOfTxs(context['num_transactions']), # Number of Transactions
                    E.CtrlSum(context['total']),            # Control Sum
                    E.ReqdExctnDt(                          # Requested Execution Date
                        (context['now']+datetime.timedelta(days=10)).strftime("%Y-%m-%d")
                    ),
                    E.Dbtr(                                 # Debtor
                        E.Nm(context['name'])
                    ),
                    E.DbtrAcct(                             # Debtor Account
                        E.Id(
                            E.IBAN(context['iban'])
                        )
                    ),
                    E.DbtrAgt(                              # Debtor Agent
                        E.FinInstnId(                       # Financial Institution Id
                            E.BIC(context['bic'])
                        )
                    ),
                *list(cls.get_credit_transactions(transactions, process))   # Transactions
                )
            )
        )
        file_name = 'credit-transfer-%i.xml' % process.id
        cls.process_xml(sepa, 'pain.001.001.03.xsd', file_name, process)
        return process
    
    @classmethod
    def process_debts(cls, transactions):
        from ..models import TransactionProcess
        process = TransactionProcess.objects.create()
        context = cls.get_context(transactions)
        # http://businessbanking.bankofireland.com/fs/doc/wysiwyg/sepa-direct-debit-pain-008-001-02-xml-file-structure-july-2013.pdf
        sepa = lxml.builder.ElementMaker(
             nsmap = {
                 'xsi': 'http://www.w3.org/2001/XMLSchema-instance',
                 None: 'urn:iso:std:iso:20022:tech:xsd:pain.008.001.02',
             }
        )
        sepa = sepa.Document(
            E.CstmrDrctDbtInitn(
                cls.get_header(context, process),
                E.PmtInf(                                   # Payment Info
                    E.PmtInfId(str(process.id)),            # Payment Id
                    E.PmtMtd("DD"),                         # Payment Method
                    E.NbOfTxs(context['num_transactions']), # Number of Transactions
                    E.CtrlSum(context['total']),            # Control Sum
                    E.PmtTpInf(                             # Payment Type Info
                        E.SvcLvl(                           # Service Level
                            E.Cd("SEPA")                    # Code
                        ),
                        E.LclInstrm(                        # Local Instrument
                            E.Cd("CORE")                    # Code
                        ),
                        E.SeqTp("RCUR")                     # Sequence Type
                    ),
                    E.ReqdColltnDt(                         # Requested Collection Date
                        context['now'].strftime("%Y-%m-%d")
                    ),
                    E.Cdtr(                                 # Creditor
                        E.Nm(context['name'])
                    ),
                    E.CdtrAcct(                             # Creditor Account
                        E.Id(
                            E.IBAN(context['iban'])
                        )
                    ),
                    E.CdtrAgt(                              # Creditor Agent
                        E.FinInstnId(                       # Financial Institution Id
                            E.BIC(context['bic'])
                        )
                    ),
                *list(cls.get_debt_transactions(transactions, process))   # Transactions
                )
            )
        )
        file_name = 'direct-debit-%i.xml' % process.id
        cls.process_xml(sepa, 'pain.008.001.02.xsd', file_name, process)
        return process
    
    @classmethod
    def get_context(cls, transactions):
        return {
            'name': settings.PAYMENTS_DD_CREDITOR_NAME,
            'iban': settings.PAYMENTS_DD_CREDITOR_IBAN,
            'bic': settings.PAYMENTS_DD_CREDITOR_BIC,
            'at02_id': settings.PAYMENTS_DD_CREDITOR_AT02_ID,
            'now': timezone.now(),
            'total': str(sum([abs(transaction.amount) for transaction in transactions])),
            'num_transactions': str(len(transactions)),
        }
    
    @classmethod
    def get_debt_transactions(cls, transactions, process):
        for transaction in transactions:
            transaction.process = process
            transaction.state = transaction.WAITTING_EXECUTION
            transaction.save(update_fields=['state', 'process'])
            account = transaction.account
            data = transaction.source.data
            yield E.DrctDbtTxInf(                           # Direct Debit Transaction Info
                E.PmtId(                                    # Payment Id
                    E.EndToEndId(str(transaction.id))       # Payment Id/End to End
                ),
                E.InstdAmt(                                 # Instructed Amount
                    str(abs(transaction.amount)),
                    Ccy=transaction.currency.upper()
                ),
                E.DrctDbtTx(                                # Direct Debit Transaction
                    E.MndtRltdInf(                          # Mandate Related Info
                        E.MndtId(str(account.id)),          # Mandate Id
                        E.DtOfSgntr(                        # Date of Signature
                            account.date_joined.strftime("%Y-%m-%d")
                        )
                    )
                ),
                E.DbtrAgt(                                  # Debtor Agent
                    E.FinInstnId(                           # Financial Institution Id
                        E.Othr(
                            E.Id('NOTPROVIDED')
                        )
                    )
                ),
                E.Dbtr(                                     # Debtor
                    E.Nm(account.name),                     # Name
                ),
                E.DbtrAcct(                                 # Debtor Account
                    E.Id(
                        E.IBAN(data['iban'].replace(' ', ''))
                    ),
                ),
            )
    
    @classmethod
    def get_credit_transactions(transactions, process):
        for transaction in transactions:
            transaction.process = process
            transaction.state = transaction.WAITTING_EXECUTION
            transaction.save(update_fields=['state', 'process'])
            account = transaction.account
            data = transaction.source.data
            yield E.CdtTrfTxInf(                            # Credit Transfer Transaction Info
                E.PmtId(                                    # Payment Id
                    E.EndToEndId(str(transaction.id))       # Payment Id/End to End
                ),
                E.Amt(                                      # Amount
                    E.InstdAmt(                             # Instructed Amount
                        str(abs(transaction.amount)),
                        Ccy=transaction.currency.upper()
                    )
                ),
                E.CdtrAgt(                                  # Creditor Agent
                    E.FinInstnId(                           # Financial Institution Id
                        E.Othr(
                            E.Id('NOTPROVIDED')
                        )
                    )
                ),
                E.Cdtr(                                     # Debtor
                    E.Nm(account.name),                     # Name
                ),
                E.CdtrAcct(                                 # Creditor Account
                    E.Id(
                        E.IBAN(data['iban'].replace(' ', ''))
                    ),
                ),
            )
    
    @classmethod
    def get_header(cls, context, process):
        return E.GrpHdr(                            # Group Header
            E.MsgId(str(process.id)),           # Message Id
            E.CreDtTm(                              # Creation Date Time
                context['now'].strftime("%Y-%m-%dT%H:%M:%S")
            ),
            E.NbOfTxs(context['num_transactions']), # Number of Transactions
            E.CtrlSum(context['total']),            # Control Sum
            E.InitgPty(                             # Initiating Party
                E.Nm(context['name']),              # Name
                E.Id(                               # Identification
                    E.OrgId(                        # Organisation Id
                        E.Othr(
                            E.Id(context['at02_id'])
                        )
                    )
                )
            )
        )
    
    @classmethod
    def process_xml(cls, sepa, xsd, file_name, process):
        # http://www.iso20022.org/documents/messages/1_0_version/pain/schemas/pain.008.001.02.zip
        path = os.path.dirname(os.path.realpath(__file__))
        xsd_path = os.path.join(path, xsd)
        schema_doc = etree.parse(xsd_path)
        schema = etree.XMLSchema(schema_doc)
        sepa = StringIO(etree.tostring(sepa).decode('utf8'))
        sepa = etree.parse(sepa)
        schema.assertValid(sepa)
        process.file = file_name
        process.save(update_fields=['file'])
        sepa.write(process.file.path,
                   pretty_print=True,
                   xml_declaration=True,
                   encoding='UTF-8')