forked from gaugendre/ofx-processor
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import re
|
|
|
|
import click
|
|
|
|
from ofx_processor.utils.base_ofx import OfxBaseLine, OfxBaseProcessor
|
|
|
|
|
|
class BpvfLine(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):
|
|
if "CB****" in name:
|
|
conversion = re.compile(r"\d+,\d{2}[a-zA-Z]{3}")
|
|
match = conversion.search(memo)
|
|
if match:
|
|
res_name = memo[: match.start() - 1]
|
|
res_memo = name + memo[match.start() - 1 :]
|
|
else:
|
|
res_name = memo
|
|
res_memo = name
|
|
|
|
return res_name, res_memo
|
|
|
|
return name, memo
|
|
|
|
|
|
class BpvfProcessor(OfxBaseProcessor):
|
|
line_class = BpvfLine
|
|
account_name = "bpvf"
|
|
|
|
@staticmethod
|
|
@click.command("bpvf")
|
|
@click.argument("ofx_filename")
|
|
def main(ofx_filename):
|
|
"""Import BPVF bank statement (OFX file)."""
|
|
BpvfProcessor(ofx_filename).push_to_ynab()
|