Luís Murta
41d599c025
All parsers defined in parsers.py. main.py is the calling script and reader/writer.
113 lines
3.2 KiB
Python
113 lines
3.2 KiB
Python
from datetime import datetime
|
|
from decimal import Decimal, InvalidOperation
|
|
|
|
|
|
class Parser:
|
|
@staticmethod
|
|
def get_transactions(file, encoding, sep="\t"):
|
|
with open(file, newline="", encoding=encoding) as f:
|
|
transactions = [line.rstrip().split(sep) for line in f]
|
|
|
|
return transactions
|
|
|
|
def parse(self, file, encoding="utf-8"):
|
|
pass
|
|
|
|
|
|
class Bank1(Parser):
|
|
"""Bank 1 parser
|
|
|
|
Bank 1 transcripts have the following properties:
|
|
encoding: utf-8
|
|
separator: ;
|
|
starting line: 5
|
|
date format: %d/%m/%Y
|
|
"""
|
|
|
|
def parse(self, file, encoding="utf-8"):
|
|
transactions = []
|
|
|
|
for transaction in self.get_transactions(file, encoding, sep=";")[5:]:
|
|
transaction = [field.rstrip() for field in transaction]
|
|
date = datetime.strptime(transaction[1], "%d/%m/%Y").date()
|
|
description = " ".join(transaction[3].split())
|
|
value = Decimal(transaction[4])
|
|
|
|
transactions.append([date.isoformat(), description, "Bank1", value])
|
|
|
|
return transactions
|
|
|
|
|
|
class Bank2(Parser):
|
|
"""Bank 2 parser
|
|
|
|
Bank 2 transcripts have the following properties:
|
|
encoding: utf-8
|
|
separator: tab
|
|
date format: %d/%m/%Y
|
|
decimal separator: ,
|
|
|
|
Bank 2 also has an associated credit card, for which the transaction value
|
|
has to be negated.
|
|
"""
|
|
|
|
def __init__(self, cc=False):
|
|
self.cc = cc
|
|
|
|
def parse(self, file, encoding="utf-8"):
|
|
transactions = []
|
|
|
|
for transaction in self.get_transactions(file, encoding):
|
|
date = datetime.strptime(transaction[0], "%d/%m/%Y").date()
|
|
description = transaction[2]
|
|
try:
|
|
value = Decimal(transaction[3])
|
|
except InvalidOperation:
|
|
transaction[3] = transaction[3].replace(",", "")
|
|
value = Decimal(transaction[3])
|
|
|
|
if not self.cc:
|
|
card = "Bank2"
|
|
else:
|
|
value = -value
|
|
card = "Bank2 CC"
|
|
|
|
transactions.append([date.isoformat(), description, card, value])
|
|
|
|
return transactions
|
|
|
|
|
|
class Bank3(Parser):
|
|
"""Bank 3 parser
|
|
|
|
Bank 3 transcripts have the following properties:
|
|
encoding: windows-1252 (passed as argument)
|
|
separator: ;
|
|
starting line: 7
|
|
finishing line: -1
|
|
date format: %d-%m-%Y
|
|
decimal separator: ,
|
|
thousands separator: .
|
|
|
|
Bank 3 has credits in a different column from debits. These also have to be
|
|
negated.
|
|
"""
|
|
|
|
def parse(self, file, encoding="utf-8"):
|
|
transactions = []
|
|
|
|
for transaction in self.get_transactions(file, encoding, sep=";")[7:-1]:
|
|
transaction = [field.rstrip() for field in transaction]
|
|
date = datetime.strptime(transaction[1], "%d-%m-%Y").date()
|
|
description = transaction[2]
|
|
if t := transaction[3]:
|
|
t = t.replace(".", "").replace(",", ".")
|
|
value = -Decimal(t)
|
|
else:
|
|
t = transaction[4].replace(".", "").replace(",", ".")
|
|
value = Decimal(t)
|
|
|
|
transactions.append([date.isoformat(), description, "Bank3", value])
|
|
|
|
return transactions
|