def get_budget_sum(self, budget_id):
t = TransactionModel()
query = {"budget": budget_id}
budget_sum = 0
for elemet in t.get(query):
budget_sum += elemet["money"]
return budget_sum
def save_transaction(self):
try:
self.money = float(self.money)
except ValueError:
self.money = 0
try:
self.budget_id = int(self.budget_id)
except ValueError:
self.budget_id = 0
try:
self.category_id = int(self.category_id)
except ValueError:
self.category_id = 0
t = TransactionModel(self.owner, self.budget_id, self.category_id, self.money, self.description)
t.save()