在 python 中终止程序

Question

您好，我在 python 代码中嵌入了时间约束，该代码正在运行带有函数的 fortran 代码。但是我意识到对另一个函数施加时间限制的函数不会终止代码，只是将其留在后台并跳过它。我想要做的是在代码超过限制时间时终止代码。这是我用来限制时间的代码，取自here。

def timeout(func, args=(), kwargs={}, timeout_duration=15, default=1):
    import signal

    class TimeoutError(Exception):
        pass

    def handler(signum, frame):
        raise TimeoutError()

    # set the timeout handler                                                                                                                                             
    signal.signal(signal.SIGALRM, handler)
    signal.alarm(timeout_duration)
    try:
        result = func(*args, **kwargs)
    except TimeoutError as exc:
        result = default
    finally:
        signal.alarm(0)

    return result

我查找了 popen.terminate()、sys.exit() 和 atexit.register()，但无法弄清楚它如何与我尝试在下面添加的这段代码一起工作，我在评论中显示。

...
        result = func(*args, **kwargs)
    except TimeoutError as exc:
        result = default
#add the terminator
    finally:
...

注意：函数位于 for 循环链中，因此我不想终止整个 python 会话，而只想终止该函数运行的程序，该程序是一个 fortran 代码并跳到 for 循环链中的另一个元素。

在一些cmets和答案之后添加以下部分：

我尝试使用 popen.terminate() 添加 SIGTERM，但是它终止了所有 python 会话，我只想终止当前正在运行的会话并跳到 for 循环中的其他元素。我所做的如下：

...
    signal.signal(signal.SIGTERM, handler) 
    signal.alarm(timeout_duration)
    try:
        result = func(*args, **kwargs)
    except TimeoutError as exc:
        result = default
        popen.terminate()
...

Answer 1

您不能期望引发异常的信号处理程序通过调用堆栈传播，它是在不同的上下文中调用的。

popen.terminate() 将在 posix 系统上生成 SIGTERM，因此您应该有一个用于 SIGTERM 而不是 SIGALRM 的信号处理程序。

与其在信号处理程序中引发异常，不如设置一些您定期检查的变量以停止活动。

或者，如果您没有SIGTERM 的信号处理程序，默认处理程序可能会生成KeyboardInterrupt 异常。