sparkasse.py

#!/usr/bin/env python3
# coding: utf-8

"""
Run with
./sparkasse.py
"""

import io
import csv
import sys
import logging
import time
from decimal import Decimal

from converter import Converter


logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


class Sparkasse(Converter):
    """
    Implementation for Sparkasse bank
    """
    def __init__(self, *args, **kwargs):
        super(Sparkasse, self).__init__(*args, **kwargs)

    def load_transactions(self, filename):
        data = []
        with io.open(filename, 'r', encoding='cp1252') as csvfile:
            rd = csv.DictReader(csvfile, delimiter=';')
            for row in rd:
                data.append(row)
        return data

    def convert_row(self, line):
        """convert csv line to ynab entry"""

        def process_comment(spk_comment):
            if spk_comment.startswith('SVWZ+'):
                spk_comment.replace('SVWZ+', '', 1)
            return spk_comment

        def process_amount(spk_amount):
            return Decimal(spk_amount.replace(',', '.'))

        def process_date(spk_date):
            timestamp = time.strptime(spk_date, "%d.%m.%y")
            return time.strftime('%m/%d/%y', timestamp)

        if 'Umsatz getätigt von' in line:
            # Credit Card
            comment = line['Transaktionsbeschreibung']
            amount = process_amount(line['Buchungsbetrag'])
            date = process_date(line['Belegdatum'])
            payee = self.find_payee(comment)
            category = ""
            memo = comment
        elif 'Auftragskonto' in line:
            # SEPA
            spk_payee = line['Beguenstigter/Zahlungspflichtiger']
            comment = process_comment(line['Verwendungszweck'])
            amount = process_amount(line['Betrag'])
            date = process_date(line['Valutadatum'])
            payee = self.find_payee(spk_payee, comment)
            category = ""
            memo = "%s: %s" % (spk_payee, comment) if spk_payee else comment
        else:
            logger.error("Unrecognized CSV format")
            sys.exit(1)
        ynab = {
            'Date': date,
            'Payee': payee,
            'Category': category,
            'Memo': memo,
            'Outflow': -amount if amount < 0 else '',
            'Inflow': amount if amount > 0 else ''
        }
        return ynab


if __name__ == '__main__':
    ynab_file = "ynab_data_sparkasse.csv"
    spk_file = sys.argv[1]
    converter = Sparkasse()
    input_data = converter.load_transactions(spk_file)
    ynab_data = []
    for row in input_data:
        ynab_data.append(converter.convert_row(row))
    converter.export_file(ynab_file, ynab_data)