如何解决Python程序退出时内置函数被垃圾回收器提前删除的问题？

原问题脚本 #

import pickle

class MyClass(object):

    def __init__(self):
        self._some_data = "lorem ipsum dolor sit"

    def __del__(self):
        with open("/some/file/to/dump/data.pickle", 'wb') as ofile:
            pickle.dump(self._some_data, ofile)

my_instance = MyClass()
# 脚本结束时，垃圾回收器清理并删除my_instance，触发__del__()

问题说明 #

程序退出阶段，垃圾回收器的清理顺序没有严格保证，可能在my_instance被销毁前就回收了内置函数open的全局引用，导致__del__()执行时抛出NameError: name 'open' is not defined。

临时修复脚本 #

添加open = open语句后，脚本可正常执行：

import pickle
open = open  # 临时修复语句

class MyClass(object):

    def __init__(self):
        self._some_data = "lorem ipsum dolor sit"

    def __del__(self):
        with open("/some/file/to/dump/data.pickle", 'wb') as ofile:
            pickle.dump(self._some_data, ofile)

my_instance = MyClass()

技术疑问解答 #

1. 临时方案为何生效？原脚本为何无法阻止open被提前删除？ #

Python程序退出时，全局命名空间的清理顺序是不确定的。原脚本中，__del__()里的open指向全局命名空间中的内置函数，但垃圾回收器可能先清空全局的open引用，等到my_instance的__del__()执行时，全局已经找不到这个名字了。

而open = open是把内置open的引用复制到全局命名空间的open变量中（相当于多了一份引用）。只要my_instance还存在，它的__del__()就会引用这个全局变量，使得open函数的引用计数不会降到0，垃圾回收器就不会提前回收它。

2. 临时方案是否可靠？在其他Python版本中是否依然有效？ #

这个方案不算完全可靠。Python不同版本、不同实现（比如CPython、PyPy）的全局命名空间清理逻辑可能存在差异，虽然在多数CPython版本中能正常工作，但如果后续Python的退出清理机制变更，这个方案可能失效。

另外，如果代码后续修改了全局的open变量（比如重新赋值），会导致__del__()调用的不是原本的内置函数，引发新问题。所以这只是个临时 workaround，不能作为长期可靠方案。

3. 是否存在更优解决方案？ #

有几种更稳健的替代方案：

• 显式调用保存方法：完全放弃依赖__del__()，在脚本结束前主动调用保存逻辑。这是最可靠的方式，避开了垃圾回收的不确定性。

  import pickle
  
  class MyClass(object):
      def __init__(self):
          self._some_data = "lorem ipsum dolor sit"
  
      def save_data(self):
          with open("/some/file/to/dump/data.pickle", 'wb') as ofile:
              pickle.dump(self._some_data, ofile)
  
  my_instance = MyClass()
  my_instance.save_data()  # 主动调用保存
  

• 绑定builtins模块的open引用：从builtins模块直接获取open，并提前保存到实例属性中，不依赖全局命名空间的open。builtins模块的清理优先级通常靠后，能降低引用被提前回收的概率。

  import pickle
  import builtins
  
  class MyClass(object):
      def __init__(self):
          self._some_data = "lorem ipsum dolor sit"
          self._open = builtins.open  # 提前保存内置open的引用
  
      def __del__(self):
          with self._open("/some/file/to/dump/data.pickle", 'wb') as ofile:
              pickle.dump(self._some_data, ofile)
  
  my_instance = MyClass()
  

• 使用上下文管理器：将类实现为上下文管理器，在__exit__()方法中完成数据保存。当with块结束时会自动触发保存，无需依赖垃圾回收。

  import pickle
  
  class MyClass(object):
      def __init__(self):
          self._some_data = "lorem ipsum dolor sit"
  
      def __enter__(self):
          return self
  
      def __exit__(self, exc_type, exc_val, exc_tb):
          with open("/some/file/to/dump/data.pickle", 'wb') as ofile:
              pickle.dump(self._some_data, ofile)
  
  with MyClass() as my_instance:
      # 执行业务逻辑
      pass
  

内容的提问来源于stack exchange，提问作者Leander Moesinger