This repository has been archived on 2023-11-10. You can view files and clone it, but cannot push or open issues or pull requests.
freecodecamp-projects/8-scientific-computing-python/3-budget-app/budget.py

94 lines
2.7 KiB
Python
Raw Normal View History

2022-11-27 11:57:33 +00:00
import math
class Category():
def __init__(self, name):
self.name = name
self.ledger = []
def __str__(self):
title = self.name.center(30, "*") + '\n'
items = ""
for item in self.ledger:
desc = item["description"]
amount = "{:0.2f}".format(float(item["amount"]))
if len(desc) > 23: desc = desc[0:23]
if len(amount) > 7: amount = amount[0:7]
items = items + desc + (30 - len(desc) - len(amount)) * " " + amount + '\n'
return title + items + "Total: " + str(float(self.get_balance()))
def deposit(self, amount, desc=""):
self.ledger.append({ "amount" : amount, "description": desc })
def withdraw(self, amount, desc=""):
if self.check_funds(amount):
amount = amount * - 1
self.ledger.append({ "amount" : amount, "description": desc })
return True
else:
return False
def get_balance(self):
balance = 0
for entry in self.ledger:
balance = balance + entry["amount"]
return balance
def transfer(self, amount, category):
if self.check_funds(amount):
self.withdraw(amount, "Transfer to " + category.name)
category.deposit(amount, "Transfer from " + self.name)
return True
else:
return False
def check_funds(self, amount):
if amount <= self.get_balance():
return True
else:
return False
def create_spend_chart(categories):
totals = {}
total = 0
chart = "Percentage spent by category\n"
for category in categories:
totals[category.name] = 0
for entry in category.ledger:
if entry["amount"] < 0:
totals[category.name] = totals[category.name] + entry["amount"]
for k, v in totals.items():
totals[k] = v * - 1
total = total - v
for k, v in totals.items():
percentage = math.trunc((v / total * 100))
totals[k] = percentage
for i in range(100, -10, -10):
chart = chart + str(i).rjust(3, " ") + "|"
for k, v in totals.items():
if v >= i:
chart = chart + " o "
else:
chart = chart + " "
chart = chart + " \n"
chart = chart + 4 * " " + (4 + 2 * len(totals)) * "-"
i = 0
while i < max(len(k) for k, v in totals.items()):
chart = chart + "\n" + 4 * " "
for key in totals:
if len(key) <= i:
chart = chart + " "
else:
chart = chart + " " + key[i] + " "
chart = chart + " "
i = i + 1
return chart