@opslane/claude-code-game 0.1.0

package/levels/09-plugins/exercise/models.py

@@ -0,0 +1,116 @@
+"""Expense data models and core operations."""
+
+from dataclasses import dataclass
+from datetime import datetime
+from typing import Optional
+import uuid
+from database import save_expense, load_expenses, remove_expense, get_expense_by_id
+
+
+@dataclass
+class Expense:
+    """Represents a single expense entry."""
+    id: str
+    amount: float
+    category: str
+    description: str
+    date: datetime
+    recurring: bool = False
+    recurring_frequency: Optional[str] = None  # 'daily', 'weekly', 'monthly'
+
+
+def generate_id() -> str:
+    """Generate a unique expense ID."""
+    return str(uuid.uuid4())[:8]
+
+
+def add_expense(
+    amount: float,
+    category: str,
+    description: str,
+    date: Optional[datetime] = None
+) -> Expense:
+    """Add a new expense with validation."""
+    if amount <= 0:
+        raise ValueError("Amount must be greater than 0")
+
+    if not category or not category.strip():
+        raise ValueError("Category cannot be empty")
+
+    expense = Expense(
+        id=generate_id(),
+        amount=amount,
+        category=category.strip(),
+        description=description,
+        date=date or datetime.now()
+    )
+    save_expense(expense)
+    return expense
+
+
+def list_expenses(
+    category: Optional[str] = None,
+    month: Optional[str] = None,
+    limit: int = 100
+) -> list[Expense]:
+    """List expenses with optional filters."""
+    expenses = load_expenses()
+
+    if category:
+        category_lower = category.lower()
+        expenses = [e for e in expenses if e.category.lower() == category_lower]
+
+    if month:
+        expenses = [e for e in expenses if e.date.strftime('%Y-%m') == month]
+
+    expenses.sort(key=lambda x: x.date, reverse=True)
+    return expenses[:limit]
+
+
+def get_expense(expense_id: str) -> Optional[Expense]:
+    """Get a single expense by ID.
+
+    Args:
+        expense_id: The unique expense ID
+
+    Returns:
+        The Expense if found, None otherwise
+    """
+    return get_expense_by_id(expense_id)
+
+
+def delete_expense(expense_id: str) -> bool:
+    """Delete an expense by ID.
+
+    Args:
+        expense_id: The unique expense ID
+
+    Returns:
+        True if deleted, False if not found
+    """
+    return remove_expense(expense_id)
+
+
+def get_total_by_category(category: str) -> float:
+    """Get total spent in a category.
+
+    Args:
+        category: The category name
+
+    Returns:
+        Total amount spent in that category
+
+    Note: Also case-sensitive - shares bug with list_expenses.
+    """
+    expenses = list_expenses(category=category)
+    return sum(e.amount for e in expenses)
+
+
+def get_categories() -> list[str]:
+    """Get all unique categories from expenses.
+
+    Returns:
+        List of category names
+    """
+    expenses = load_expenses()
+    return list(set(e.category for e in expenses))

package/levels/09-plugins/exercise/pyproject.toml

@@ -0,0 +1,52 @@
+[project]
+name = "expense-tracker"
+version = "0.1.0"
+description = "A simple CLI expense tracker for the Claude Code course"
+readme = "README.md"
+requires-python = ">=3.10"
+license = { text = "MIT" }
+
+# No runtime dependencies - uses only stdlib
+dependencies = []
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0",
+    "ruff>=0.1.0",
+]
+
+[project.scripts]
+expense = "main:main"
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+python_functions = ["test_*"]
+addopts = "-v --tb=short"
+filterwarnings = [
+    "ignore::DeprecationWarning",
+]
+
+[tool.ruff]
+line-length = 100
+target-version = "py310"
+
+[tool.ruff.lint]
+select = [
+    "E",      # pycodestyle errors
+    "F",      # pyflakes
+    "I",      # isort
+    "N",      # pep8-naming
+    "W",      # pycodestyle warnings
+    "UP",     # pyupgrade
+]
+ignore = [
+    "E501",   # Line too long (handled by formatter)
+]
+
+[tool.ruff.lint.isort]
+known-first-party = ["models", "database", "utils", "reports"]
+
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"

package/levels/09-plugins/exercise/reports.py

