使用timeout()设置Python函数的超时时间

在Python中，使用timeout来设置函数的超时时间是通过signal模块实现的。signal模块提供了与信号相关的操作，包括捕获和处理信号。我们可以使用signal.alarm()函数来设置定时器超时时间，并使用signal.signal()函数来注册信号处理函数。当超出指定时间后，将触发相应的信号，从而中断函数的执行。

下面是一个使用timeout()设置Python函数超时时间的例子：

import signal
import time

class TimeoutError(Exception):  # 自定义超时异常类
    pass

def timeout(seconds):
    def decorator(func):
        def handler(signum, frame):
            raise TimeoutError("Time is up!")  # 超时时触发自定义异常
        
        def wrapper(*args, **kwargs):
            signal.signal(signal.SIGALRM, handler)  # 注册信号处理函数
            signal.alarm(seconds)  # 设置超时时间
            try:
                result = func(*args, **kwargs)  # 执行函数
            finally:
                signal.alarm(0)  # 取消定时器
            
            return result
        
        return wrapper
        
    return decorator

@timeout(3)  # 设置超时时间为3秒
def long_running_function():
    time.sleep(5)  # 模拟耗时操作
    return "Function completed successfully!"

try:
    result = long_running_function()  # 执行函数
    print(result)
except TimeoutError as e:
    print(e)

在上面的例子中，我们定义了一个TimeoutError类作为超时异常的自定义类。然后，我们定义了一个名为timeout的装饰器函数，该函数接受一个超时时间作为参数。在装饰器函数内部，我们定义了一个信号处理函数handler，该函数在超时时触发自定义异常。然后，我们定义了一个装饰器内部的包装函数wrapper，该函数通过signal.signal()和signal.alarm()函数设置超时时间并执行函数。最后，我们将装饰器应用于指定函数long_running_function()，并在主程序中捕获超时异常。

在上述例子中，long_running_function()模拟了一个耗时操作，它使用time.sleep(5)函数来让程序休眠5秒。然而，我们通过为该函数添加@timeout(3)装饰器，将超时时间设置为3秒。因此，在调用long_running_function()时，由于执行时间超过了设置的超时时间，将触发自定义的超时异常TimeoutError。在try块中捕获这个异常，并打印相应的错误信息。

这是一个简单的例子来展示如何使用timeout()函数设置Python函数的超时时间。你可以通过修改超时时间和执行时间来测试这个例子，并根据实际需求进行定制。