Я пытаюсь использовать atexit в Process, но, к сожалению, он не работает. Вот пример кода:

import time
import atexit
import logging
import multiprocessing

logging.basicConfig(level=logging.DEBUG)

class W(multiprocessing.Process):
    def run(self):
        logging.debug("%s Started" % self.name)

        @atexit.register
        def log_terminate():
             # ever called?
             logging.debug("%s Terminated!" % self.name)

        while True:
            time.sleep(10)

@atexit.register
def log_exit():
    logging.debug("Main process terminated")

logging.debug("Main process started")

a = W()
b = W()
a.start()
b.start()
time.sleep(1)
a.terminate()
b.terminate()

Вывод этого кода:

DEBUG:root:Main process started
DEBUG:root:W-1 Started
DEBUG:root:W-2 Started
DEBUG:root:Main process terminated

Я ожидал бы, что W.run.log_terminate() будет вызываться, когда вызываются a.terminate() и b.terminate(), а вывод должен быть чем-то подобным (выделено мной)!:

DEBUG:root:Main process started
DEBUG:root:W-1 Started
DEBUG:root:W-2 Started
DEBUG:root:W-1 Terminated!
DEBUG:root:W-2 Terminated!
DEBUG:root:Main process terminated

Почему это не работает, и есть ли лучший способ для регистрации сообщения (из контекста Process), когда Process завершается?

Спасибо за ваш вклад - он очень ценится.

Решение

РЕДАКТИРОВАТЬ: Основываясь на решении, предложенном Алексом Мартелли, следующее работает как ожидалось:

import sys
import time
import atexit
import signal
import logging
import multiprocessing

logging.basicConfig(level=logging.DEBUG)

class W(multiprocessing.Process):
    def run(self):
        logging.debug("%s Started" % self.name)

        def log_terminate(num, frame):
             logging.debug("%s Terminated" % self.name)
             sys.exit()
        signal.signal(signal.SIGTERM, log_terminate)
        while True:
            time.sleep(10)

@atexit.register
def log_exit():
    logging.debug("Main process terminated")

logging.debug("Main process started")
a = W()
b = W()
a.start()
b.start()
time.sleep(1)
a.terminate()
b.terminate()

Стоит отметить следующий комментарий в документации atexit:

Примечание: функции, зарегистрированные через этот модуль, не вызываются, когда программа убивается сигналом, когда обнаружена фатальная внутренняя ошибка Python или когда вызывается os._exit().

Процесс Python не будет вызывать atexit

Решение

Ответ 1