@@ -0,0 +1,63 @@
+"""Report generation for expense tracker."""
+
+from datetime import datetime
+from pathlib import Path
+from typing import Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from models import Expense
+
+
+def generate_monthly_report(month: Optional[str] = None) -> str:
+    """Generate a monthly expense report."""
+    from models import list_expenses
+    from utils import format_currency
+
+    if month is None:
+        month = datetime.now().strftime("%Y-%m")
+
+    expenses = list_expenses(month=month)
+
+    if not expenses:
+        return f"# Expense Report: {month}\n\nNo expenses found."
+
+    total = sum(e.amount for e in expenses)
+    breakdown = get_category_breakdown(expenses)
+    top = get_top_expenses(expenses, 5)
+
+    report = f"# Expense Report: {month}\n\n"
+    report += f"## Summary\n"
+    report += f"- **Total Spent:** {format_currency(total)}\n"
+    report += f"- **Number of Expenses:** {len(expenses)}\n\n"
+
+    report += "## By Category\n"
+    for cat, amount in sorted(breakdown.items(), key=lambda x: x[1], reverse=True):
+        pct = (amount / total) * 100
+        report += f"- {cat}: {format_currency(amount)} ({pct:.1f}%)\n"
+
+    report += "\n## Top 5 Expenses\n"
+    for i, exp in enumerate(top, 1):
+        report += f"{i}. {format_currency(exp.amount)} - {exp.description} ({exp.category})\n"
+
+    return report
+
+
+def get_category_breakdown(expenses: list["Expense"]) -> dict[str, float]:
+    """Get spending breakdown by category."""
+    breakdown: dict[str, float] = {}
+    for exp in expenses:
+        breakdown[exp.category] = breakdown.get(exp.category, 0) + exp.amount
+    return breakdown
+
+
+def get_top_expenses(expenses: list["Expense"], n: int = 5) -> list["Expense"]:
+    """Get the top N expenses by amount."""
+    return sorted(expenses, key=lambda x: x.amount, reverse=True)[:n]
+
+
+def save_report(report: str, filename: str) -> Path:
+    """Save report to a file."""
+    path = Path("reports") / filename
+    path.parent.mkdir(exist_ok=True)
+    path.write_text(report)
+    return path

package/levels/09-plugins/exercise/seed_data.py

@@ -0,0 +1,91 @@
+#!/usr/bin/env python3
+"""Seed the database with sample expenses for learning.
+
+Run this script to populate the database with example data:
+    python seed_data.py
+
+This creates a realistic set of expenses for students to work with.
+"""
+
+from datetime import datetime
+from database import init_db, clear_all_expenses
+from models import add_expense
+
+# Sample expenses for the course
+# Note: One entry has lowercase "food" to trigger the case-sensitivity bug
+SAMPLE_EXPENSES = [
+    # January 2026 - Primary month for exercises
+    (45.50, "Food", "Grocery shopping at Whole Foods", "2026-01-15"),
+    (12.00, "Transport", "Uber to airport", "2026-01-14"),
+    (89.99, "Shopping", "Running shoes from Nike", "2026-01-10"),
+    (150.00, "Bills", "Electric bill", "2026-01-01"),
+    (8.50, "food", "Coffee at Starbucks", "2026-01-20"),  # lowercase - triggers bug!
+    (35.00, "Entertainment", "Movie tickets for two", "2026-01-18"),
+    (22.00, "Food", "Lunch with team", "2026-01-19"),
+    (65.00, "Health", "Gym membership monthly", "2026-01-01"),
+    (15.75, "Food", "Thai takeout dinner", "2026-01-17"),
+    (42.00, "Transport", "Weekly metro pass", "2026-01-13"),
+    (28.99, "Entertainment", "Netflix + Spotify", "2026-01-05"),
+    (120.00, "Shopping", "Winter jacket on sale", "2026-01-08"),
+
+    # December 2025 - For month filtering tests
+    (200.00, "Shopping", "Holiday gifts for family", "2025-12-20"),
+    (55.00, "Food", "Holiday dinner groceries", "2025-12-25"),
+    (75.00, "Entertainment", "Concert tickets", "2025-12-15"),
+    (30.00, "Transport", "Airport parking", "2025-12-23"),
+
+    # November 2025 - Additional historical data
+    (95.00, "Bills", "Internet bill", "2025-11-15"),
+    (180.00, "Health", "Doctor visit copay", "2025-11-10"),
+    (45.00, "Food", "Birthday dinner", "2025-11-22"),
+]
+
+
+def seed_database(verbose: bool = True) -> int:
+    """Initialize and seed the database with sample data.
+
+    Args:
+        verbose: Whether to print progress messages
+
+    Returns:
+        Number of expenses added
+    """
+    if verbose:
+        print("Initializing database...")
+    init_db()
+
+    if verbose:
+        print("Clearing existing data...")
+    clear_all_expenses()
+
+    if verbose:
+        print("Adding sample expenses...")
+
+    count = 0
+    for amount, category, desc, date_str in SAMPLE_EXPENSES:
+        date = datetime.strptime(date_str, "%Y-%m-%d")
+        expense = add_expense(amount, category, desc, date)
+        count += 1
+        if verbose:
+            print(f"  Added: {expense.id} - ${amount:.2f} - {desc[:40]}")
+
+    if verbose:
+        print(f"\n{'='*50}")
+        print(f"Seeded {count} expenses successfully!")
+        print(f"{'='*50}")
+        print("\nTry these commands:")
+        print("  python main.py list")
+        print("  python main.py list --category Food")
+        print("  python main.py summary")
+        print("  pytest -v")
+
+    return count
+
+
+def main():
+    """Entry point for seeding script."""
+    seed_database()
+
+
+if __name__ == "__main__":
+    main()

