python中通过__init__参数实现的超时装饰器

import signal

class TimeoutDecorator:
    def __init__(self, timeout):
        self.timeout = timeout

    def __call__(self, func):
        def wrapper(*args, **kwargs):
            def handle_timeout(signum, frame):
                raise TimeoutError("Function execution timed out")

            # 设置超时信号处理函数
            signal.signal(signal.SIGALRM, handle_timeout)
            # 设置超时时间
            signal.alarm(self.timeout)

            try:
                # 执行被装饰的函数
                result = func(*args, **kwargs)
                return result
            finally:
                # 取消超时信号
                signal.alarm(0)

        return wrapper

@TimeoutDecorator(5)  # 设置超时时间为5秒
def long_running_function():
    # 执行耗时较长的操作
    ...

try:
    long_running_function()
except TimeoutError:
    print("Function execution timed out")