gather_transactions_daily.py

import os; os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'brmburo.settings')

from brmburo.transactions import payment_income, account_sum
from brmburo.models import BankTransaction, Currency, BankAccount
from brmburo import settings
from fiobank import FioBank
from django.core.exceptions import ObjectDoesNotExist

from decimal import Decimal

from time import sleep
import datetime
import logging

#FIRST_DATE = datetime.date(2010, 9, 1)
FIRST_DATE = datetime.date(2015,1,1)
logger = logging.getLogger(__name__)



for token in settings.BANK_TOKENS:
    client = FioBank(token=token)
    info = client.info()
    curr,created = Currency.objects.get_or_create(symbol=info.get('currency'))
    if created:
        curr.name = curr.symbol
        curr.save()
    my,created = BankAccount.objects.get_or_create(account_number=info['account_number'], bank_code=info['bank_code'])
    if created:
        my.currency = curr
        my.save()
    logger.info('Processing account %s...' % my)
    try:
        from_id = BankTransaction.objects.filter(my_account=my).latest('date').tid
        sleep(30) # to avoid HTTP 409 - "Conflict" response
        trans = client.last(from_id=from_id)
    except ObjectDoesNotExist:
        sleep(30) # to avoid HTTP 409 - "Conflict" response
        trans = client.last(from_date=FIRST_DATE)

    for tran in trans:
        tid =  tran.get('transaction_id')
        if BankTransaction.objects.filter(tid=tid).exists():
            logger.warning('Transaction %s already here.'%tid)
            continue

        if 'account_number' not in tran or ( 'bank_code' not in  tran and 'bic' not in tran ):
            continue

        acc,created = BankAccount.objects.get_or_create(account_number=tran['account_number'], bank_code=tran['bic'] if 'bic' in tran else tran['bank_code'])
        if created:
            acc.account_name = tran.get('account_name')
            acc.save()

        curr,created = Currency.objects.get_or_create(symbol=tran.get('currency'))
        if created:
            curr.name = curr.symbol
            curr.save()

        todecimal = lambda x: Decimal(str(x)) if x is not None else x

        t = BankTransaction.objects.create(
            tid = tran.get('transaction_id'),
            my_account = my,
            their_account = acc,
            amount = todecimal(tran.get('amount')),
            currency = curr,
            constant_symbol = tran.get('constant_symbol'),
            specific_symbol = tran.get('specific_symbol'),
            variable_symbol = tran.get('variable_symbol'),
            recipient_message = tran.get('recipient_message'),
            comment = tran.get('comment'),
            date = tran.get('date'),
        )


        t.save()
        logger.info(u'Imported transaction %s.' % t)

        lt, b = payment_income(t)
        if lt is not None and b is not None:
            template = u'Paired incomming payment %(amount)0.2f[%(curr1)s] from @%(nickname)s accounted as transaction #%(tid)d. New balance for @%(nickname)s is %(balance)s %(curr2)s.'
            logger.info(template % dict(
                amount = t.amount,
                curr1 = t.currency.symbol,
                curr2 = b.logic_account.currency.symbol,
                nickname = b.nickname,
                tid = lt.id,
                balance = account_sum(b),
            ))
        else:
            logger.info(u'Not paired.')