Python中的进程执行检查和获取PID

Question

我需要在后台运行一个 bash 命令，但稍后需要 kill(os.kill()) 它。我还想确保命令运行我有这个以确保命令运行。

if subprocess.Popen("tcpdump -i eth0 -XX -w /tmp/tmp.cap &", shell=True).wait() == 0:

我不确定如何更改此设置，因此我可以使用 Popen.pid 获取 pid，同时仍然能够检查执行是否成功。 任何帮助，将不胜感激。 谢谢。

Answer 1

要启动一个子进程，等待一段时间并终止它，并检查它的退出状态是否为零：

import shlex
from subprocess import Popen
from threading import Timer

def kill(process):
    try:
        process.kill()
    except OSError: 
        pass # ignore

p = Popen(shlex.split("tcpdump -i eth0 -XX -w /tmp/tmp.cat"))
t = Timer(10, kill, [p]) # run kill in 10 seconds
t.start()
returncode = p.wait()
t.cancel()
if returncode != 0:
   # ...

或者您可以自己实现超时：

import shlex
from subprocess import Popen
from time import sleep, time as timer # use time.monotonic instead

p = Popen(shlex.split("tcpdump -i eth0 -XX -w /tmp/tmp.cat"))

deadline = timer() + 10 # kill in 10 seconds if not complete
while timer() < deadline:
    if p.poll() is not None: # process has finished
        break 
    sleep(1) # sleep a second
else: # timeout happened
    try:
        p.kill()
    except OSError:
        pass

if p.wait() != 0:
   # ...

假设sleep 使用与timer 相似的时钟。

threading.Timer 变体允许您的代码在子进程退出后立即继续。

Answer 2

使用Popen.poll() 方法。您也可以获取Popen.returncode 来确定流程是否成功完成。

import subprocess

tasks = [subprocess.Popen('ping www.stackoverflow.com -n 5 && exit 0', shell=True),
         subprocess.Popen('ping www.stackoverflow.com -n 5 && exit 1', shell=True)]

for task in tasks:
    while task.poll() is None:
        # the task has not finished
        pass

    print task
    print task.pid
    print task.returncode