pyMt940/pymt940/unicredit.py

from .main import CsvToMt940, TypeId
from datetime import datetime

import pandas as pd
import re
import os


class Unicredit(CsvToMt940):

    def __init__(self, filename, iban):
        super().__init__()
        self.filename = filename
        self.iban = iban
        self._info_df = pd.read_csv(self.filename, nrows=1, sep=';')
        # TO DO: pd parse error at some utilities
        self._trans_df = pd.read_csv(self.filename, skiprows=2, sep=';', keep_default_na=False)
        self._trans_df.to_csv('logged_bacx.csv', mode='a', sep=';', header=False)  # not os.path.exists('logged_bacx.csv')
        self.transactions = map(self._statm_transactions, self._trans_df.values)
        self._ibans = {'RO23BACX0000001138725001': 'MZK', 'RO93BACX0000001138725002': 'MZK', 'RO50BACX0000001138725000': 'MZK',
                       'RO26BTRL01301202N63584XX': 'MZK', 'RO65BACX0000001144439000': 'ALP', 'RO44BTRL01301205G46199XX': 'GE'}
        if 'EUR' in self._info_df['Currency'][0]:
            self.iban = 'RO23BACX0000001138725001'
            # print(self._info_df[['Currency']])
        elif 'USD' in self._info_df['Currency'][0]:
            self.iban = 'RO93BACX0000001138725002'
        # else:
        #     self.iban = iban
        # iban = 'RO50BACX0000001138725000'
        # st_nr = randint(100000000, 199999999)

    def convert(self):
        with open('unicredit_mt940-{}-{}-{}.txt'.format(self.st_date, self._ibans[self.iban], self.iban), 'w') as export:
            export.write(self._statm_header())
            for elem in self.transactions:
                export.write(elem)
            export.write(self._statm_footer())

    def _statm_header(self):
        self.o_bal = self.dig_it(self._info_df['Opening Balance'][0])
        if self.o_bal > 0:
            cd_mark = 'C'
        else:
            cd_mark = 'D'
        if re.search('.', self._info_df['Opening Balance'][0]):
            n = self._info_df['Opening Balance'][0].replace('.', '')
        else:
            n = self._info_df['Opening Balance'][0]
        tr_ref_nr = ':20:' + str(self._info_df['Account'][0]) + '\n'
        acc_id = ':25:' + self.iban + '\n'
        st_ref = ':28:' + str(self.st_nr) + '\n'
        in_op_balance = ':60F:' + cd_mark + self.st_date +\
            self._info_df['Currency'][0] + n + '\n'
        return tr_ref_nr + acc_id + st_ref + in_op_balance

    def _statm_transactions(self, elem):
        if self.dig_it(elem[1]) > 0:
            CD_mark = 'C'
        else:
            CD_mark = 'D'
        if re.search('-', elem[1]):
            elem[1] = elem[1].replace('-', '')
        if re.search('.', elem[1]):
            elem[1] = elem[1].replace('.', '')
        if re.search(TypeId.types, str(elem[14]), flags=re.RegexFlag.IGNORECASE):
            trans_type = TypeId.COMMISION + 'OPH0001138725000'  # temp workarround for missing type on comm.
        else:
            trans_type = TypeId.TRANSFER
        tr = ':61:' + elem[3][-2:] + elem[3][3:-5] + elem[3][:2] + elem[3][3:-5] +\
            elem[3][:2] + CD_mark + elem[1] + trans_type + str(elem[-2]) +\
            elem[5] + '\n'
        inf = str(elem[14]) + elem[16] + '\n'
        more_inf = ':86:' + '+31' + elem[9] + '\n' + '+32' + elem[10] + '+33/' +\
            str(elem[-7]) + '\n'
        return tr + inf + more_inf

    def _statm_footer(self):
        c_bal = self.dig_it(self._info_df['Closing Balance'][0])
        if c_bal > 0:
            cd_mark = 'C'
        else:
            cd_mark = 'D'
        if re.search('.', self._info_df['Closing Balance'][0]):
            n = self._info_df['Closing Balance'][0].replace('.', '')
        else:
            n = self._info_df['Closing Balance'][0]
        fin_op_balance = ':62F:' + cd_mark + self.st_date + self._info_df['Currency'][0] + n
        return fin_op_balance