python 限制函数计算时间

在Python中，如果你想要限制某个函数的运行时间，可以使用`time`模块配合异常处理来实现。一种常见的做法是使用`signal`模块或者`threading`模块结合计时器。这里我将展示一个简单的使用`time`模块的例子：

import time

def limited_execution(function, max_time, *args, **kwargs):
    start_time = time.time()
    
    def timeout_handler(signum, frame):
        if time.time() - start_time > max_time:
            raise TimeoutError("Function execution exceeded the maximum allowed time.")
    
    # 注册超时信号处理器
    signal.signal(signal.SIGALRM, timeout_handler)
    signal.alarm(max_time)  # 设置超时时长
    
    try:
        result = function(*args, **kwargs)
    except TimeoutError as e:
        print(f"Timeout occurred: {e}")
        return None  # 或者你想有的其他处理方式
    finally:
        # 移除定时器，防止多次触发
        signal.alarm(0)

    return result

# 调用函数并设置最大运行时间为5秒
limited_execution(your_function, 5, arg1, arg2)

在这个例子中，如果`your_function`在5秒内未完成，会抛出一个`TimeoutError`。你可以根据需要调整`max_time`。