forked from gaugendre/ofx-processor
39 lines
1 KiB
Python
39 lines
1 KiB
Python
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):
|
|
"""Import CE bank statement (OFX file)."""
|
|
processor = CeProcessor(filename)
|
|
if send:
|
|
processor.send_reconciled_amount()
|
|
processor.push_to_ynab(keep)
|