Python实现函数超时终止并执行后续操作的方法咨询

刚好碰到过类似的需求，给你两种靠谱的实现思路，正好对应你提到的SIGALARM信号和异步进程方案：

方法一：用SIGALARM信号实现超时（Unix/Linux/macOS专属） #

这是利用系统信号机制的轻量级方案，思路是给主线程设置一个闹钟，超时后触发SIGALRM信号，我们捕获这个信号并抛出异常，从而中断目标函数的执行。

具体代码实现如下：

import signal

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

# 信号处理函数：收到SIGALRM时抛出超时异常
def timeout_handler(signum, frame):
    raise TimeoutError("函数执行超过设定时长")

def may_take_a_long_time():
    while 1:
        # do something
        if met_condition():
            break

timeout_dur = 600

# 设置信号处理
signal.signal(signal.SIGALRM, timeout_handler)
signal.alarm(timeout_dur)  # 启动闹钟，timeout_dur秒后触发信号

try:
    may_take_a_long_time()
except TimeoutError:
    print("函数执行超时，已终止")
finally:
    signal.alarm(0)  # 无论正常结束还是超时，都关闭闹钟

do_something_else()

注意事项： #

• 这个方法只能在主线程使用，因为Python的信号处理逻辑是由主线程负责的，子线程无法接收信号。
• Windows系统不支持SIGALARM，所以只适用于Unix/Linux/macOS。
• 如果你的may_take_a_long_time函数内部有自己的信号处理逻辑，可能会和这个方案冲突，需要提前协调。

方法二：用多进程实现通用超时控制（跨平台） #

如果需要跨平台支持，或者不想受信号机制的限制，用多进程是更稳妥的选择。思路是把耗时函数放到子进程中运行，主线程等待指定时长，超时就强制终止子进程，然后继续执行后续代码。

代码示例：

import multiprocessing

def may_take_a_long_time():
    while 1:
        # do something
        if met_condition():
            break

def do_something_else():
    print("执行后续任务")

timeout_dur = 600

# 创建子进程，指定要执行的函数
proc = multiprocessing.Process(target=may_take_a_long_time)
proc.start()

# 等待子进程结束，最多等待timeout_dur秒
proc.join(timeout_dur)

# 如果子进程还活着，说明超时了，强制终止它
if proc.is_alive():
    print("函数执行超时，已终止子进程")
    proc.terminate()
    proc.join()  # 等待进程完全终止，避免僵尸进程

# 继续执行后续任务
do_something_else()

优势说明： #

• 跨平台支持，Windows、Unix系统都能用。
• 不依赖信号机制，不会和函数内部的信号处理冲突。
• 可以在任何线程中使用，没有主线程限制。
• 进程可以被强制终止，哪怕目标函数是死循环也能生效。

小提示： #

如果需要从may_take_a_long_time获取返回值，可以用multiprocessing.Queue或者Pipe来实现进程间通信，稍微麻烦一点，但完全可行。

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