subprocessを使って後からメソッドを実行したい

Question

実現したいこと

カウントを表示し続けるプログラムをsubprocessを用いて後からカウントを停止させたい。

前提

pythonで以下のようなcnt.pyを書いています。
毎秒カウントをするコードです。

import time

is_start = True
cnt = 1

def stop():
    is_start = False

while is_start:
    print(cnt)
    time.sleep(1)
    cnt += 1

これを以下の様なsub.pyで実行します。

import subprocess
import time
p = subprocess.Popen(['python', 'cnt.py'], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

これではcnt.pyは永遠にカウントをし続けてしまいます。
5秒後にsub.pyからsubprocessを使ってカウントを停止させる方法を教えてください。

Answer 1

5秒後にsub.pyからsubprocessを使ってカウントを停止させる方法

cnt.py を実行している Python プロセスの stdin へ sub.py から文字列を出力する方法もあるかと思います。cnt.py 側では stdin への入力を別スレッドで待ちます。
※ カウントアップの状況が判らないので、cnt.py を実行している Python プロセスの stdout を sub.py 側の stdout にリダイレクトしています

sub.py

import subprocess
import sys
import threading

p = subprocess.Popen(['python', 'cnt.py'],
                     stdin=subprocess.PIPE, stdout=sys.stdout, stderr=subprocess.PIPE,
                     encoding='utf-8')

# send stop command after 5 seconds
t = threading.Timer(5, lambda: p.stdin.write('stop\n'))
t.start()
t.join()
print('stopped.')

cnt.py

import sys
import threading
import time

def stop():
    global is_start
    is_start = False

def read_command():
    cmd = sys.stdin.readline().rstrip()
    if cmd == 'stop': stop()

if __name__ == '__main__':
    is_start = True
    cnt = 1

    thread = threading.Thread(target=read_command)
    thread.start()

    while is_start:
        print(cnt)
        time.sleep(1)
        cnt += 1

    thread.join()

実行結果

$ python sub.py
1
2
3
4
5
stopped.

Answer 2

プロセスの外からそのプロセスを強制的に止めるというのはあまりよい作法ではありません。方法はあります。

また、cnt.py の stop()関数を外から呼ぶ方法があるのではないかと思っているのだと思いますが、そのようなことはできません。

プロセスを外から止める場合、プロセス間通信という仕組みを使います。プロセス間通信にはいろいろな方法があります。調べればいろいろ出てくるので適切なものを選べばいいでしょう。

一例としては、ファイルを使った方法です。
止められる側のプロセスが、特定のディレクトリの特定の名前のファイルを定期的に監視し、そのファイルが有れば(もしくは無ければ)自分は停止するようにします。

Answer 3

cnt.pyで、５秒経過で停止するようにコードを組みましょう

Comments

[taka]

そういう話ではありません。
やりたいことはsubprocessで実行したプログラムに後からきっかけを与えてstop()を実行させることができるかという話です。
それを簡略化して伝えるためにカウントのコードを提示しているだけです。そりゃやりたいことが5秒後に停止させるということならばこんな回りくどいコードは書きません^^;

[y_waiwai]

なら、相互に通信させるようにして、cnt.py側のコードで、何らかの信号を受信したらコードを終了させるようにする、ように組みましょう