在脚本中捕获脚本的Ctrl-C

在脚本中捕获脚本的Ctrl-C（即SIGINT信号），是一种常见的需求，尤其是在需要优雅地关闭脚本或执行清理操作时。以下是关于这个问题的基础概念、优势、类型、应用场景以及解决方案。

基础概念

• 信号（Signal）：信号是一种软件中断，用于通知进程发生了某种情况。例如，用户按下Ctrl-C会发送SIGINT信号给当前进程。
• SIGINT信号：这是当用户按下Ctrl-C时，操作系统发送给前台进程组的信号，通常用于请求进程终止。

优势

• 优雅关闭：捕获SIGINT信号允许脚本在退出前执行清理操作，如关闭文件、释放资源等。
• 用户友好：提供更好的用户体验，因为用户可以通过Ctrl-C随时中断脚本，而不必等待脚本自然结束。

类型

• 同步信号处理：在信号处理函数中直接处理信号。
• 异步信号处理：设置信号处理函数，在信号发生时调用。

应用场景

• 长时间运行的脚本：如服务器监控、数据处理等。
• 需要清理资源的脚本：如文件操作、数据库连接等。

解决方案

以下是一个使用Python捕获SIGINT信号的示例：

import signal
import time

def signal_handler(sig, frame):
    print('捕获到SIGINT信号，正在退出...')
    # 在这里执行清理操作
    exit(0)

signal.signal(signal.SIGINT, signal_handler)

print('脚本开始运行，按Ctrl-C退出...')
while True:
    time.sleep(1)

解释

1. 导入模块：导入signal和time模块。
2. 定义信号处理函数：定义一个名为signal_handler的函数，该函数将在捕获到SIGINT信号时被调用。
3. 设置信号处理函数：使用signal.signal()函数设置SIGINT信号的处理函数为signal_handler。
4. 主循环：在主循环中，脚本会不断运行，直到捕获到SIGINT信号。

参考链接

• Python signal模块文档

通过这种方式，脚本可以在捕获到Ctrl-C时执行必要的清理操作，从而优雅地退出。