以指定的最大运行时间运行外部程序

Question

我想在多线程python程序的每个线程中执行一个外部程序。

假设最大运行时间设置为1秒。如果启动的过程在1秒内完成，主程序将捕获其输出以供进一步处理。如果它在1秒内没有完成，主程序只是终止它并开始另一个新进程。

如何实现这个？

Answer 1

你可以定期轮询：

import subprocess, time 

s = subprocess.Popen(['foo', 'args']) 
timeout = 1 
poll_period = 0.1 
s.poll() 
while s.returncode is None and timeout > 0: 
    time.sleep(poll_period) 
    timeout -= poll_period 
    s.poll() 
if timeout <= 0: 
    s.kill() # timed out 
else: 
    pass # completed 

然后，您可以只把上面的功能，并启动它作为一个线程。

Answer 2

linux上令人讨厌的黑客攻击是使用timeout程序来运行该命令。但是，您可以选择更好的所有Python解决方案。

Answer 3

这是辅助函数使用：

def run_with_timeout(command, timeout): 
    import time 
    import subprocess 

    p = subprocess.Popen(command, shell=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE) 

    while timeout > 0: 
     if p.poll() is not None: 
      return p.communicate() 
     time.sleep(0.1) 
     timeout -= 0.1 
    else: 
     try: 
      p.kill() 
     except OSError as e: 
      if e.errno != 3: 
       raise 
    return (None, None) 

Answer 4

这里是使用pexpect模块（我需要捕捉程序的输出才跑进超时，我没能做到的解决方案这与subprocess.Popen）：

import pexpect 

timeout = ... # timeout in seconds 

proc = pexpect.spawn('foo', ['args'], timeout = timeout) 

result = proc.expect([ pexpect.EOF, pexpect.TIMEOUT]) 

if result == 0: 
    # program terminated by itself 
    ... 
else: 
    # result is 1 here, we ran into the timeout 
    ... 

print "program's output:", print proc.before