Function select_from_options

deeptutor_cli/init_wizard.py:269–326

Render a numbered/keyed menu, return the selected key. ``options`` is the visible numbered list. ``extra_keys`` adds letter shortcuts (e.g. ``{"s": "Show all providers", "c": "Custom"}``) — these show up after the numbered rows and are accepted as input.

(
    console: Console,
    *,
    title: str,
    options: list[tuple[str, str, str]],  # [(key, label, hint), ...]
    default_key: str | None = None,
    extra_keys: dict[str, str] | None = None,
    prompt_label: str = "Choice",
    invalid_label: str = "Invalid choice. Try again.",
)

def select_from_options(
    console: Console,
    *,
    title: str,
    options: list[tuple[str, str, str]],  # [(key, label, hint), ...]
    default_key: str | None = None,
    extra_keys: dict[str, str] | None = None,
    prompt_label: str = "Choice",
    invalid_label: str = "Invalid choice. Try again.",
) -> str:
    """Render a numbered/keyed menu, return the selected key.

    ``options`` is the visible numbered list. ``extra_keys`` adds letter
    shortcuts (e.g. ``{"s": "Show all providers", "c": "Custom"}``) — these
    show up after the numbered rows and are accepted as input.
    """

    # Titles come from i18n and may contain `[c]`-style brackets that Rich
    # would otherwise interpret as markup tags.
    console.print(f"[bold]{rich_escape(title)}[/bold]")
    console.print()
    table = Table.grid(padding=(0, 1))
    table.add_column(style="bright_cyan", justify="right")
    table.add_column(style="bold")
    table.add_column(style="dim")

    # Rich Table cells parse markup, so e.g. `[s]` would be eaten as a
    # nonexistent tag. Wrap markers in Text so they render verbatim.
    def _marker(text: str) -> Text:
        return Text(text, style="bright_cyan", justify="right")

    for idx, (_key, label, hint) in enumerate(options, start=1):
        table.add_row(_marker(f"[{idx}]"), label, hint or "")
    if extra_keys:
        for short, label in extra_keys.items():
            table.add_row(_marker(f"[{short}]"), label, "")
    console.print(table)
    console.print()

    valid_numbers = {str(i): options[i - 1][0] for i in range(1, len(options) + 1)}
    valid_letters = {k.lower(): k.lower() for k in (extra_keys or {})}
    default_input: str | None = None
    if default_key is not None:
        for idx, (key, _label, _hint) in enumerate(options, start=1):
            if key == default_key:
                default_input = str(idx)
                break
        if default_input is None and default_key.lower() in valid_letters:
            default_input = default_key.lower()

    while True:
        raw = typer.prompt(prompt_label, default=default_input or "")
        choice = str(raw).strip().lower()
        if choice in valid_numbers:
            return valid_numbers[choice]
        if choice in valid_letters:
            return valid_letters[choice]
        fail(console, invalid_label)