Python 仿APT命令行界面的Yes/No输入

在本文中，我们将介绍如何使用Python编写一个类似于APT（高级软件包工具）命令行界面的程序，用于实现Yes/No输入的功能。

问题背景

在开发交互式命令行工具或自动化脚本时，经常需要获取用户的确认输入。常见的需求是询问用户是否确定继续执行某个操作。传统的实现方式是要求用户输入”y”或”n”来表示确认或取消。然而，这种方式在用户体验和代码可读性方面存在一些问题。

解决方案

为了改善用户体验和代码可读性，我们可以编写一个类似于APT命令行界面的程序，使用”yes”和”no”来代替”y”和”n”。这样的实现方式更符合人们正常的语言表达习惯，并且更容易被用户理解。

下面是一个示例实现：

def get_yes_or_no(prompt):
    while True:
        user_input = input(prompt).lower()

        if user_input == "yes":
            return True
        elif user_input == "no":
            return False
        else:
            print("Invalid input. Please enter 'yes' or 'no'.")


# 使用示例
if get_yes_or_no("Do you want to continue? "):
    print("Continuing...")
else:
    print("Aborted.")

在上面的示例中，我们定义了一个get_yes_or_no函数，该函数接受一个提示信息作为参数，并通过input函数获取用户的输入。然后，我们将用户输入转换为小写字母，以便不区分大小写。

如果用户输入是”yes”，则函数返回True，表示用户确认执行某个操作。如果用户输入是”no”，则函数返回False，表示用户取消了该操作。如果用户输入既不是”yes”也不是”no”，则提示用户重新输入。

通过调用get_yes_or_no函数，我们可以很方便地获取到用户的确认或取消的选择，并根据不同的结果执行相应的逻辑。

改进与扩展

除了实现类似于APT命令行界面的Yes/No输入，我们还可以进行一些改进和扩展，以满足不同的需求。

默认值

有时，我们希望在用户不提供输入时设置一个默认值。我们可以在get_yes_or_no函数中添加一个可选参数来指定默认值：

def get_yes_or_no(prompt, default=False):
    options = "[yes]/no" if default else "yes/[no]"
    prompt = f"{prompt} ({options})"

    while True:
        user_input = input(prompt).lower()

        if user_input in ("", "yes"):
            return True
        elif user_input == "no":
            return False
        else:
            print("Invalid input. Please enter 'yes' or 'no'.")


# 使用示例
if get_yes_or_no("Do you want to continue?", default=True):
    print("Continuing...")
else:
    print("Aborted.")

在上面的示例中，我们根据default参数的值来设置提示信息中的默认值和方括号的位置。如果用户直接回车，则返回默认值；如果用户输入”yes”，则返回True；如果用户输入”no”，则返回False。

循环输入

有时，我们希望用户可以多次输入直到输入有效的选项为止。我们可以修改get_yes_or_no函数，在输入无效时要求用户重新输入：

def get_yes_or_no(prompt, default=False):
    options = "[yes]/no" if default else "yes/[no]"
    prompt = f"{prompt} ({options})"

    while True:
        user_input = input(prompt).lower()

        if user_input in ("", "yes"):
            return True
        elif user_input == "no":
            return False
        else:
            print("Invalid input. Please enter 'yes' or 'no'.")


# 使用示例
while True:
    if get_yes_or_no("Do you want to continue?", default=True):
        print("Continuing...")
        break
    else:
        print("Aborted. Please try again.")

在上面的示例中，如果用户输入无效，则会提示用户重新输入。只有当用户输入有效的选项时，才会跳出循环。

其他选项

除了”yes”和”no”之外，有时我们还希望提供其他的选项，例如”skip”、”cancel”等。我们可以扩展get_yes_or_no函数以支持更多的选项：

def get_yes_or_no(prompt, default=False, options=("yes", "no")):
    prompt_options = "/".join("[{}]".format(option) for option in options)
    prompt_default = options[0] if default else options[1]
    prompt = f"{prompt} ({prompt_options}) [{prompt_default}]"

    while True:
        user_input = input(prompt).lower()

        if not user_input:
            return default
        elif user_input in options:
            return user_input
        else:
            print("Invalid input. Please enter one of the following options:", ", ".join(options))


# 使用示例
if get_yes_or_no("Do you want to continue?", default=True, options=("yes", "no", "cancel")) == "cancel":
    print("Cancelled.")

在上面的示例中，我们通过一个新的options参数来指定所有的选项。如果用户输入为空，则返回默认值；如果用户输入是有效的选项之一，则返回该选项；否则提示用户重新输入，并列出所有可用的选项。