Python

概述

使用 python 的rich和prompt_toolkit做一个漂亮的单选样式。

代码示例

1
2
pip3 install rich
pip3 install prompt_toolkit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from prompt_toolkit import Application
from prompt_toolkit.key_binding import KeyBindings
from prompt_toolkit.layout import Layout, HSplit
from prompt_toolkit.widgets import Label
from prompt_toolkit.layout.controls import FormattedTextControl
from prompt_toolkit.layout.containers import Window
from rich.console import Console

console = Console()


def interactive_select(options: dict[str, str], prompt: str = "请选择一个选项:") -> str:
    keys = list(options.keys())
    selected_idx = [0]  # 使用列表是为了能在内部函数中修改

    def get_option_text():
        items = []
        for idx, key in enumerate(keys):
            if idx == selected_idx[0]:
                items.append(("reverse", f" ● {key} "))
            else:
                items.append(("", f" ○ {key} "))
            items.append(("", "   "))  # 间隔
        return items

    control = FormattedTextControl(get_option_text)
    window = Window(content=control, always_hide_cursor=True)
    kb = KeyBindings()

    @kb.add("left")
    def move_left(event):
        selected_idx[0] = (selected_idx[0] - 1) % len(keys)
        event.app.invalidate()

    @kb.add("right")
    def move_right(event):
        selected_idx[0] = (selected_idx[0] + 1) % len(keys)
        event.app.invalidate()

    @kb.add("enter")
    def enter(event):
        event.app.exit(result=options[keys[selected_idx[0]]])

    app = Application(
        layout=Layout(HSplit([Label(text=prompt), window])),
        key_bindings=kb,
        mouse_support=False,
        full_screen=False,
    )

    result = app.run()
    return result


# if __name__ == "__main__":
#     source_options = {
#         "国内源（中科大）": "cn-ustc",
#         "国内源（清华）": "cn-tsinghua",
#         "国外源（Google）": "us-google",
#     }

#     selected = interactive_select(
#         source_options, prompt="请选择需要使用的源 (← → 选择，回车确认)"
#     )
#     console.print(f"你选择了: [bold green]{selected}[/]")

效果

效果大概长这样，其实rich不装也行，无所谓的，但是如果你想后面输出的东西好看点，还是装上比较好。