table_plain.py

Go to the documentation of this file.

# Copyright (c) 2025-present Aquin Labs Private Limited. All Rights Reserved.
"""Plain Unicode tables for quiet CLI surfaces (no Rich)."""
 
from __future__ import annotations
 
import sys
from collections.abc import Sequence
 
 
# Box-drawing (falls back to ASCII if stdout can't encode them).
_UNICODE = {
    "tl": "┌", "tr": "┐", "bl": "└", "br": "┘",
    "h": "─", "v": "│",
    "l": "├", "r": "┤", "t": "┬", "b": "┴", "x": "┼",
}
_ASCII = {
    "tl": "+", "tr": "+", "bl": "+", "br": "+",
    "h": "-", "v": "|",
    "l": "+", "r": "+", "t": "+", "b": "+", "x": "+",
}
 
 
def _glyphs() -> dict[str, str]:
    try:
        encoding = getattr(sys.stdout, "encoding", None) or "utf-8"
        "┌─│".encode(encoding)
        return _UNICODE
    except Exception:
        return _ASCII
 
 
def _cell(text: str, width: int, *, align: str = "left") -> str:
    s = str(text)
    if len(s) > width:
        if width <= 1:
            s = s[:width]
        else:
            s = s[: width - 1] + "…"
    if align == "right":
        return s.rjust(width)
    return s.ljust(width)
 
 
def format_table(
    headers: Sequence[str],
    rows: Sequence[Sequence[str]],
    *,
    aligns: Sequence[str] | None = None,
    max_col: int = 48,
) -> str:
    """Return a bordered table as a multi-line string."""
    g = _glyphs()
    n = len(headers)
    if aligns is None:
        aligns = ["left"] * n
    elif len(aligns) < n:
        aligns = list(aligns) + ["left"] * (n - len(aligns))
 
    widths = [min(max_col, len(str(h))) for h in headers]
    for row in rows:
        for i, cell in enumerate(row[:n]):
            widths[i] = min(max_col, max(widths[i], len(str(cell))))
 
    def line(left: str, join: str, right: str) -> str:
        return left + join.join(g["h"] * (w + 2) for w in widths) + right
 
    def row_line(cells: Sequence[str]) -> str:
        parts = [
            f" {_cell(cells[i] if i < len(cells) else '', widths[i], align=aligns[i])} "
            for i in range(n)
        ]
        return g["v"] + g["v"].join(parts) + g["v"]
 
    out: list[str] = [
        line(g["tl"], g["t"], g["tr"]),
        row_line(headers),
        line(g["l"], g["x"], g["r"]),
    ]
    for row in rows:
        out.append(row_line(row))
    out.append(line(g["bl"], g["b"], g["br"]))
    return "\n".join(out)
 
 
def print_table(
    headers: Sequence[str],
    rows: Sequence[Sequence[str]],
    *,
    aligns: Sequence[str] | None = None,
    max_col: int = 48,
    title: str | None = None,
) -> None:
    try:
        sys.stdout.reconfigure(encoding="utf-8")  # type: ignore[attr-defined]
    except Exception:
        pass
    if title:
        print(title)
    print(format_table(headers, rows, aligns=aligns, max_col=max_col))