如何使用timeout()函数在Python中处理长时间运行的函数

在Python中，可以使用timeout函数来处理长时间运行的函数。timeout函数可以设置一个时间限制，在该时间内运行的函数如果超时，则会抛出TimeoutError异常。

timeout函数有多种实现方式，下面给出其中一种实现方式的示例代码：

import signal

# 定义一个自定义的超时异常类
class TimeoutException(Exception):
    def __init__(self, error_message):
        self.error_message = error_message

# 定义timeout装饰器
def timeout(seconds):
    def decorator(func):
        def handler(signum, frame):
            raise TimeoutException("Function execution timed out")
        
        def wrapper(*args, **kwargs):
            # 设置信号处理器
            signal.signal(signal.SIGALRM, handler)
            signal.alarm(seconds)  # 设置超时时间
            
            try:
                result = func(*args, **kwargs)
                return result
            finally:
                signal.alarm(0)  # 取消超时信号
        
        return wrapper
    
    return decorator

通过上面的示例代码，我们可以使用timeout函数来处理长时间运行的函数。下面给出一个使用示例：

@timeout(2)  # 设置超时时间为2秒
def long_running_function():
    import time
    time.sleep(5)  # 模拟一个长时间运行的函数

try:
    long_running_function()
except TimeoutException as e:
    print(e.error_message)
else:
    print("Function executed successfully.")

在上面的示例中，long_running_function函数模拟一个长时间运行的函数，但是由于我们在timeout装饰器中设置了最长运行时间为2秒，所以在实际运行时该函数会超时，抛出TimeoutException异常。在我们的代码中，我们捕获了这个异常，并打印出错误信息。如果函数在超时时间内能够正常执行完毕，就会打印出"Function executed successfully."。

使用timeout函数可以很方便地处理长时间运行的函数，可以避免由于函数执行时间过长而导致程序无法继续执行的问题。