加入的 Python 进程不会调用 atexit

Question

我认为 Python 进程在终止时会调用它们的 atexit 函数。请注意，我使用的是 Python 2.7。这是一个简单的例子:

from __future__ import print_function
import atexit
from multiprocessing import Process


def test():
    atexit.register(lambda: print("atexit function ran"))

process = Process(target=test)
process.start()
process.join()

我希望它打印出“atexit function ran”，但它没有。

注意这个问题: Python process won't call atexit 类似，但它涉及以信号终止的进程，答案涉及拦截该信号。这个问题中的进程正在正常退出，所以(据我所知)这个问题和答案不适用(除非这些进程由于某种信号而退出？)。

Answer 1

我通过查看 CPython 中的实现方式做了一些研究。这是假设您在 Unix 上运行。如果您在 Windows 上运行，则以下内容可能无效，因为 multiprocessing 中进程的实现不同。

事实证明，os._exit() 总是在进程结束时被调用。这与 atexit 文档中的以下注释一起，应该解释为什么您的 lambda 没有运行。

Note: The functions registered via this module are not called when the program is killed by a signal not handled by Python, when a Python fatal internal error is detected, or when os._exit() is called.

---

这是 CPython 2.7 的 Popen 类的摘录，用于 fork 进程。请注意， fork 进程的最后一条语句是对 os._exit() 的调用。

# Lib/multiprocessing/forking.py

class Popen(object):

    def __init__(self, process_obj):
        sys.stdout.flush()
        sys.stderr.flush()
        self.returncode = None

        self.pid = os.fork()
        if self.pid == 0:
            if 'random' in sys.modules:
                import random
                random.seed()
            code = process_obj._bootstrap()
            sys.stdout.flush()
            sys.stderr.flush()
            os._exit(code)

在 Python 3.4 中，如果您开始 fork 进程，os._exit() 仍然存在，这是默认设置。但似乎您可以更改它，请参阅 Contexts and start methods了解更多信息。我还没有尝试过，但也许使用 spawn 的启动方法会起作用吗？但不适用于 Python 2.7。