使用prompt_toolkit实现基于选择的命令行交互界面

prompt_toolkit 是一个 Python 库，用于构建命令行交互界面。它提供了丰富的组件来构建交互式的命令行应用程序，包括自动补全、提示、多行输入、颜色输出等功能。

下面是使用 prompt_toolkit 实现基于选择的命令行交互界面的一个示例：

from prompt_toolkit import prompt
from prompt_toolkit.completion import WordCompleter
from prompt_toolkit.shortcuts import menu_dialog, radiolist_dialog

def main_menu():
    # 创建菜单项的名称和对应的动作
    choices = [("Option 1", option1),
               ("Option 2", option2),
               ("Exit", exit_program)]
    
    # 显示菜单对话框，并获取选择的菜单项
    result = menu_dialog(
        title="Main Menu",
        text="Please select an option:",
        values=choices,
    ).run()
    
    # 根据选择的菜单项执行对应的动作
    if result is not None:
        result()

def option1():
    # 选项 1 的逻辑
    choices = [("Task A", task_a),
               ("Task B", task_b),
               ("Back to main menu", main_menu)]
    
    # 显示选择对话框，并获取选择的选项
    result = radiolist_dialog(
        title="Option 1 Menu",
        text="Please select a task:",
        values=choices,
    ).run()
    
    # 根据选择的选项执行对应的任务
    if result is not None:
        result()

def option2():
    # 选项 2 的逻辑
    prompt("Enter some input: ")

def task_a():
    # 任务 A 的逻辑
    prompt("Enter some input for task A: ")

def task_b():
    # 任务 B 的逻辑
    prompt("Enter some input for task B: ")

def exit_program():
    # 退出程序
    print("Exiting...")
    raise KeyboardInterrupt

# 启动程序
main_menu()

这个例子实现了一个简单的命令行交互界面，用户可以选择不同的选项来执行不同的任务。主菜单中有两个选项，分别是选项 1 和选项 2。选择选项 1 后，可以进一步选择任务 A 或任务 B。选择选项 2 后，用户可以输入一些文本。

使用 prompt_toolkit 的 menu_dialog 和 radiolist_dialog 可以快速创建菜单和选择对话框。prompt 函数用于输入文本内容。

这只是 prompt_toolkit 的一个基本用例，你可以根据自己的具体需求进行定制和扩展。prompt_toolkit 还提供了很多其他功能，如自定义补全、选择列表、历史记录等，你可以根据实际需求选择使用。