您的位置:首页 > 编程语言 > Python开发

python 设置方法超时

2015-04-16 16:56 232 查看
#!/usr/bin/python
#-*-coding:utf-8-*-
import os,time,signal,platform,subprocess

class TimeoutError(Exception):
pass

def run_command(cmd, timeout=60):
is_linux = platform.system() == 'Linux'

p = subprocess.Popen(cmd, stderr=subprocess.STDOUT, stdout=subprocess.PIPE, shell=True, preexec_fn=os.setsid if is_linux else None)
t_beginning = time.time()
seconds_passed = 0
while True:
if p.poll() is not None:
break
seconds_passed = time.time() - t_beginning
if timeout and seconds_passed > timeout:
if is_linux:
os.killpg(p.pid, signal.SIGTERM)
else:
p.terminate()
raise TimeoutError(cmd, timeout)
time.sleep(0.1)
return p.stdout.read()

if __name__ == '__main__':
cmd = 'ping www.google.com
timeout = 10
try:
result = run_command(cmd, timeout)
except TimeoutError:
print 'excute command=<%s> timeout after %i' %(cmd,timeout)
else:
print = 'other error'
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: