使用子进程终止正在运行的脚本

Question

我正在使用subprocess 执行python 脚本。我可以成功运行脚本。

import subprocess
import time

proc = subprocess.Popen(['python', 'test.py', ''], shell=True)
proc.communicate()
time.sleep(10)
proc.terminate()

test.py

import time
while True:
    print("I am inside test.py")
    time.sleep(1)
    

我可以看到消息I am inside test.py 每秒打印一次。但是，当脚本仍在运行时，我无法终止 proc.terminate()。

有人可以帮帮我吗？

Answer 1

proc.communicate 等待进程完成，除非您包含超时。您的进程不会退出并且您没有超时，因此communicate 永远不会返回。您可以在通话后使用print 进行验证。相反，添加超时并捕获其异常

import subprocess
import time

proc = subprocess.Popen(['python', 'test.py', ''], shell=True)
try:
    proc.communicate(timeout=10)
except subprocess.TimeoutExpired:
    proc.terminate()

Answer 2

首先，不要使用列表来传递subprocess.Popen() 的参数，如果您使用的是shell = True!，请将命令更改为string、"python test.py"。

Popen.communicate(input=None, timeout=None)是一个阻塞类方法，它与进程交互，等待进程终止并设置returncode属性。

由于您的test.py 运行无限while 循环，他将永远不会返回！

您有 2 个选项可让您生成的进程 proc 超时：

1. 分配超时关键字参数，例如将过程计时 5 秒，communicate(timeout=5) 方法。如果进程proc 在timeout 秒后没有终止，则会引发TimeoutExpired exception。捕获此异常并重试通信不会丢失任何输出（在您的情况下，您不需要子输出，但我将在下面的示例中提到这一点）。 注意如果timeout expires，子进程不会被杀死，因此为了正确清理，行为良好的应用程序应该杀死子进程（proc）并完成通信。
2. 通过使用poll 方法并通过您的调用方法进行计时。

超时通信

try:
    outs, errs = proc.communicate(timeout=15)
except TimeoutExpired:
    proc.kill()
    outs, errs = proc.communicate()

使用 time.sleep 进行投票

proc = subprocess.Popen('python test.py', shell=True)
t=10
while proc.poll() is None and t >= 0:
    print('Still sleeping')
    time.sleep(1)
    t -= 1
proc.kill()