import_revolut.py

from beancount.core import data

import beangulp
from beancount_importers.bank_classifier import payee_to_account_mapping
from beangulp.importers import csv

Col = csv.Col

# UNCATEGORIZED_EXPENSES_ACCOUNT = "Expenses:Uncategorized:Revolut"
UNCATEGORIZED_EXPENSES_ACCOUNT = "Expenses:FIXME"


def categorizer(txn, row):
    payee = row[4]
    comment = row[4]
    if comment.startswith("To "):
        payee = comment[3:]

    posting_account = None
    if txn.postings[0].units.number < 0:
        # Expenses
        posting_account = payee_to_account_mapping.get(payee)

        # Default by category
        if not posting_account:
            posting_account = UNCATEGORIZED_EXPENSES_ACCOUNT
    else:
        if "Withdrawing savings" in comment:
            posting_account = "Assets:Revolut:Savings"
        elif "Metal Cashback" in comment:
            posting_account = "Income:Revolut:Cashback"
        elif "Referral reward" in comment:
            posting_account = "Income:Revolut:Referrals"
        else:
            posting_account = "Income:Uncategorized:Revolut"
            # Ignore most incoming transactions as they will mostly duplicate Monzo's
            txn.meta["skip_transaction"] = True

    txn.postings.append(
        data.Posting(posting_account, -txn.postings[0].units, None, None, None, None)
    )

    return txn


def get_importer(account, currency):
    return csv.CSVImporter(
        {
            Col.DATE: "Started Date",
            Col.NARRATION: "Description",
            Col.AMOUNT: "Amount",
            Col.PAYEE: "Description",
            Col.CURRENCY: "Currency",
            Col.BALANCE: "Balance",
        },
        account,
        currency,
        categorizer=categorizer,
    )


if __name__ == "__main__":
    ingest = beangulp.Ingest([get_importer("Assets:Revolut:Cash", "GBP")], [])
    ingest()