67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
"""Utilities for rendering execution plan information in CLI output."""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any
|
||
|
||
|
||
def fallback_plan_summary(execution_plan: Any) -> str:
|
||
"""Build a short plan summary from summary and summary_by_reason.
|
||
|
||
This is used when execution_plan.human_summary is missing or empty.
|
||
"""
|
||
summary = getattr(execution_plan, "summary", {}) or {}
|
||
summary_by_reason = getattr(execution_plan, "summary_by_reason", {}) or {}
|
||
operations = getattr(execution_plan, "operations", []) or []
|
||
|
||
total = summary.get("total", len(operations))
|
||
parts = [
|
||
f"计划操作统计：共 {total} 条（move {summary.get('move', 0)}，rename {summary.get('rename', 0)}，"
|
||
f"quarantine {summary.get('quarantine', 0)}，no-op {summary.get('no-op', 0)}）"
|
||
]
|
||
if summary_by_reason:
|
||
parts.append(
|
||
"原因分布：" + "；".join(f"{reason}: {count}" for reason, count in list(summary_by_reason.items())[:8])
|
||
)
|
||
return "\n".join(parts)
|
||
|
||
|
||
def preferred_plan_summary(execution_plan: Any) -> str:
|
||
"""Return human_summary if available, otherwise fallback summary."""
|
||
human_summary = getattr(execution_plan, "human_summary", "")
|
||
if isinstance(human_summary, str) and human_summary.strip():
|
||
return human_summary
|
||
return fallback_plan_summary(execution_plan)
|
||
|
||
|
||
def render_review_preview(
|
||
rows: list[dict[str, str]],
|
||
preview_limit: int = 10,
|
||
show_all: bool = False,
|
||
) -> tuple[list[str], int]:
|
||
"""Render high-risk review rows for console preview.
|
||
|
||
Returns a tuple of (lines, remaining_count).
|
||
"""
|
||
if preview_limit < 1:
|
||
raise ValueError("preview_limit must be >= 1")
|
||
|
||
selected = rows if show_all else rows[:preview_limit]
|
||
remaining = 0 if show_all else max(0, len(rows) - len(selected))
|
||
|
||
lines: list[str] = []
|
||
for row in selected:
|
||
idx = row.get("index", "?")
|
||
operation_type = row.get("operation_type", "")
|
||
flags = row.get("risk_flags", "") or "none"
|
||
source_path = row.get("source_path", "")
|
||
destination_path = row.get("destination_path", "")
|
||
reason = row.get("reason", "")
|
||
|
||
lines.append(f" - [{idx}] {operation_type} | flags={flags}")
|
||
lines.append(f" source: {source_path}")
|
||
lines.append(f" destination: {destination_path or '(none)'}")
|
||
lines.append(f" reason: {reason}")
|
||
|
||
return lines, remaining
|