Python 在 Python 中如何捕获 SIGINT 信号

在本文中，我们将介绍如何在 Python 中捕获 SIGINT（Interrupt信号）并对其处理。SIGINT信号通常是在按下Ctrl+C键时发送给正在运行的进程，用于请求进程终止。

什么是信号？

在操作系统中，信号是用于进程间通信（IPC）的一种基本机制。信号是操作系统向进程发送的软件中断。常见的信号有SIGINT（由Ctrl+C发送）、SIGHUP（由终端关闭发送）和SIGKILL（用于强制终止进程）等。

Python 中的信号处理

Python提供了signal模块来处理信号。signal模块包含了捕获信号、注册处理函数和发送信号的相关函数和常量。

捕获 SIGINT 信号

在Python中，我们可以使用signal模块的signal函数来捕获SIGINT信号，并指定一个处理函数。下面的示例演示了如何捕获SIGINT信号并执行自定义的处理函数：

import signal

def handle_signal(signal, frame):
    print("Received SIGINT signal. Exiting...")
    exit(0)

signal.signal(signal.SIGINT, handle_signal)

在上述示例中，我们定义了一个handle_signal函数来处理SIGINT信号。当接收到SIGINT信号时，该函数将打印一条消息并调用exit函数来终止进程。

恢复默认信号处理

如果我们希望在某个时刻恢复默认的信号处理，可以使用signal模块的default_int_handler函数。以下示例演示了如何在处理SIGINT信号后恢复默认的信号处理：

import signal

def handle_signal(signal, frame):
    print("Received SIGINT signal. Exiting...")
    signal.signal(signal.SIGINT, signal.default_int_handler)

signal.signal(signal.SIGINT, handle_signal)

在上面的示例中，我们在handle_signal函数中先打印一条消息，然后使用default_int_handler函数恢复默认的SIGINT信号处理。

发送信号

通过在Python中使用os.kill函数，我们可以发送一个信号给指定的进程。以下示例演示了如何向进程发送SIGINT信号：

import os
import signal

pid = os.getpid()
os.kill(pid, signal.SIGINT)

在上面的示例中，我们使用os.getpid函数获取当前进程的ID，然后使用os.kill函数向该进程发送SIGINT信号。

示例：使用捕获SIGINT信号的应用

下面是一个示例应用程序，它在接收到SIGINT信号后会询问用户是否退出，并根据用户的选择来进行相应动作。

import signal

def handle_signal(signal, frame):
    answer = input("Received SIGINT signal. Do you want to exit? (y/n) ")
    if answer.lower() == "y":
        exit(0)

signal.signal(signal.SIGINT, handle_signal)

while True:
    pass

在上述示例中，当接收到SIGINT信号时，handle_signal函数将询问用户是否退出。如果用户选择退出（输入”y”），则调用exit函数终止进程；如果用户选择不退出，则程序将继续运行。

总结

Python提供了signal模块来处理信号，包括捕获信号、注册处理函数和发送信号等功能。通过捕获SIGINT信号并指定处理函数，我们可以在Python程序中实现对该信号的自定义处理。此外，还可以通过os.kill函数向指定进程发送信号。在开发Python应用程序时，了解和使用信号处理机制可以提高程序的稳定性和可靠性。