如何让python上下文管理器捕获SIGINT或SIGTERM信号

在Python中，上下文管理器通常用于管理资源，如文件操作或数据库连接，确保它们在使用后被正确释放。要使上下文管理器能够捕获SIGINT（通常由Ctrl+C触发）或SIGTERM（通常由系统请求终止进程触发）信号，我们可以使用signal模块来注册信号处理函数，并在上下文管理器中处理这些信号。

基础概念

上下文管理器：Python中的上下文管理器通过with语句使用，用于管理资源，确保资源在使用后被正确释放。
信号：操作系统发出的消息，用于通知进程某个事件已经发生。SIGINT和SIGTERM是常见的信号类型。

类型与应用场景

SIGINT：通常用于用户中断程序执行，如在终端中按下Ctrl+C。
SIGTERM：用于请求程序正常终止，通常由系统管理工具发送。

示例代码

以下是一个简单的上下文管理器示例，它能够捕获SIGINT和SIGTERM信号，并在接收到这些信号时执行清理操作：

import signal
import time
from contextlib import contextmanager

class SignalHandlingContextManager:
    def __init__(self):
        self.original_sigint_handler = signal.getsignal(signal.SIGINT)
        self.original_sigterm_handler = signal.getsignal(signal.SIGTERM)

    def __enter__(self):
        signal.signal(signal.SIGINT, self.handle_signal)
        signal.signal(signal.SIGTERM, self.handle_signal)
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        signal.signal(signal.SIGINT, self.original_sigint_handler)
        signal.signal(signal.SIGTERM, self.original_sigterm_handler)

    def handle_signal(self, signum, frame):
        print(f"Received signal {signum}. Cleaning up and exiting gracefully...")
        # 在这里执行清理操作
        exit(0)

# 使用上下文管理器
with SignalHandlingContextManager():
    try:
        while True:
            print("Running...")
            time.sleep(1)
    except SystemExit:
        pass  # 这里可以添加额外的清理代码，如果需要的话