fix ofx lcl
This commit is contained in:
parent
c7acfcd099
commit
8f1662a861
2 changed files with 17 additions and 7 deletions
|
@ -1,5 +1,6 @@
|
|||
import sys
|
||||
from datetime import datetime
|
||||
from operator import truediv
|
||||
|
||||
import click
|
||||
import dateparser
|
||||
|
@ -67,19 +68,27 @@ class LclProcessor(OfxBaseProcessor):
|
|||
click.secho("Couldn't find ofx file", fg="red")
|
||||
sys.exit(1)
|
||||
|
||||
if "Content-Type:" in data[0]:
|
||||
new_lines = [line for line in data if is_valid_line(line)]
|
||||
|
||||
with open(self.filename, "w") as temp_file:
|
||||
temp_file.writelines(data[1:])
|
||||
temp_file.writelines(new_lines)
|
||||
|
||||
ofx = super()._parse_file()
|
||||
|
||||
if "Content-Type:" in data[0]:
|
||||
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:
|
||||
|
@ -91,6 +100,7 @@ def main(filename, keep, download, send_method, push_to_ynab):
|
|||
)
|
||||
filename = LclDownloader().download()
|
||||
processor = LclProcessor(filename)
|
||||
processor.parse_file()
|
||||
if send_method:
|
||||
processor.send_reconciled_amount(send_method)
|
||||
if push_to_ynab:
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
[tool.poetry]
|
||||
name = "ofx-processor"
|
||||
version = "4.5.1"
|
||||
version = "4.5.2"
|
||||
description = "Personal ofx processor"
|
||||
readme = "README.md"
|
||||
authors = ["Gabriel Augendre <gabriel@augendre.info>"]
|
||||
|
|
Loading…
Reference in a new issue