Python3のmultiprocessingで特に何も言われずに終了したい

Question

前提・実現したいこと

Python3.6.8 にて、multiprocessing を使用している最中に keyboardInterrupt して正常終了したい

発生している問題・エラーメッセージ

Error in atexit._run_exitfuncs:
Traceback (most recent call last):
  File "/usr/lib/python3.6/multiprocessing/popen_fork.py", line 28, in poll
    pid, sts = os.waitpid(self.pid, flag)
KeyboardInterrupt

該当のソースコード

import multiprocessing
import time

def hoge(n):
    try:
        while True:
            print("hoge", n)
            time.sleep(n)
    except KeyboardInterrupt:
        pass

for i in [1, 2]:
    multiprocessing.Process(target=hoge, args=(i,)).start()

Answer 1

キー割り込みの例外を捕捉する処理を施した、Process.join() を追加してみてください

import multiprocessing
import time

def hoge(n):
    try:    
        while True:
            time.sleep(n)
    except KeyboardInterrupt:
        pass

processes = []
for i in [1, 2]:
    p = multiprocessing.Process(target=hoge, args=(i,))
    p.start()
    processes.append(p)
for p in processes:
    try:
        p.join()
    except KeyboardInterrupt:
        # 割り込み処理でプロセスが終了してるのでJoinできるはず
        p.join(1)

Comments

[shirasublue]

できました。ありがとうございます。

join()でブロックすればいいんですね。

except でもう一度タイムアウト付きのjoinをしているのは、
子が例外割り込みでも終わらない場合の保険という認識でよろしいでしょうか？

[magichan]

はい。そのとおりです。
基本的には絶対に成功すると思うのですが、万が一があるので。