From ff324ca0a61bedaa0092c0df62cb4e9c3c525d75 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lu=C3=ADs=20Murta?= Date: Tue, 2 Feb 2021 22:07:17 +0000 Subject: [PATCH] Removes unnecessary Transactions class --- pfbudget/transactions.py | 23 ++++------------------- 1 file changed, 4 insertions(+), 19 deletions(-) diff --git a/pfbudget/transactions.py b/pfbudget/transactions.py index fd0f3a2..cd8c1da 100644 --- a/pfbudget/transactions.py +++ b/pfbudget/transactions.py @@ -102,23 +102,8 @@ class Transaction: ) -class Transactions(list): - def sort_by_bank(self): - self.sort(key=lambda k: k.bank) - - def get_transactions_by_year(self, start=None, end=None): - if not start: - start = self[0].date - if not end: - end = self[-1].date - - years = dict() - for year in range(start.year, end.year + 1): - years[year] = Transactions( - t for t in self if start <= t.date <= end and t.date.year == year - ) - - return years +def sort_by_bank(transactions: list): + transactions.sort(key=lambda k: k.bank) def daterange(start, end, period): @@ -187,8 +172,8 @@ def by_month_and_category(transactions, start, end) -> dict: return monthly_transactions_by_categories -def load_transactions(data_dir) -> Transactions: - transactions = Transactions() +def load_transactions(data_dir) -> list: + transactions = [] for df in Path(data_dir).iterdir(): try: trs = read_transactions(df)