有没有办法永远不会在 python 中的 KeyboardInterrupt 上退出？

Question

我正在 python 中创建一种交互式命令行。我有这样的事情：

def complete_menu():
    while True:
        cmd = input('cmd> ')
        if cmd == "help":
            print('help')
        elif cmd == "?":
            print('?')

当用户按下 CTRL-C，而不是退出程序时，我试图让它打印“请键入 exit 以退出”并返回到 while True。我现在有这样的事情：

if __name__ == "__main__":
    try:
       main()
    except KeyboardInterrupt:
        print('Please use exit to exit')
        complete_menu()

虽然这可行，但存在许多问题。首先，当第一次按下 CTRL-C 时，它会打印出文本并完美运行。但是，当用户第二次按下 CTRL-C 时，它会像其他任何程序一样在按下 CTRL-C 后出现一堆乱七八糟的文本。这可以解决吗？

Answer 1

更好的方法是注册一个信号处理程序：

import signal

def handler(signum, frame):
    print("Please use exit to exit")
    # or: just call sys.exit("goodbye")

...

def main():
    signal.signal(signal.SIGINT, handler)  # prevent "crashing" with ctrl+C
    ...


if __name__ == "__main__":
    main()

现在，当您的代码中收到 Ctrl+C 时，将执行函数 handler，而不是引发 KeyboardInterrupt 异常。这是一个基本示例，在处理程序中自定义代码以执行您想要的操作。

注意：我的建议是实际上让用户使用 Ctrl+C 退出，即执行您可能需要执行的任何清理代码运行，然后在此处调用sys.exit。需要更强信号才能杀死的程序很烦人。

Answer 2

你快到了。这应该可以解决问题：

if __name__ == "__main__":
    while True:
        try:
            main()
        except KeyboardInterrupt:
            print('Please use exit to exit')
            complete_menu()