如何限制Python循环中函数调用的执行时长？

因为perhaps_long_exec()是不可修改的黑盒，且单线程/信号类方案在循环里容易误触发（信号可能打断循环本身而非目标函数），最可靠的解决方式是用多进程隔离每次函数调用——进程级的隔离能让我们安全终止超时的函数，完全不影响主循环的执行。

具体实现方案 #

1. 用multiprocessing.Process创建独立子进程来执行目标黑盒函数
2. 主进程等待子进程执行，同时设置超时时间
3. 一旦超时，直接终止子进程，然后继续下一次循环

完整代码示例 #

import random
import multiprocessing

def perhaps_long_exec():
    if random.random() < 0.5:
        while True:
            pass

def run_with_timeout(func, timeout):
    # 创建子进程执行目标函数
    proc = multiprocessing.Process(target=func)
    proc.start()
    # 等待指定时长，超时则终止进程
    proc.join(timeout)
    if proc.is_alive():
        proc.terminate()
        proc.join()  # 确保进程彻底退出
        print("函数调用超时，已终止进程")

if __name__ == "__main__":
    for i in range(100):
        print(f"第{i+1}次调用函数")
        run_with_timeout(perhaps_long_exec, timeout=2)  # 这里设置2秒超时，可按需调整

方案优势 #

• 精准控制：每次函数调用的超时逻辑独立，绝不会误打断主循环的执行流程
• 安全可靠：子进程的挂起、死循环不会影响主进程，超时后能彻底终止
• 兼容性强：不管黑盒函数内部是何种阻塞逻辑，都能有效限制执行时长

注意事项 #

• 如果需要给黑盒函数传参，要确保参数是可序列化的（符合Pythonpickle模块的要求）
• 频繁创建销毁进程会有轻微性能开销，若循环次数极多，可改用multiprocessing.Pool复用进程

内容的提问来源于stack exchange，提问作者olivarb