"""Shared renderers for typer-test output tables. This module centralizes Rich table rendering so typer-test.py stays lean. It includes specialized handling for planets plus generic list/dict/object renderers. """ from __future__ import annotations from typing import Any, Dict, List, Optional, Sequence from rich import box from rich.console import Console, Group from rich.panel import Panel from rich.table import Table from tarot import Card from utils.attributes import Color, Planet class Renderer: """Render arbitrary Tarot objects using Rich tables.""" def __init__(self, console: Console) -> None: self.console = console self.planet_renderer = PlanetRenderer(self) # Public helpers ----------------------------------------------------- def render_value(self, value: Any) -> None: if isinstance(value, Planet): self.planet_renderer.render(value) return if isinstance(value, Card): self._render_card(value) return if isinstance(value, list) and value and isinstance(value[0], Card): self.print_cards_table(value) return if hasattr(value, "upright") and hasattr(value, "reversed"): self.print_kv_table( "Meaning", {"upright": value.upright, "reversed": value.reversed} ) return if isinstance(value, dict): self._render_dict(value) return if isinstance(value, list): self._render_list(value) return if hasattr(value, "__dict__"): self._render_object(value) return self.console.print(value) def print_kv_table(self, title: str, data: Dict[str, Any]) -> None: table = Table(title=title, show_header=False) table.add_column("Field", style="bold") table.add_column("Value") for key, value in data.items(): table.add_row(self._humanize_key(str(key)), "" if value is None else str(value)) self.console.print(table) def print_rows_table(self, title: str, columns: List[str], rows: List[List[Any]]) -> None: table = Table(title=title, show_header=True, show_lines=True) for col in columns: table.add_column(col) for row in rows: str_row = ["" if v is None else str(v) for v in row] table.add_row(*str_row) self.console.print(table) def print_cards_table(self, cards: List[Card]) -> None: table = Table(title="Cards", show_lines=True) table.add_column("#", justify="right") table.add_column("Name") table.add_column("Arcana") table.add_column("Suit") table.add_column("Pip/Court") for card in cards: suit = getattr(getattr(card, "suit", None), "name", "") pip = str(getattr(card, "pip", "")) if getattr(card, "pip", None) else "" court = getattr(card, "court_rank", "") pip_display = court or pip table.add_row( str(getattr(card, "number", "")), getattr(card, "name", ""), getattr(card, "arcana", ""), suit, pip_display, ) self.console.print(table) # Internal renderers ------------------------------------------------- def _render_card(self, card: Card) -> None: """Render a card in a rectangular layout with top/middle/bottom bars.""" name = getattr(card, "name", "") arcana = getattr(card, "arcana", "") suit_obj = getattr(card, "suit", None) suit = getattr(suit_obj, "name", "") if suit_obj is not None else "" pip = getattr(card, "pip", None) number = getattr(card, "number", "") keywords = getattr(card, "keywords", None) or [] meaning = getattr(card, "meaning", None) top = Table.grid(expand=True) for align in ("left", "center", "right"): top.add_column(justify=align) top_left = suit or arcana top_center = name top_right = f"#{number}" if number not in (None, "") else "" top.add_row(top_left, top_center, top_right) body_lines = [f"Arcana: {arcana or '-'}"] if suit: body_lines.append(f"Suit: {suit}") if pip: body_lines.append(f"Pip: {pip}") if getattr(card, "court_rank", ""): body_lines.append(f"Court Rank: {card.court_rank}") if keywords: body_lines.append(f"Keywords: {', '.join(keywords)}") guidance = getattr(card, "guidance", None) if guidance: body_lines.append(f"Guidance: {guidance}") body = Panel("\n".join(body_lines), box=box.MINIMAL, padding=(1, 1)) bottom = Table( show_header=False, expand=True, box=box.SIMPLE_HEAVY, show_edge=True, pad_edge=False, ) bottom.add_column(justify="left") bottom.add_column(justify="center") bottom.add_column(justify="right") bottom.add_row( self._label_block("Upright", getattr(meaning, "upright", None)), self._label_block("Reversed", getattr(meaning, "reversed", None)), self._label_block("Keywords", ", ".join(keywords) if keywords else None), ) card_panel = Panel( Group(top, body, bottom), box=box.SQUARE, padding=0, title=name, expand=False, ) self.console.print(card_panel) def _render_object(self, obj: Any) -> None: attrs = {k: v for k, v in vars(obj).items() if not k.startswith("_")} display = {k: self._format_value(v) for k, v in attrs.items()} title = getattr(obj, "name", None) or obj.__class__.__name__ self.print_kv_table(str(title), display) def _render_dict(self, data: Dict[Any, Any], title: str = "Result") -> None: if not data: self.console.print(f"{title}: {{}}") return values = list(data.values()) if all(self._is_scalar(v) for v in values): self.print_kv_table(title, data) return if all(isinstance(v, list) for v in values) and values and values[0]: first_item = values[0][0] if hasattr(first_item, "__dict__"): self._render_dict_of_list_objects(data, title) return if all(hasattr(v, "__dict__") for v in values): rows = [] sample_attrs = self._simple_attrs(values[0]) columns = ["Key"] + [self._humanize_key(k) for k in sample_attrs.keys()] for key, obj in data.items(): attrs = self._simple_attrs(obj) rows.append([key] + [attrs.get(k, "") for k in sample_attrs.keys()]) self.print_rows_table(title, columns, rows) return self.print_kv_table(title, {k: self._short_value(v) for k, v in data.items()}) def _render_list(self, items: List[Any], title: str = "Result") -> None: if not items: self.console.print(f"{title}: []") return first = items[0] if isinstance(first, Card): self.print_cards_table(items) return if hasattr(first, "__dict__"): sample_attrs = self._simple_attrs(first) columns = [self._humanize_key(k) for k in sample_attrs.keys()] rows = [] for obj in items: attrs = self._simple_attrs(obj) rows.append([attrs.get(k, "") for k in sample_attrs.keys()]) self.print_rows_table(title, columns, rows) return if isinstance(first, dict): keys = list(first.keys()) columns = [self._humanize_key(k) for k in keys] rows = [] for obj in items: rows.append([obj.get(k, "") for k in keys]) self.print_rows_table(title, columns, rows) return self.print_kv_table(title, {i: self._short_value(v) for i, v in enumerate(items)}) def _render_dict_of_list_objects(self, data: Dict[Any, List[Any]], title: str) -> None: first_key = next(iter(data)) first_list = data[first_key] if not first_list: self.print_kv_table(title, {k: "[]" for k in data.keys()}) return sample_attrs = self._simple_attrs(first_list[0]) columns = ["Key"] + [self._humanize_key(k) for k in sample_attrs.keys()] rows: List[List[Any]] = [] for key, lst in data.items(): for obj in lst: attrs = self._simple_attrs(obj) rows.append([key] + [attrs.get(k, "") for k in sample_attrs.keys()]) self.print_rows_table(title, columns, rows) # Value helpers ------------------------------------------------------ @staticmethod def _simple_attrs(obj: Any) -> Dict[str, Any]: return {k: v for k, v in vars(obj).items() if not k.startswith("_")} @staticmethod def _is_scalar(val: Any) -> bool: return isinstance(val, (str, int, float, bool)) or val is None def _format_value(self, val: Any, depth: int = 0) -> str: if isinstance(val, list): if not val: return "[]" lines = [] for item in val: lines.append(" " * depth + "- " + self._format_value(item, depth + 1).lstrip()) return "\n".join(lines) if isinstance(val, dict): lines = [] for k, v in val.items(): lines.append( " " * depth + f"{self._humanize_key(str(k))}: " + self._format_value(v, depth + 1) ) return "\n".join(lines) if hasattr(val, "__dict__"): inner = {k: v for k, v in vars(val).items() if not k.startswith("_")} if not inner: return str(val) parts = [ f"{self._humanize_key(str(k))}={self._short_value(v)}" for k, v in inner.items() ] name = getattr(val, "name", val.__class__.__name__) return f"{name}: " + "; ".join(parts) return str(val) def _short_value(self, val: Any) -> str: if isinstance(val, Card): return f"Card {getattr(val, 'number', '?')}: {getattr(val, 'name', '')}" if isinstance(val, list): if not val: return "[]" preview = ", ".join(self._short_value(v) for v in val[:4]) suffix = "..." if len(val) > 4 else "" return f"[{preview}{suffix}]" if isinstance(val, dict): return f"dict({len(val)})" return str(val) @staticmethod def _humanize_key(key: str) -> str: return key.replace("_", " ") @staticmethod def _label_block(label: str, value: Optional[str]) -> str: safe_val = value if value not in (None, "") else "—" return f"{label}:\n{safe_val}" class PlanetRenderer: """Type-specific table rendering for planets.""" def __init__(self, renderer: Renderer) -> None: self.renderer = renderer @property def console(self) -> Console: return self.renderer.console def render(self, planet: Planet) -> None: base_fields = { "Symbol": planet.symbol, "Element": planet.element, "Ruling Zodiac": self._join_list(planet.ruling_zodiac), "Associated Letters": self._join_list(planet.associated_letters), "Keywords": self._join_list(planet.keywords), "Description": planet.description, } self.renderer.print_kv_table(f"Planet: {planet.name}", self._compact(base_fields)) if planet.associated_numbers: rows = [] for number in planet.associated_numbers: color_label = self._color_label(number.color) rows.append( [ getattr(number, "value", ""), getattr(number, "sephera", ""), getattr(number, "element", ""), getattr(number, "compliment", ""), color_label, ] ) self.renderer.print_rows_table( "Associated Numbers", ["Value", "Sephera", "Element", "Compliment", "Color"], rows, ) if planet.color: color = planet.color color_fields = { "Name": color.name, "Hex": color.hex_value, "Element": color.element, "Scale": color.scale, "Meaning": color.meaning, "Tarot Associations": self._join_list(getattr(color, "tarot_associations", [])), "Description": color.description, } self.renderer.print_kv_table("Planetary Color", self._compact(color_fields)) @staticmethod def _join_list(items: Sequence[Any]) -> str: return ", ".join(str(item) for item in items) if items else "" @staticmethod def _compact(data: Dict[str, Any]) -> Dict[str, Any]: return {k: v for k, v in data.items() if v not in (None, "", [], {})} @staticmethod def _color_label(color: Optional[Color]) -> str: if isinstance(color, Color): hex_part = f" ({color.hex_value})" if getattr(color, "hex_value", "") else "" return f"{color.name}{hex_part}" return ""