forked from gaugendre/ofx-processor
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import sys
|
|
|
|
import click
|
|
|
|
from ofx_processor.utils.base_ofx import OfxBaseLine, OfxBaseProcessor
|
|
|
|
|
|
class LclLine(OfxBaseLine):
|
|
pass
|
|
|
|
|
|
class LclProcessor(OfxBaseProcessor):
|
|
line_class = LclLine
|
|
account_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(True)
|
|
except FileNotFoundError:
|
|
click.secho("Couldn't find ofx file", fg="red")
|
|
sys.exit(1)
|
|
|
|
if "Content-Type:" in data[0]:
|
|
with open(self.filename, "w") as temp_file:
|
|
temp_file.writelines(data[1:])
|
|
|
|
transactions = super(LclProcessor, self).parse_file()
|
|
|
|
if "Content-Type:" in data[0]:
|
|
with open(self.filename, "w") as temp_file:
|
|
temp_file.writelines(data)
|
|
|
|
return transactions
|
|
|
|
@staticmethod
|
|
@click.command("lcl")
|
|
@click.argument("ofx_filename")
|
|
def main(ofx_filename):
|
|
"""Import LCL bank statement (OFX file)."""
|
|
LclProcessor(ofx_filename).push_to_ynab()
|