ofx-processor/ofx_processor/processors/revolut.py

66 lines
1.7 KiB
Python
Raw Normal View History

import csv
2020-02-26 19:59:31 +01:00
import sys
import click
import dateparser
from ofx_processor.utils.base_processor import BaseProcessor, BaseLine
def _amount_str_to_float(amount):
if amount:
return float(amount.replace(",", "."))
return ""
class RevolutLine(BaseLine):
def _process_inflow(self):
return _amount_str_to_float(self.data.get("Paid In (EUR)"))
def _process_outflow(self):
return _amount_str_to_float(self.data.get("Paid Out (EUR)"))
def get_date(self):
return dateparser.parse(self.data.get("Completed Date")).strftime("%Y-%m-%d")
def get_amount(self):
outflow = self._process_outflow()
inflow = self._process_inflow()
amount = -outflow if outflow else inflow
amount = int(amount * 1000)
return amount
def get_memo(self):
return " - ".join(
filter(
None,
map(
str.strip,
[self.data.get("Category", ""), self.data.get("Exchange Rate", "")],
),
)
)
def get_payee(self):
return self.data.get("Reference")
class RevolutProcessor(BaseProcessor):
line_class = RevolutLine
account_name = "revolut"
command_name = "revolut"
def parse_file(self):
2020-02-26 19:59:31 +01:00
try:
2020-02-26 20:07:33 +01:00
with open(self.filename, encoding="utf-8") as f:
2020-02-26 19:59:31 +01:00
reader = csv.DictReader(f, delimiter=";")
return [line for line in reader]
except FileNotFoundError:
click.secho("File not found", fg="red")
sys.exit(1)
def main(filename, keep):
"""Import Revolut bank statement (CSV file)."""
RevolutProcessor(filename).push_to_ynab(keep)