如何运行具有特定持续时间的函数？ [复制]

Question

我是 python 新手，想知道如何设置函数运行的特定时间。

考虑到这里我有一个名为 df 的包含 891 行的数据框，它的列名 body_text 我想通过一行数据帧并将它们存储到一个名为 df_summary 的新数据帧中。在这里，我的目标是设置一个函数 summary 最多运行 2 分钟，如果它可以在 2 分钟之前完成并进入下一次迭代，那就太好了。如果不是，我想调用另一个函数名称 preprocess 而不是无法在 120 秒内完成的函数 summary。

假设我们已经创建了具有 891 行和列名 body_text 的 df_summary。这是我的代码+一些伪代码，因为我不知道如何准确地编写它：

import time 

for i in tqdm(range(0,891)):
  # code to detect running time here
    df_summary['body_text'][i] = summary(df['body_text'][i])# perform the summary function with at most 120s 
  # when 120s is over but the summary function cannot finish than perform a preprocessing function instead and move to the next iteration
    df_summary['body_text'][i] = preprocess(df['body_text'][i])

请注意，如果函数摘要最多可以运行 120 秒或低于该限制时间，则它不会运行回退函数 preproces，然后移动到下一次迭代。

我怎样才能做到这一点？任何帮助将不胜感激。

Answer 1

这是我的答案：

import sys
import threading
try:
    import thread
except ImportError:
    import _thread as thread


def exit_after(s):
    '''
    use as decorator to exit process if 
    function takes longer than s seconds
    '''

    def quit_function(fn_name):
        # print to stderr, unbuffered in Python 2.
        sys.stderr.flush() # Python 3 stderr is likely buffered.
        thread.interrupt_main() # raises KeyboardInterrupt

    def outer(fn):
        def inner(*args, **kwargs):
            timer = threading.Timer(s, quit_function, args=[fn.__name__])
            timer.start()
            try:
                result = fn(*args, **kwargs)
            finally:
                timer.cancel()
            return result
        return inner
    return outer

# I DIDN'Y WRITE THE CODE ABOVE I GOT IT FROM THERE : https://stackoverflow.com/questions/492519/timeout-on-a-function-call

@exit_after(120) # the function will kill itself after 120 seconds
def process(data): 
    summary(data)

for i in tqdm(range(0,891)):
    try:
        df_summary['body_text'][i] = process(df['body_text'][i])
    except KeyboardInterrupt: # took over 120 seconds
        df_summary['body_text'][i] = preprocess(df['body_text'][i])

120s 后停止处理，运行预处理功能。 有什么不懂的告诉我