如何在 Python 中捕捉 SIGINT 信号并优雅地处理键盘中断？

使用信号处理库捕捉 SIGINT

在 Python 脚本中，捕捉键盘中断对于优雅地处理用户输入至关重要。SIGINT 信号（通常通过按下 Ctrl+C 触发）允许你在程序收到该信号时执行自定义操作。本文将指导你如何使用信号处理库来捕捉和处理 SIGINT 信号。

引入信号处理库

首先，导入信号处理库：

import signal

定义信号处理函数

接下来，定义一个名为 signal_handler 的函数，该函数将在收到 SIGINT 信号时执行。在函数内部，你可以编写自定义的清理操作或其他所需的代码：

def signal_handler(signum, frame):
    print("捕捉到 Ctrl+C 信号。执行清理操作...")
    # 在此处执行清理操作
    exit(0)

注册信号处理函数

使用 signal.signal 函数将 signal_handler 注册为 SIGINT 信号的处理函数：

signal.signal(signal.SIGINT, signal_handler)

启动主循环

在你的脚本中，启动一个主循环，该循环将持续运行，直到按下 Ctrl+C。在循环中，你可以执行脚本的主要逻辑：

# 主循环
while True:
    # 你的循环逻辑
    time.sleep(1)

示例代码

以下是完整示例代码：

import signal

def signal_handler(signum, frame):
    print("捕捉到 Ctrl+C 信号。执行清理操作...")
    exit(0)

signal.signal(signal.SIGINT, signal_handler)

# 主循环
while True:
    # 你的循环逻辑
    time.sleep(1)

当按下 Ctrl+C 时，signal_handler 函数将被调用，你的脚本将执行清理操作并退出。

注意事项

• 始终在你的脚本中处理 SIGINT 信号。如果没有处理，脚本将在收到 Ctrl+C 时立即终止。
• 你可以使用 signal.pause() 函数暂停信号处理，直到收到信号。
• signal.getsignal 和 signal.setsignal 函数可用于获取和设置特定信号的当前处理函数。

结论

通过使用信号处理库，你可以轻松地捕捉和处理 SIGINT 信号。这使你能够在脚本收到键盘中断时执行自定义操作，从而增强用户体验并提高脚本的稳健性。

常见问题解答

1. 为什么在 Python 脚本中捕捉 SIGINT 信号很重要？

捕捉 SIGINT 信号允许你优雅地处理用户中断，从而避免脚本意外终止并丢失数据。

2. 除了清理操作，我还可以使用信号处理函数做什么？

你可以使用信号处理函数来执行各种任务，例如日志记录、保存数据或关闭数据库连接。

3. 如何确定要捕捉的信号类型？

SIGINT 信号对应于 Ctrl+C，但信号库支持多种信号，具体取决于你的操作系统。

4. signal.pause() 函数的作用是什么？

signal.pause() 暂停信号处理，直到收到信号。在某些情况下，这可能是有用的，例如等待用户输入。

5. 如何在其他语言中捕捉 SIGINT 信号？

在其他语言中，捕捉 SIGINT 信号的方法可能有所不同。例如，在 C 中，可以使用 signal.h 头文件。