Python subprocess设置超时不生效问题

使用Python脚本写了个测试程序,测试程序中会执行SHELL脚本,并且设置了超时时间,大概是这样的:

subprocess.run("do something", shell=True, stdout=subprocess.PIPE, timeout=30.0)

但是却发现运行的时候,测试程序一直无法执行完成,通过ps -efpystack发现进程一直在等待"do something"进程,这个进程的状态是,就是僵尸进程,不过这个进程还创建了子进程。很明显,这个超时并没有生效。
于是Google了一下,确实有类似的情况。网上的说法是shell=Truestdout=subprocess.PIPE时会有问题,原因是python的subprocess库在捕获到TimeoutExpire异常时,会执行 p.kill,然后执行p.communicate,也就是他还会等待那个脚本的子进程结束。
这个说法经过测试不完全正确,我把shell=True去掉也有这个问题。于是按照自己的改法,变成了这样子:

with subprocess.Popen(command.split(), 
                          stdout=subprocess.PIPE, 
                          stderr=subprocess.STDOUT, 
                          universal_newlines=True) as proc:
      try:
        stdout, stderr = proc.communicate(timeout=timeout)
        returncode = proc.poll()
      except subprocess.TimeoutExpired:
        # os.killpg(os.getpgid(proc.pid), signal.SIGKILL) # 使用killpg会把自己杀死
        os.kill(proc.pid, signal.SIGKILL)

注意:网上还有人提到使用 os.killpg 杀死所有子进程,不过这会把自己也杀死。但是不适用os.killpg不能把所有子进程杀死,所以正确的做法是找到所有的子进程,然后杀死他们。

我后面阅读过psutil的手册,发现psutil可以很方便的把子进程都列出来,所以这个代码改成psutil会更好。不过我没测试也就先不写出来了。

参考资料
Subprocess timeout failure
Using module ‘subprocess’ with timeout
Python subprocess timeout?

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值