Luís Murta
647af8be5b
Also checks for transactions before loop incase of months without transactions.
86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
from .categories import (
|
|
get_income_categories,
|
|
get_fixed_expenses,
|
|
get_required_expenses,
|
|
get_health_expenses,
|
|
get_discretionary_expenses,
|
|
)
|
|
from .transactions import (
|
|
load_transactions,
|
|
daterange,
|
|
by_category,
|
|
by_month,
|
|
by_month_and_category,
|
|
by_year_and_category,
|
|
)
|
|
|
|
|
|
def net(state, start=None, end=None):
|
|
transactions = load_transactions(state.data_dir)
|
|
if not start:
|
|
start = transactions[0].date
|
|
if not end:
|
|
end = transactions[-1].date
|
|
|
|
income, fixed, required, health, discretionary = [], [], [], [], []
|
|
yearly_transactions_by_categories = by_year_and_category(transactions, start, end)
|
|
for _, transactions_by_category in yearly_transactions_by_categories.items():
|
|
income.append(
|
|
sum(
|
|
float(t.value)
|
|
for category, transactions in transactions_by_category.items()
|
|
if transactions and category in get_income_categories()
|
|
for t in transactions
|
|
)
|
|
)
|
|
fixed.append(
|
|
sum(
|
|
-float(t.value)
|
|
for category, transactions in transactions_by_category.items()
|
|
if transactions and category in get_fixed_expenses()
|
|
for t in transactions
|
|
)
|
|
)
|
|
required.append(
|
|
sum(
|
|
-float(t.value)
|
|
for category, transactions in transactions_by_category.items()
|
|
if transactions and category in get_required_expenses()
|
|
for t in transactions
|
|
)
|
|
)
|
|
health.append(
|
|
sum(
|
|
-float(t.value)
|
|
for category, transactions in transactions_by_category.items()
|
|
if transactions and category in get_health_expenses()
|
|
for t in transactions
|
|
)
|
|
)
|
|
discretionary.append(
|
|
sum(
|
|
-float(t.value)
|
|
for category, transactions in transactions_by_category.items()
|
|
if transactions and category in get_discretionary_expenses()
|
|
for t in transactions
|
|
)
|
|
)
|
|
|
|
for i, year in enumerate(yearly_transactions_by_categories.keys()):
|
|
print(year)
|
|
print(
|
|
"Income: {:.2f}, Expenses: {:.2f}, Net: {:.2f}\n"
|
|
"Fixed Expenses: {:.2f}\n"
|
|
"Required Expenses: {:.2f}\n"
|
|
"Health Expenses: {:.2f}\n"
|
|
"Discretionary Expenses: {:.2f}\n".format(
|
|
income[i],
|
|
fixed[i] + required[i] + health[i] + discretionary[i],
|
|
income[i] - (fixed[i] + required[i] + health[i] + discretionary[i]),
|
|
fixed[i],
|
|
required[i],
|
|
health[i],
|
|
discretionary[i],
|
|
)
|
|
)
|