ofx-processor/ofx_processor/processors/ce.py

import re

from ofx_processor.utils.base_ofx import OfxBaseLine, OfxBaseProcessor


class CeLine(OfxBaseLine):
    def get_memo(self):
        return self._process_name_and_memo(self.data.name, self.data.memo)[1]

    def get_payee(self):
        return self._process_name_and_memo(self.data.name, self.data.memo)[0]

    @staticmethod
    def _process_name_and_memo(name: str, memo: str):
        name = name.strip()
        cb_format = re.compile(r"FACT \d{6}$")
        match = cb_format.search(name)
        if match:
            res_name = name[: match.start() - 1].strip()
            res_memo = name[match.start() - 1 :].strip()
        else:
            res_name = name
            res_memo = memo
        return res_name, res_memo


class CeProcessor(OfxBaseProcessor):
    account_name = "ce"
    command_name = "ce"
    line_class = CeLine


def main(filename, keep, send_method):
    """Import CE bank statement (OFX file)."""
    processor = CeProcessor(filename)
    if send_method:
        processor.send_reconciled_amount(send_method)
    processor.push_to_ynab(keep)