Posted on 2013-05-10 21:26
Onway 阅读(2717)
评论(0) 编辑 收藏 引用 所属分类:
码儿快跑
python程序里面需要执行一个系统命令程序，如果命令在限定时间之内结束，则python程序读取其输出（如果有）并马上返回，否则强行终止命令程序。
原本这个功能是用系统信号SIGALARM和python的异常解决的，但这不能用在多线程的环境里。然后考虑用threading.Timer进行计时，但这个计时是在一个单独线程进行的，如何将超时信息传给主线程也是一个问题。
百度一下，用select可以解决需求：
但select并不完美，当命令程序输出的内容多于管道容量的时候，select就会返回，如果此时命令程序再进入阻塞，则时间限制就不起作用了。
select.py:
import select
import subprocess
popen = subprocess.Popen("./test.sh", stdout=subprocess.PIPE)
fs = select.select([popen.stdout], [], [], 3)
if popen.stdout in fs[0]:
output = popen.stdout.read()
print len(output)
else:
print "timeout"
test.sh:
#!/bin/bash
# a.txt contains 65536 characters
cat a.txt
sleep 10
cat a.txt