在程序关闭期间捕获 Python 中的 KeyboardInterrupt

Question

在程序关闭期间捕获 Python 中的 KeyboardInterrupt

139

我正在用Python编写一个命令行实用程序。由于它是生产代码，因此应该能够在关闭时干净地关闭，而不会将一大堆内容（错误代码、堆栈跟踪等）倾泻到屏幕上。这意味着我需要捕捉键盘中断。

我已经尝试过使用try catch块，例如：

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        print 'Interrupted'
        sys.exit(0)

并捕获信号本身（如此帖子中所述）：

import signal
import sys

def sigint_handler(signal, frame):
    print 'Interrupted'
    sys.exit(0)
signal.signal(signal.SIGINT, sigint_handler)

这两种方法在正常操作期间似乎都能很好地工作。但是，如果中断发生在应用程序末尾的清理代码期间，Python似乎总是向屏幕打印一些内容。捕获中断可以

^CInterrupted
Exception KeyboardInterrupt in <bound method MyClass.__del__ of <path.to.MyClass object at 0x802852b90>> ignored

处理这个信号会得到以下结果：

^CInterrupted
Exception SystemExit: 0 in <Finalize object, dead> ignored

或者

^CInterrupted
Exception SystemExit: 0 in <bound method MyClass.__del__ of <path.to.MyClass object at 0x802854a90>> ignored

这些错误不仅难看，而且对于没有源代码的最终用户来说也没有什么帮助！

该应用程序的清理代码相当庞大，因此存在实际用户遇到此问题的可能性。是否有任何方法可以捕获或阻止此输出，还是我必须处理它？

- Dan

2

为什么不替换 sys.stdout/sys.stderr 呢？比如 sys.stderr = open(os.devnull, 'w')。如果你真的不关心最终的输出，那这似乎是显而易见的解决方案。 - Bakuriu

2

有一个 os._exit 但对我而言它看起来像是鼻妖。你的清理代码在哪里，你是否使用 atexit 模块来做清理工作？ - wim

1

@Bakuriu：虽然重定向 stderr 可以使输出变得安静，但它也会扼杀用户可以采取措施解决的合法错误，比如文件未找到或主机不可达。 - Dan

4

@Dan 不一定要使用/dev/null。您可以编写一个自定义的文件对象，仅隐藏特定格式的消息。 - Bakuriu

2

@Bakuriu：对我来说，这仍然看起来很粗糙。如果必须这样做，我会这样做，但我觉得这应该是语言内置的东西。 - Dan

第三丹的答案不是正确的吗？ - Alexey

2个回答

14

在开始清理代码之前，您可以通过调用signal.signal(signal.SIGINT, signal.SIG_IGN)来忽略关机后发出的SIGINT信号。

- Dan Getz

1

如果代码预计在其他脚本中使用，那么在清理代码后应该返回 signal.SIGINT 到 signal.SIG_DFL，以将其恢复为默认状态。 - StickySli

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Dan Hogan · Accepted Answer

请查看此问题串，其中包含一些有关退出和回溯的有用信息。

如果您只想终止程序，请尝试类似于以下代码（这也会使清理代码失效）：

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        print('Interrupted')
        try:
            sys.exit(130)
        except SystemExit:
            os._exit(130)

[根据评论建议更改退出代码。130是通过Ctrl-C终止脚本时在Linux上通常返回的代码。虽然我们可能不在Linux上，但重要的是返回一个非零值，而130与其他任何值一样好。]