import datetime import lxml.builder import os from lxml import etree from lxml.builder import E from StringIO import StringIO from django import forms from django.utils import timezone from django.utils.translation import ugettext_lazy as _ from django_iban.forms import IBANFormField from django_iban.validators import IBANValidator, IBAN_COUNTRY_CODE_LENGTH from rest_framework import serializers from .. import settings from .options import PaymentSourceDataForm, PaymentMethod class SEPADirectDebitForm(PaymentSourceDataForm): iban = IBANFormField(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) 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, source): return _("This bill will been automatically charged to your bank account " " with IBAN number
%s.") % source.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 account = transaction.account data = transaction.source.data transaction.state = transaction.WAITTING_CONFIRMATION transaction.save() 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.register_date.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']) ), ), ) @classmethod def get_credit_transactions(transactions, process): for transaction in transactions: transaction.process = process account = transaction.account data = transaction.source.data transaction.state = transaction.WAITTING_CONFIRMATION transaction.save() 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']) ), ), ) @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 = etree.parse(StringIO(etree.tostring(sepa))) schema.assertValid(sepa) process.file = file_name process.save() sepa.write(process.file.path, pretty_print=True, xml_declaration=True, encoding='UTF-8')