48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
|
"""Output helpers — step/ok/warn/die + aligned text table."""
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
def step(msg: str) -> None:
|
||
|
|
"""Print a step header."""
|
||
|
|
print(f"\n==> {msg}", flush=True)
|
||
|
|
|
||
|
|
|
||
|
|
def ok(msg: str) -> None:
|
||
|
|
"""Print a success/info line."""
|
||
|
|
print(f" {msg}", flush=True)
|
||
|
|
|
||
|
|
|
||
|
|
def warn(msg: str) -> None:
|
||
|
|
"""Print a warning to stderr."""
|
||
|
|
print(f" WARN: {msg}", file=sys.stderr, flush=True)
|
||
|
|
|
||
|
|
|
||
|
|
def die(msg: str) -> None:
|
||
|
|
"""Print an error to stderr and exit."""
|
||
|
|
print(f"\nERROR: {msg}", file=sys.stderr, flush=True)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
|
||
|
|
def table(rows: list[list[str]], headers: list[str]) -> str:
|
||
|
|
"""Return an aligned text table. Columns padded to max width."""
|
||
|
|
if not headers:
|
||
|
|
return ""
|
||
|
|
# Compute column widths
|
||
|
|
col_widths = [len(h) for h in headers]
|
||
|
|
for row in rows:
|
||
|
|
for i, cell in enumerate(row):
|
||
|
|
if i < len(col_widths):
|
||
|
|
col_widths[i] = max(col_widths[i], len(cell))
|
||
|
|
# Format header
|
||
|
|
header_line = " ".join(h.ljust(col_widths[i]) for i, h in enumerate(headers))
|
||
|
|
separator = " ".join("-" * w for w in col_widths)
|
||
|
|
lines = [header_line, separator]
|
||
|
|
# Format rows
|
||
|
|
for row in rows:
|
||
|
|
cells = []
|
||
|
|
for i in range(len(headers)):
|
||
|
|
val = row[i] if i < len(row) else ""
|
||
|
|
cells.append(val.ljust(col_widths[i]))
|
||
|
|
lines.append(" ".join(cells))
|
||
|
|
return "\n".join(lines)
|