web-dev-qa-db-ja.com

親が死んだときにsubprocess.check_output（）で作成されたpython子プロセス）を殺す方法は？

Linuxマシンで実行しているpythonスクリプトは、subprocess.check_output（）を使用して次のように子プロセスを作成します。

subprocess.check_output(["ls", "-l"], stderr=subprocess.STDOUT)

問題は、親プロセスが停止しても、子プロセスがまだ実行されていることです。親が死んだときにも子プロセスを殺すことができる方法はありますか？

pythonlinuxsubprocess
27
Clara

あなたの問題はsubprocess.check_outputの使用にあります-あなたは正しいです、あなたはそのインターフェースを使用して子PIDを取得することはできません。代わりにPopenを使用します。

proc = subprocess.Popen(["ls", "-l"], stdout=PIPE, stderr=PIPE)

# Here you can get the PID
global child_pid
child_pid = proc.pid

# Now we can wait for the child to complete
(output, error) = proc.communicate()

if error:
    print "error:", error

print "output:", output

終了時に子を確実に殺すには：

import os
import signal
def kill_child():
    if child_pid is None:
        pass
    else:
        os.kill(child_pid, signal.SIGTERM)

import atexit
atexit.register(kill_child)
21
cdarke

はい、2つの方法でこれを達成できます。両方とも、check_outputの代わりにPopenを使用する必要があります。最初の方法は、次のようにtry..finallyを使用した簡単な方法です。

from contextlib import contextmanager

@contextmanager
def run_and_terminate_process(*args, **kwargs):
try:
    p = subprocess.Popen(*args, **kwargs)
    yield p        
finally:
    p.terminate() # send sigterm, or ...
    p.kill()      # send sigkill

def main():
    with run_and_terminate_process(args) as running_proc:
        # Your code here, such as running_proc.stdout.readline()

これはsigint（キーボード割り込み）とsigtermをキャッチしますが、sigkill（-9でスクリプトを強制終了した場合）はキャッチしません。

もう1つの方法はもう少し複雑で、ctypesのprctl PR_SET_PDEATHSIGを使用します。何らかの理由で（sigkillであっても）親が終了すると、システムは子にシグナルを送信します。

import signal
import ctypes
libc = ctypes.CDLL("libc.so.6")
def set_pdeathsig(sig = signal.SIGTERM):
    def callable():
        return libc.prctl(1, sig)
    return callable
p = subprocess.Popen(args, preexec_fn = set_pdeathsig(signal.SIGTERM))
24
micromoses

詳細はわかりませんが、最良の方法は、シグナル（おそらくはすべてのエラー）をシグナルでキャッチし、そこに残っているプロセスを終了することです。

import signal
import sys
import subprocess
import os

def signal_handler(signal, frame):
    sys.exit(0)
signal.signal(signal.SIGINT, signal_handler)

a = subprocess.check_output(["ls", "-l"], stderr=subprocess.STDOUT)

while 1:
    pass # Press Ctrl-C (breaks the application and is catched by signal_handler()

これは単なるモックアップであり、SIGINT以上のものをキャッチする必要がありますが、そのアイデアから始めることができ、まだ生成されたプロセスをチェックする必要があります。

http://docs.python.org/2/library/os.html#os.killhttp://docs.python.org/2/library/subprocess.html# subprocess.Popen.pidhttp://docs.python.org/2/library/subprocess.html#subprocess.Popen.kill

Check_outputは実行中にあまり対話できないため、check_outputは実際には単純なデバッグなどのためだけであることに気付いたので、check_output causeのパーソナライズバージョンを書き換えることをお勧めします。

Check_outputを書き換えます。

from subprocess import Popen, PIPE, STDOUT
from time import sleep, time

def checkOutput(cmd):
    a = Popen('ls -l', Shell=True, stdin=PIPE, stdout=PIPE, stderr=STDOUT)
    print(a.pid)
    start = time()
    while a.poll() == None or time()-start <= 30: #30 sec grace period
        sleep(0.25)
    if a.poll() == None:
        print('Still running, killing')
        a.kill()
    else:
        print('exit code:',a.poll())
    output = a.stdout.read()
    a.stdout.close()
    a.stdin.close()
    return output

そして、あなたがそれで好きなことをして、おそらくアクティブな実行を一時変数に保存し、シグナルまたはメインループのエラー/シャットダウンを受け入れる他の手段で終了時にそれらを強制終了します。

最終的には、子を安全に殺すためにメインアプリケーションで終了をキャッチする必要があります。これにアプローチする最善の方法は、try & exceptまたはsignalを使用することです。

1
Torxed