如何确保所有weakref finalizer执行完毕后再运行顶层退出清理函数

Python的atexit退出钩子严格遵循*后注册、先执行（LIFO）*的顺序规则。查看weakref模块源码可以确认：模块内置的全局finalizer执行钩子，只会在第一次实例化weakref.finalize对象时才会注册到atexit队列中，这个钩子负责在解释器退出阶段批量执行所有已注册的finalizer回调。

如果用户代码在导入你的模块前就已经创建过finalize实例，weakref的全局执行钩子注册时间就会早于你通过atexit.register注册的模块顶层清理函数。退出时你的模块顶层清理会先执行，之后才轮到所有finalizer（包括你模块内对象的清理回调）运行，顺序完全不受模块本身控制，和用户代码行为强绑定。

不要依赖全局atexit的注册顺序保证执行先后，改为在模块内部自主跟踪本模块创建的所有finalizer状态，从逻辑上强制顶层清理仅在本模块所有对象的清理全部完成后触发，完全不受外部代码行为影响。

修改后的模块实现代码如下：

# module.py
from weakref import finalize
import atexit

# 模块级状态跟踪
_pending_obj_cleanup = 0  # 待执行清理的本模块对象计数
_top_cleanup_triggered = False  # 退出流程已启动标记
_top_cleanup_done = False  # 顶层清理已执行标记，防止重复调用

class MyObject:
    @staticmethod
    def _cleanup():
        global _pending_obj_cleanup
        print('Cleaning up module object')
        _pending_obj_cleanup -= 1
        # 所有本模块对象清理完成、且已进入退出流程时，直接触发顶层清理
        if _pending_obj_cleanup == 0 and _top_cleanup_triggered:
            _do_top_cleanup()
    
    def __init__(self):
        global _pending_obj_cleanup
        finalize(self, MyObject._cleanup)
        _pending_obj_cleanup += 1

def _do_top_cleanup():
    global _top_cleanup_done
    if _top_cleanup_done:
        return
    _top_cleanup_done = True
    print('Cleaning up module')

def _module_atexit_handler():
    global _top_cleanup_triggered
    _top_cleanup_triggered = True
    # 进入退出流程时如果已经没有待清理对象，直接执行顶层清理
    if _pending_obj_cleanup == 0:
        _do_top_cleanup()

# 保险操作：强制提前触发weakref的atexit钩子注册，消除边界场景影响
_temp_guard = object()
finalize(_temp_guard, lambda: None)
del _temp_guard

# 注册模块自身的退出处理器
atexit.register(_module_atexit_handler)

• 模块通过计数器跟踪所有未完成清理的内部对象，每创建一个MyObject实例计数加1，每完成一个实例的清理计数减1。
• 退出阶段首先触发模块自身的atexit处理器，标记进入退出流程：如果此时所有对象已经被GC清理完成，直接执行顶层清理；如果还有未清理的对象，就等待最后一个对象的清理回调执行完成时，自动触发顶层清理。
• 开头创建临时可回收对象的操作是保险逻辑，无论用户代码是否提前创建过finalizer，都能保证weakref的全局钩子不会打乱本模块的清理流程，且不会产生任何额外副作用。

这个实现完全不干涉用户侧自身的finalizer执行顺序，仅保证本模块的对象清理回调全部执行完毕后，才会运行本模块的顶层清理逻辑，不受用户代码导入顺序、对象创建时机的影响。

内容的提问来源于stack exchange，提问作者Craig White