如何让python上下文管理器捕获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  # 这里可以添加额外的清理代码，如果需要的话