在 Windows 上设置超时限制

Question

我有一个函数，运行时间不会太长。我想为其设置超时限制。我可以在互联网上找到一个建议的解决方案。请参阅以下 SO 帖子。函数调用超时

该解决方案使用信号，这在 Windows 上不可用。有类似的信号用于制作股票代码，它有一个 windows 端口，如这篇 SO 帖子中所述： python：Windows 相当于 SIGALRM 这不是直接对超时的回答，但可以适应超时工作。不过，它是为python 2.7编写的。

由于答案大约有 10 年的历史，我的问题是：是否有更现代的 python(例如 python 3.7)方法来创建上下文管理器/装饰器/类似的包装器确实使"普通函数"成为 Windows 系统上的超时限制函数？

Answer 1

这里有一种方法可以将@Praveenkumar的答案转换为易于使用的装饰器，您现在似乎想知道：

import time
import concurrent.futures as futures

def timeout(timelimit):
def decorator(func):
def decorated(*args, **kwargs):
with futures.ThreadPoolExecutor(max_workers=1) as executor:
future = executor.submit(func, *args, **kwargs)
try:
result = future.result(timelimit)
except futures.TimeoutError:
print('Timedout!')
raise TimeoutError from None
else:
print(result)
executor._threads.clear()
futures.thread._threads_queues.clear()
return result
return decorated
return decorator

@timeout(3)
def test(n):
print(f'Sleeping for {n} seconds')
time.sleep(n)
# Real code here.
return 'Done'
test(2)  # OK
test(5)  # -> Causes timeout.