ofx-processor/ofx_processor/main.py
2020-01-15 14:42:57 +01:00

53 lines
1.5 KiB
Python

import os
import re
import sys
from xml.etree import ElementTree
import click
from ofxtools.Parser import OFXTree
from ofxtools.header import make_header
@click.command()
@click.argument("ofx_filename")
def cli(ofx_filename):
parser = OFXTree()
try:
parser.parse(ofx_filename)
except FileNotFoundError:
click.secho("Couldn't open ofx file", fg="red")
sys.exit(1)
ofx = parser.convert()
if ofx is None:
click.secho("Couldn't parse ofx file", fg="red")
sys.exit(1)
for transaction in ofx.statements[0].transactions:
if "CB****" in transaction.name:
click.secho("Edited transaction {}".format(transaction.checknum), fg="blue")
name = transaction.name
memo = transaction.memo
conversion = re.compile(r"\d+,\d{2}[a-zA-Z]{3}")
match = conversion.search(memo)
if match:
transaction.name = memo[: match.start() - 1]
transaction.memo = name + memo[match.start() - 1 :]
else:
transaction.name = memo
transaction.memo = name
header = str(make_header(version=102))
root = ofx.to_etree()
data = ElementTree.tostring(root).decode()
processed_file = os.path.join(os.path.dirname(ofx_filename), "processed.ofx")
with open(processed_file, "w") as f:
f.write(header + data)
click.secho("{} written".format(processed_file), fg="green")
if __name__ == "__main__":
cli()