feat(mma): Implement track progress calculation and refactor get_all_tracks

This commit is contained in:
2026-03-07 11:24:05 -05:00
parent 34673ee32d
commit 87902d82d8
2 changed files with 85 additions and 4 deletions

View File

@@ -318,10 +318,10 @@ def get_all_tracks(base_dir: Union[str, Path] = ".") -> list[dict[str, Any]]:
track_info["id"] = state.metadata.id or track_id
track_info["title"] = state.metadata.name or track_id
track_info["status"] = state.metadata.status or "unknown"
track_info["complete"] = len([t for t in state.tasks if t.status == "completed"])
track_info["total"] = len(state.tasks)
if track_info["total"] > 0:
track_info["progress"] = track_info["complete"] / track_info["total"]
progress = calculate_track_progress(state.tasks)
track_info["complete"] = progress["completed"]
track_info["total"] = progress["total"]
track_info["progress"] = progress["percentage"] / 100.0
state_found = True
except Exception:
pass
@@ -352,3 +352,35 @@ def get_all_tracks(base_dir: Union[str, Path] = ".") -> list[dict[str, Any]]:
pass
results.append(track_info)
return results
def calculate_track_progress(tickets: list) -> dict:
"""
Calculates track progress based on ticket statuses.
percentage (float), completed (int), total (int), in_progress (int), blocked (int), todo (int)
"""
total = len(tickets)
if total == 0:
return {
"percentage": 0.0,
"completed": 0,
"total": 0,
"in_progress": 0,
"blocked": 0,
"todo": 0
}
completed = sum(1 for t in tickets if t.status == "completed")
in_progress = sum(1 for t in tickets if t.status == "in_progress")
blocked = sum(1 for t in tickets if t.status == "blocked")
todo = sum(1 for t in tickets if t.status == "todo")
percentage = (completed / total) * 100.0
return {
"percentage": float(percentage),
"completed": completed,
"total": total,
"in_progress": in_progress,
"blocked": blocked,
"todo": todo
}

View File

@@ -0,0 +1,49 @@
import pytest
from src.project_manager import calculate_track_progress
from src.models import Ticket
def test_calculate_track_progress_empty():
results = calculate_track_progress([])
assert results["total"] == 0
assert results["percentage"] == 0.0
assert results["completed"] == 0
assert results["in_progress"] == 0
assert results["blocked"] == 0
assert results["todo"] == 0
def test_calculate_track_progress_all_todo():
tickets = [
Ticket(id="1", description="desc 1", status="todo"),
Ticket(id="2", description="desc 2", status="todo")
]
results = calculate_track_progress(tickets)
assert results["total"] == 2
assert results["percentage"] == 0.0
assert results["completed"] == 0
assert results["todo"] == 2
def test_calculate_track_progress_mixed():
tickets = [
Ticket(id="1", description="desc 1", status="completed"),
Ticket(id="2", description="desc 2", status="in_progress"),
Ticket(id="3", description="desc 3", status="blocked"),
Ticket(id="4", description="desc 4", status="todo")
]
results = calculate_track_progress(tickets)
assert results["total"] == 4
assert results["completed"] == 1
assert results["in_progress"] == 1
assert results["blocked"] == 1
assert results["todo"] == 1
assert results["percentage"] == 25.0
def test_calculate_track_progress_all_completed():
tickets = [
Ticket(id="1", description="desc 1", status="completed"),
Ticket(id="2", description="desc 2", status="completed")
]
results = calculate_track_progress(tickets)
assert results["total"] == 2
assert results["percentage"] == 100.0
assert results["completed"] == 2