如何在python中终止进程之前执行代码?

时间:2017-03-02 16:26:13

标签: python multiprocessing python-multiprocessing

这个问题涉及python中的多处理。我希望在终止进程时执行一些代码,在它终止之前更具体。我正在寻找一个解决方案,它可以作为atexit.register用于python程序。

我有一个方法工作者,看起来:

def worker():   而真:   打印(“工作”)   time.sleep(2)   返回

我通过以下方式运行:

proc = multiprocessing.Process(target = worker,args =())   proc.start()

我的目标是在终止之前执行一些额外的代码,我这样做:

proc.terminate()

1 个答案:

答案 0 :(得分:3)

使用信号处理和拦截SIGTERM:

import multiprocessing
import time
import sys
from signal import signal, SIGTERM

def before_exit(*args):
    print('Hello')
    sys.exit(0)  # don't forget to exit!


def worker():
    signal(SIGTERM, before_exit)
    time.sleep(10)

proc = multiprocessing.Process(target=worker, args=())
proc.start()
time.sleep(3)
proc.terminate()

在子进程终止之前生成所需的输出。