ofx-processor/ofx_processor/processors/lcl.py

import sys
from datetime import datetime
from operator import truediv

import click
import dateparser

from ofx_processor.downloaders import LclDownloader
from ofx_processor.utils.base_ofx import OfxBaseLine, OfxBaseProcessor


class LclLine(OfxBaseLine):
    def _extract_payee_and_date(self):
        default_date = (
            self.data.dtposted.isoformat().split("T")[0] if self.data.dtposted else None
        )

        if self.data.trntype.lower() == "check":
            return "CHQ", default_date

        if not self.data.name:
            return "", default_date

        split = self.data.name.split()
        if not split:
            return "", default_date

        date_spot = split[-1]
        if "/" not in date_spot:
            return " ".join(split), default_date

        date = dateparser.parse(
            split[-1], date_formats=["%d/%m/%Y"], languages=["fr"]
        )  # type: datetime
        if date:
            name = " ".join(split[:-1])
            date = date.strftime("%Y-%m-%d")  # type: str
        else:
            name = " ".join(split)
            date = default_date

        return name, date

    def get_payee(self):
        return self._extract_payee_and_date()[0]

    def get_date(self):
        return self._extract_payee_and_date()[1]

    def get_memo(self):
        if self.data.trntype.lower() == "check":
            return f"CHQ {self.data.checknum}"
        return super().get_memo()


class LclProcessor(OfxBaseProcessor):
    line_class = LclLine
    account_name = "lcl"
    command_name = "lcl"

    def _parse_file(self):
        # The first line of this file needs to be removed.
        # It contains something that is not part of the header of an OFX file.
        try:
            with open(self.filename, "r") as user_file:
                data = user_file.read().splitlines(keepends=True)
        except FileNotFoundError:
            click.secho("Couldn't find ofx file", fg="red")
            sys.exit(1)

        new_lines = [line for line in data if is_valid_line(line)]

        with open(self.filename, "w") as temp_file:
            temp_file.writelines(new_lines)

        ofx = super()._parse_file()

        with open(self.filename, "w") as temp_file:
            temp_file.writelines(data)

        return ofx


def is_valid_line(line):
    if "Content-Type:" in line:
        return False
    if "MKTGINFO" in line:
        return False
    return True


def main(filename, keep, download, send_method, push_to_ynab):
    """Import LCL bank statement (OFX file)."""
    if download:
        if filename:
            click.secho(
                "You can't specify a file name with auto download. "
                "Specified file name will be ignored.",
                fg="yellow",
            )
        filename = LclDownloader().download()
    processor = LclProcessor(filename)
    processor.parse_file()
    if send_method:
        processor.send_reconciled_amount(send_method)
    if push_to_ynab:
        processor.push_to_ynab(keep)