package/levels/09-plugins/exercise/tests/__init__.py

@@ -0,0 +1 @@
+"""Test package for expense tracker."""

package/levels/09-plugins/exercise/tests/conftest.py

@@ -0,0 +1,69 @@
+"""Pytest configuration and shared fixtures."""
+
+import os
+import sys
+import pytest
+from pathlib import Path
+
+# Add parent directory to path so we can import our modules
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+# Use a separate test database
+TEST_DB_PATH = Path(__file__).parent / "test_expenses.db"
+os.environ["EXPENSE_DB_PATH"] = str(TEST_DB_PATH)
+
+
+@pytest.fixture(autouse=True)
+def clean_test_db():
+    """Ensure clean database for each test.
+
+    This fixture runs automatically before and after each test.
+    It initializes the database and cleans up after.
+    """
+    from database import init_db, clear_all_expenses
+
+    # Setup: initialize fresh database
+    init_db()
+    clear_all_expenses()
+
+    yield  # Run the test
+
+    # Teardown: clean up
+    clear_all_expenses()
+
+
+@pytest.fixture
+def sample_expenses():
+    """Create a set of sample expenses for testing.
+
+    Returns:
+        List of created Expense objects
+    """
+    from datetime import datetime
+    from models import add_expense
+
+    expenses = [
+        add_expense(50.00, "Food", "Groceries", datetime(2026, 1, 15)),
+        add_expense(30.00, "Transport", "Uber ride", datetime(2026, 1, 14)),
+        add_expense(25.00, "Food", "Lunch", datetime(2026, 1, 16)),
+        add_expense(100.00, "Shopping", "Clothes", datetime(2026, 1, 10)),
+        add_expense(15.00, "Entertainment", "Movie", datetime(2026, 1, 12)),
+    ]
+    return expenses
+
+
+@pytest.fixture
+def mixed_case_expenses():
+    """Create expenses with mixed case categories for bug testing.
+
+    This fixture specifically tests the case-sensitivity bug.
+    """
+    from datetime import datetime
+    from models import add_expense
+
+    expenses = [
+        add_expense(50.00, "Food", "Groceries", datetime(2026, 1, 15)),
+        add_expense(25.00, "food", "Coffee", datetime(2026, 1, 16)),  # lowercase
+        add_expense(30.00, "FOOD", "Dinner", datetime(2026, 1, 17)),  # uppercase
+    ]
+    return expenses

package/levels/09-plugins/exercise/tests/test_database.py

