personal-finance/budget
2
1
Fork 0
Code Issues 11 Pull Requests Projects 6 Releases 2 Wiki Activity

Removes unnecessary Transactions class

Browse Source
This commit is contained in:
Luís Murta 2021-02-02 22:07:17 +00:00
parent 311c5358f1
commit ff324ca0a6
Signed by: satprog
GPG Key ID: DDF2EFC6179009DC
1 changed files with 4 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pfbudget/transactions.py
View File

@ -102,23 +102,8 @@ class Transaction:
) )
class Transactions(list): def sort_by_bank(transactions: list):
def sort_by_bank(self): transactions.sort(key=lambda k: k.bank)
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 daterange(start, end, period): def daterange(start, end, period):
@ -187,8 +172,8 @@ def by_month_and_category(transactions, start, end) -> dict:
return monthly_transactions_by_categories return monthly_transactions_by_categories
def load_transactions(data_dir) -> Transactions: def load_transactions(data_dir) -> list:
transactions = Transactions() transactions = []
for df in Path(data_dir).iterdir(): for df in Path(data_dir).iterdir():
try: try:
trs = read_transactions(df) trs = read_transactions(df)