时间段后停止代码[重复]

Question

我想打电话给foo(n)，但如果它运行超过 10 秒就停止它。有什么好的方法可以做到这一点？

我可以看到理论上我可以修改 foo 本身以定期检查它已经运行了多长时间，但我不想这样做。

Answer 1

给你：

import multiprocessing
import time

# Your foo function
def foo(n):
    for i in range(10000 * n):
        print "Tick"
        time.sleep(1)

if __name__ == '__main__':
    # Start foo as a process
    p = multiprocessing.Process(target=foo, name="Foo", args=(10,))
    p.start()

    # Wait 10 seconds for foo
    time.sleep(10)

    # Terminate foo
    p.terminate()

    # Cleanup
    p.join()

这将等待foo 10 秒，然后将其杀死。

更新

只有在进程正在运行时才终止它。

# If thread is active
if p.is_alive():
    print "foo is running... let's kill it..."

    # Terminate foo
    p.terminate()

更新 2：推荐

将join 与timeout 一起使用。如果foo 在超时之前完成，那么 main 可以继续。

# Wait a maximum of 10 seconds for foo
# Usage: join([timeout in seconds])
p.join(10)

# If thread is active
if p.is_alive():
    print "foo is running... let's kill it..."

    # Terminate foo
    p.terminate()
    p.join()

Answer 2

import signal

#Sets an handler function, you can comment it if you don't need it.
signal.signal(signal.SIGALRM,handler_function) 

#Sets an alarm in 10 seconds
#If uncaught will terminate your process.
signal.alarm(10) 

超时时间不是很精确，但如果你不需要极高的精确度也可以。

另一种方法是使用资源模块，并设置最大CPU时间。