@@ -0,0 +1,244 @@
+"""Tests for database operations."""
+
+import pytest
+from datetime import datetime
+from database import (
+    init_db,
+    save_expense,
+    load_expenses,
+    get_expense_by_id,
+    remove_expense,
+    clear_all_expenses,
+    get_expense_count
+)
+from models import Expense
+
+
+class TestDatabaseInit:
+    """Tests for database initialization."""
+
+    def test_init_creates_table(self):
+        """Test that init_db creates the expenses table."""
+        init_db()
+        # If we get here without error, table was created
+        expenses = load_expenses()
+        assert isinstance(expenses, list)
+
+    def test_init_idempotent(self):
+        """Test that init_db can be called multiple times safely."""
+        init_db()
+        init_db()
+        init_db()
+        # Should not raise any errors
+
+
+class TestSaveAndLoad:
+    """Tests for saving and loading expenses."""
+
+    def test_save_and_load_single(self):
+        """Test saving and loading a single expense."""
+        expense = Expense(
+            id="test123",
+            amount=50.0,
+            category="Food",
+            description="Test expense",
+            date=datetime(2026, 1, 15, 12, 0)
+        )
+        save_expense(expense)
+
+        expenses = load_expenses()
+        assert len(expenses) == 1
+        assert expenses[0].id == "test123"
+        assert expenses[0].amount == 50.0
+        assert expenses[0].category == "Food"
+
+    def test_save_multiple(self):
+        """Test saving multiple expenses."""
+        for i in range(5):
+            expense = Expense(
+                id=f"test{i}",
+                amount=float(i * 10),
+                category="Test",
+                description=f"Expense {i}",
+                date=datetime.now()
+            )
+            save_expense(expense)
+
+        expenses = load_expenses()
+        assert len(expenses) == 5
+
+    def test_load_preserves_data(self):
+        """Test that all expense fields are preserved on load."""
+        original = Expense(
+            id="preserve123",
+            amount=99.99,
+            category="Shopping",
+            description="Test all fields",
+            date=datetime(2026, 6, 15, 14, 30),
+            recurring=True,
+            recurring_frequency="monthly"
+        )
+        save_expense(original)
+
+        loaded = load_expenses()[0]
+        assert loaded.id == original.id
+        assert loaded.amount == original.amount
+        assert loaded.category == original.category
+        assert loaded.description == original.description
+        assert loaded.date == original.date
+        assert loaded.recurring == original.recurring
+        assert loaded.recurring_frequency == original.recurring_frequency
+
+    def test_load_ordered_by_date(self):
+        """Test that load_expenses returns results ordered by date descending."""
+        dates = [
+            datetime(2026, 1, 1),
+            datetime(2026, 1, 15),
+            datetime(2026, 1, 10),
+        ]
+        for i, date in enumerate(dates):
+            expense = Expense(
+                id=f"date{i}",
+                amount=10.0,
+                category="Test",
+                description=f"Expense {i}",
+                date=date
+            )
+            save_expense(expense)
+
+        expenses = load_expenses()
+        loaded_dates = [e.date for e in expenses]
+        assert loaded_dates == sorted(loaded_dates, reverse=True)
+
+
+class TestGetExpenseById:
+    """Tests for getting expense by ID."""
+
+    def test_get_existing(self):
+        """Test retrieving an existing expense by ID."""
+        expense = Expense(
+            id="findme123",
+            amount=25.0,
+            category="Transport",
+            description="Bus fare",
+            date=datetime.now()
+        )
+        save_expense(expense)
+
+        result = get_expense_by_id("findme123")
+        assert result is not None
+        assert result.id == "findme123"
+        assert result.description == "Bus fare"
+
+    def test_get_nonexistent(self):
+        """Test that non-existent ID returns None."""
+        result = get_expense_by_id("does-not-exist")
+        assert result is None
+
+    def test_get_after_multiple_saves(self):
+        """Test getting specific expense after saving multiple."""
+        for i in range(10):
+            expense = Expense(
+                id=f"multi{i}",
+                amount=float(i),
+                category="Test",
+                description=f"Expense {i}",
+                date=datetime.now()
+            )
+            save_expense(expense)
+
+        result = get_expense_by_id("multi5")
+        assert result is not None
+        assert result.amount == 5.0
+
+
+class TestRemoveExpense:
+    """Tests for removing expenses."""
+
+    def test_remove_existing(self):
+        """Test removing an existing expense."""
+        expense = Expense(
+            id="removeme",
+            amount=10.0,
+            category="Test",
+            description="To be removed",
+            date=datetime.now()
+        )
+        save_expense(expense)
+        assert get_expense_by_id("removeme") is not None
+
+        result = remove_expense("removeme")
+        assert result is True
+        assert get_expense_by_id("removeme") is None
+
+    def test_remove_nonexistent(self):
+        """Test removing non-existent expense returns False."""
+        result = remove_expense("fake-id")
+        assert result is False
+
+    def test_remove_only_target(self):
+        """Test that remove only deletes the target expense."""
+        for i in range(3):
+            expense = Expense(
+                id=f"keep{i}",
+                amount=float(i),
+                category="Test",
+                description=f"Expense {i}",
+                date=datetime.now()
+            )
+            save_expense(expense)
+
+        remove_expense("keep1")
+
+        expenses = load_expenses()
+        ids = [e.id for e in expenses]
+        assert "keep0" in ids
+        assert "keep1" not in ids
+        assert "keep2" in ids
+
+
+class TestClearAll:
+    """Tests for clearing all expenses."""
+
+    def test_clear_removes_all(self):
+        """Test that clear_all_expenses removes everything."""
+        for i in range(5):
+            expense = Expense(
+                id=f"clear{i}",
+                amount=float(i),
+                category="Test",
+                description=f"Expense {i}",
+                date=datetime.now()
+            )
+            save_expense(expense)
+
+        assert len(load_expenses()) == 5
+
+        clear_all_expenses()
+        assert len(load_expenses()) == 0
+
+    def test_clear_empty_db(self):
+        """Test that clearing empty database doesn't error."""
+        clear_all_expenses()  # Should not raise
+
+
+class TestExpenseCount:
+    """Tests for getting expense count."""
+
+    def test_count_empty(self):
+        """Test count on empty database."""
+        assert get_expense_count() == 0
+
+    def test_count_with_expenses(self):
+        """Test count with expenses."""
+        for i in range(7):
+            expense = Expense(
+                id=f"count{i}",
+                amount=float(i),
+                category="Test",
+                description=f"Expense {i}",
+                date=datetime.now()
+            )
+            save_expense(expense)
+
+        assert get_expense_count() == 7