feat(mma): Implement track progress calculation and refactor get_all_tracks
This commit is contained in:
@@ -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["id"] = state.metadata.id or track_id
|
||||||
track_info["title"] = state.metadata.name or track_id
|
track_info["title"] = state.metadata.name or track_id
|
||||||
track_info["status"] = state.metadata.status or "unknown"
|
track_info["status"] = state.metadata.status or "unknown"
|
||||||
track_info["complete"] = len([t for t in state.tasks if t.status == "completed"])
|
progress = calculate_track_progress(state.tasks)
|
||||||
track_info["total"] = len(state.tasks)
|
track_info["complete"] = progress["completed"]
|
||||||
if track_info["total"] > 0:
|
track_info["total"] = progress["total"]
|
||||||
track_info["progress"] = track_info["complete"] / track_info["total"]
|
track_info["progress"] = progress["percentage"] / 100.0
|
||||||
state_found = True
|
state_found = True
|
||||||
except Exception:
|
except Exception:
|
||||||
pass
|
pass
|
||||||
@@ -352,3 +352,35 @@ def get_all_tracks(base_dir: Union[str, Path] = ".") -> list[dict[str, Any]]:
|
|||||||
pass
|
pass
|
||||||
results.append(track_info)
|
results.append(track_info)
|
||||||
return results
|
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
|
||||||
|
}
|
||||||
|
|||||||
49
tests/test_progress_viz.py
Normal file
49
tests/test_progress_viz.py
Normal 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
|
||||||
|
|
||||||
Reference in New Issue
Block a user