expense_tracker.py

from expense import Expense
import calendar
import datetime


def main():
    print("Running expense tracker!")
    expense_file_path = "expenses.csv"
    budget = 2000
    
    # Get the user to input their expenses.
    expense = get_user_expense()
    
    # Write the expenses to a file.
    save_expense_to_file(expense, expense_file_path)
    
    # Read the file and summarize the expenses.
    summarize_expenses(expense_file_path, budget)


def get_user_expense():
    print("Getting user expense.")
    expense_name = input("Enter the expense name: ")
    expense_amount = float(input("Enter the expense amount: ")) 
    expense_categories = [
        "Food", 
        "Rent",
        "Work",
        "Fun",
        "Miscellaneous"
    ]

    while True:
        print("Select a category: ")
        for i, category_name in enumerate(expense_categories):
            print(f"  {i + 1}. {category_name}")
        
        value_range = f"[1 - {len(expense_categories)}]"
        selected_index = int(input(f"Enter a category number {value_range}: "))
        
        if selected_index in range(1, len(expense_categories) + 1):
            selected_category = expense_categories[selected_index - 1]
            new_expense = Expense(
                name = expense_name, category = selected_category, amount = expense_amount)
            return new_expense
        else:
            print("Invalid category. Please try again.")


def save_expense_to_file(expense, expense_file_path):
    print(f"Saving user expense: {expense} to {expense_file_path}.")
    with open(expense_file_path, "a") as f:
        f.write(f"{expense.name},{expense.amount},{expense.category}\n")


def summarize_expenses(expense_file_path, budget):
    print("Summarizing user expense.")
    expenses: list[Expense] = []
    with open(expense_file_path, "r") as f:
        lines = f.readlines()
        for line in lines:
            expense_name, expense_amount, expense_category = line.strip().split(",")
            line_expense = Expense(
                name=expense_name, 
                amount=float(expense_amount),
                category=expense_category
            )
            expenses.append(line_expense)
                
    amount_by_category = {}
    for expense in expenses:
        key = expense.category
        if key in amount_by_category:
            amount_by_category[key] += expense.amount
        else:
            amount_by_category[key] = expense.amount
        
    print("Expenses By Category:")
    for key, amount in amount_by_category.items():
        print(f"  {key}: ${amount:.2f}")

    total_spent = sum([ex.amount for ex in expenses])
    print(f"Total spent: ${total_spent:.2f}")

    remaining_budget = budget - total_spent
    print(f"Budget Remaining: ${remaining_budget:.2f}")

    now = datetime.datetime.now()
    days_in_month = calendar.monthrange(now.year, now.month)[1]
    remaining_days = days_in_month - now.day

    daily_budget = remaining_budget / remaining_days
    print(green(f"Budget per day: ${daily_budget:.2f}"))


def green(text):
    return f"\033[92m{text}\033[0m"

if __name__ == "__main__":
    main()