如何在Python中限制函数调用的执行时间

Question

如何在Python中限制函数调用的执行时间在我的代码中有一个与套接字相关的函数调用，该函数来自另一个模块，因此我无法控制，问题是它偶尔阻塞几个小时，这是完全不能接受的，我如何从代码中限制函数的执行时间？我想解决方案必须利用另一个线程。

Answer 1

使用with陈述为了给超时功能一些语法糖：

import signalfrom contextlib import contextmanagerclass TimeoutException(Exception): pass@contextmanagerdef time_limit(seconds):
    def signal_handler(signum, frame):
        raise TimeoutException("Timed out!")
    signal.signal(signal.SIGALRM, signal_handler)
    signal.alarm(seconds)
    try:
        yield
    finally:
        signal.alarm(0)try:
    with time_limit(10):
        long_function_call()except TimeoutException as e:
    print("Timed out!")