app.py

import pprint
from datetime import datetime
import json
from pathlib import Path


def menu():
    options = [
        "0. Add an expense",
        "1. View all expenses",
        "2. Summary by Category",
        "3. Quit",
    ]
    for option in options:
        print(option)


def main():
    try:
        with open("data.json", "r") as f:
            expenses = json.load(f)
    except (FileNotFoundError, json.JSONDecodeError):
        expenses = {}

    while True:
        menu()
        choice = int(input("Enter option number: "))
        if choice >= 0 and choice <= 4:
            if choice == 0:
                while True:
                    category = input("Enter category for the expense: ")
                    if category.lower() == "q":
                        break
                    amount = float(input("Enter the amount of expense: "))

                    expenses.setdefault(category, []).append(
                        {"amount": amount, "date": datetime.now().isoformat()}
                    )
                    with open("data.json", "w") as f:
                        json.dump(expenses, f)
                    print(f"Expense added to {category}!")
                    break
            elif choice == 1:
                for category in expenses:
                    print(f"{category}:")
                    for index in range(len(expenses[category])):
                        print(expenses[category][index])
            elif choice == 2:
                for category in expenses:
                    total = 0
                    for item in expenses[category]:
                        total += item["amount"]
                    print(f"{category}: {total}$")
            elif choice == 3:
                exit()

        else:
            continue


main()