|
| 1 | +class InvestmentTracker: |
| 2 | + # test |
| 3 | + def __init__(self): |
| 4 | + self.expenses = [] |
| 5 | + self.categories = set( |
| 6 | + ["food", "transport", "utilities", "entertainment", "other"] |
| 7 | + ) |
| 8 | + |
| 9 | + ## new and improved defs below |
| 10 | + |
| 11 | + def record_transaction(self, amount, category, description): |
| 12 | + """Add a new expense to the tracker.""" |
| 13 | + if not isinstance(amount, (int, float)) or amount <= 0: |
| 14 | + raise ValueError("Amount must be a positive number") |
| 15 | + |
| 16 | + if category.lower() not in self.categories: |
| 17 | + raise ValueError(f"Category must be one of: {', '.join(self.categories)}") |
| 18 | + |
| 19 | + expense = { |
| 20 | + "amount": amount, |
| 21 | + "category": category.lower(), |
| 22 | + "description": description, |
| 23 | + } |
| 24 | + self.expenses.append(expense) |
| 25 | + return True |
| 26 | + |
| 27 | + def calculate_overall_spending(self): |
| 28 | + """Calculate total expenses.""" |
| 29 | + return sum(expense["amount"] for expense in self.expenses) |
| 30 | + |
| 31 | + def filter_by_category(self, category): |
| 32 | + """Get all expenses for a specific category.""" |
| 33 | + if category.lower() not in self.categories: |
| 34 | + raise ValueError(f"Category must be one of: {', '.join(self.categories)}") |
| 35 | + |
| 36 | + return [ |
| 37 | + expense |
| 38 | + for expense in self.expenses |
| 39 | + if expense["category"] == category.lower() |
| 40 | + ] |
| 41 | + |
| 42 | + def compute_category_sum(self, category): |
| 43 | + """Get total expenses for a specific category.""" |
| 44 | + if category.lower() not in self.categories: |
| 45 | + raise ValueError(f"Category must be one of: {', '.join(self.categories)}") |
| 46 | + |
| 47 | + return sum( |
| 48 | + expense["amount"] |
| 49 | + for expense in self.expenses |
| 50 | + if expense["category"] == category.lower() |
| 51 | + ) |
| 52 | + |
| 53 | + def register_new_category(self, category): |
| 54 | + """Add a new expense category.""" |
| 55 | + if not isinstance(category, str) or not category.strip(): |
| 56 | + raise ValueError("Category must be a non-empty string") |
| 57 | + |
| 58 | + category = category.lower().strip() |
| 59 | + if category in self.categories: |
| 60 | + return False |
| 61 | + |
| 62 | + self.categories.add(category) |
| 63 | + return True |
| 64 | + |
| 65 | + |
| 66 | +def main(): |
| 67 | + # Example usage |
| 68 | + tracker = InvestmentTracker() |
| 69 | + |
| 70 | + # Add some sample expenses |
| 71 | + tracker.record_transaction(25.50, "food", "Lunch at cafe") |
| 72 | + tracker.record_transaction(35.00, "transport", "Uber ride") |
| 73 | + tracker.record_transaction(150.00, "utilities", "Electricity bill") |
| 74 | + |
| 75 | + # Print total expenses |
| 76 | + print(f"Total expenses: ${tracker.calculate_overall_spending():.2f}") |
| 77 | + |
| 78 | + # Print food expenses |
| 79 | + food_expenses = tracker.filter_by_category("food") |
| 80 | + print("\nFood expenses:") |
| 81 | + for expense in food_expenses: |
| 82 | + print(f"${expense['amount']:.2f} - {expense['description']}") |
| 83 | + |
| 84 | + |
| 85 | +if __name__ == "__main__": |
| 86 | + main() |
0 commit comments