在Python2.x中的幾個用來執行命令行的庫或函數在執行命令是均不能設置一個命令執行的超時時間,用來在命令執行時間超時時終端這個命令的執行,這個功能在3.x(?)中解決了,但是在2.x還是只能自己實現。
下面就簡單實現了一個版本:
import subprocess
from threading import Timer
def call(args, timeout):
p = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
timer = Timer(timeout, lambda process: process.kill(), [p])
try:
timer.start()
stdout, stderr = p.communicate()
return_code = p.returncode
return (stdout, stderr, return_code)
finally:
timer.cancel()
測試
print call(['hostname'], 2)
print call(['ping', 'www.baidu.com'], 2)