如何通过辅助函数返回值控制main函数中的while循环（替代sys.exit）

你的错误点 #

第一种尝试的问题 #

• 条件判断写在了while循环外面，只会在循环结束后执行一次，完全起不到控制循环的作用。
• 错误使用==进行赋值（应该用=），且重复调用getValidInput会导致用户被重复要求输入，逻辑混乱。
• getValidInput当前返回的是print()的结果（即None），永远不可能等于"0"，因为print()没有返回值。

第二种尝试的问题 #

• 把main()的循环控制逻辑写到了getValidInput里，违反了需求中“所有逻辑需在main()函数内执行”的要求。
• selection变量在getValidInput里未定义，会直接报错。
• 同样犯了用==赋值的错误，且函数返回的还是print()的None，无法传递状态给main()。

正确实现思路 #

核心是让getValidInput只做输入验证和结果反馈，把输入的有效状态返回给main()，由main()控制循环：

1. getValidInput接收用户输入，验证是否合法（0-5）。
2. 无效输入：打印错误信息，返回None表示需要重新输入。
3. 输入1-5：打印对应消息，返回True表示继续循环。
4. 输入0：打印 goodbye，返回False表示终止循环。

修正后的代码 #

def getValidInput(selection):
    if selection in {"0", "1", "2", "3", "4", "5"}:
        if selection == "0":
            print("goodbye")
            return False  # 通知main终止循环
        elif selection == "1":
            print("message_1")
        elif selection == "2":
            print("message_2")
        elif selection == "3":
            print("message_3")
        elif selection == "4":
            print("message_4")
        elif selection == "5":
            print("message_5")
        return True  # 有效输入，继续循环
    else:
        print("That is not a valid option.")
        return None  # 无效输入，需重新输入

def main():
    askAgain = True
    while askAgain:
        user_input = input("Select number 1-5. 0 exits program.")
        result = getValidInput(user_input)
        # 根据验证结果更新循环条件
        if result is not None:
            askAgain = result

if __name__ == "__main__":
    main()

代码说明 #

• getValidInput仅负责验证输入和打印消息，通过返回值传递状态给main()，完全符合用while条件控制退出的要求。
• main()中每次获取用户输入后，根据验证结果更新askAgain：
  • 无效输入时，result为None，askAgain保持True，继续循环。
  • 输入1-5时，result为True，继续循环。
  • 输入0时，result为False，循环终止。
• 所有控制逻辑都在main()内，仅把输入验证的工作交给辅助函数，满足需求要求。

内容的提问来源于stack exchange，提问作者user20219794