无法从 python 2.7 中的 atexit 注销函数

Question

首先，我用flush方法写了一个记录类：

class Recorder
    def __init__(self, buffer_size, path):
        self._big_buffer = np.array(*buffer_size)
        self._path = path
    def push(self, data):
        # insert in self._big_buffer
        # if self._big_buffer is full:
        #     self._flush()
    def flush(self):
        # write buffer to disk (self._path)

然后，我想在退出时刷新：手动停止时，崩溃或任何原因。

所以我用了：

def __init__(self):
    (...)
    atexit.register(self.flush)

而且效果很好。

但是现在，我想用不同的缓冲区大小和不同的路径进行多次录制、停止录制、再次录制。所以我不得不丢弃，然后实例化几个Recorder。它有点工作，但旧的Recorder 的内存（包含一些胖self._big_buffer̀）没有被释放，因为它被atexit 保留。即使我明确调用del。 我不能atexit.unregister(self._flush)，因为它只是 Python 3。

我不希望重复使用现有实例，而是丢弃旧实例并创建新实例。

遇到这种情况你会怎么处理？

Answer 1

您可以尝试使用对atexit 处理程序的弱引用，这样对象 如果在别处删除则不会保留：

import atexit
import weakref

class CallableMethodWeakRef:
    def __init__(self, object, method_name):
        self.object_ref = weakref.ref(object)
        self.method_name = method_name
    def __call__(self):
        object = self.object_ref()
        if object:
            getattr(object, self.method_name)()

class Recorder:
    def __init__(self, *args):
        atexit.register(CallableMethodWeakRef(self, 'flush'))

    def flush(self):
        print 'flushing'

该方法作为字符串传递以避免很多问题 绑定方法弱引用，如果您发现它令人不安，您可以随时使用 像这样的BoundMethodWeakref 实现：http://code.activestate.com/recipes/578298-bound-method-weakref/

Answer 2

我会说您尝试使用错误的工具。 with 语句和上下文管理器是一个非常好的工具。文件 IO 是大多数 Python 用户会接触到 with 语句的主要示例。

f = open("somefile.txt", "w")
try:
    f.write("...")
    # more file operations
finally:
    # regardless of what happens, make sure the files is closed 
    f.close()

变成：

with open("somefile.txt", "w") as f:
    f.write("...")
    # more file operations
# close automatically called at the end of the block

您可以通过为您的类编写__enter__ 和__exit__ 方法来创建自己的上下文管理器。

class Recorder
    def __init__(self, buffer_size, path):
        self._big_buffer = np.array(*buffer_size)
        self._path = path
    def push(self, data):
        # insert in self._big_buffer
        # if self._big_buffer is full:
        #     self._flush()
    def flush(self):
        # write buffer to disk (self._path)
    def __enter__(self):
        return self
    def __exit__(self, exctype, exception, traceback):
        # If an exception was thrown in the with block you will get the details here. 
        # If you want the say that the exception has been handled and for it not to be 
        # raised outside the with block then return True
        self.flush()
        # self.close() ?

然后您将使用您的 Recorder 对象，例如：

with Recorder(...) as recorder:
    # operations with recorder
    ...
# regardless of what happens the recorder will be flushed at this point

Answer 3

答案当然是 允许您的Recorder 随意更改路径和缓冲区特征。你说 “我宁愿不重用现有实例，而是丢弃旧实例并创建新实例。” 但你没有给出任何理由， 除非您假设 “旧 Recorder 的内存（包含一些胖 self._big_buffer̀）没有被释放，因为它被 atexit 保留”，我认为这是不正确的。

虽然atexit 确实保留了对记录器对象的引用，但这仅意味着只要记录器引用它，缓冲存储器就会被保留。添加close() 方法很容易，例如

    def close(self):
        self.flush()
        self._big_buffer = None

还有宾果游戏！不存在对缓冲内存的引用，它是可回收的。

您的__init__() 方法应该简单地向atexit 注册，然后open() 方法（它完成了__init__() 当前所做的其余部分）可以多次使用，每次调用后跟一个close() .

总之，我认为你的问题需要一个对象。

Answer 4

您可以从（未记录的）atexit._exithandlers 列表中手动移除句柄。

import atexit

def unregister(func, *targs, **kargs):

    """unregister a function previously registered with atexit.
       use exactly the same aguments used for before register.
    """
    for i in range(0,len(atexit._exithandlers)):
        if (func, targs, kargs) == atexit._exithandlers[i] :
            del atexit._exithandlers[i]
            return True
    return False

希望对您有所帮助。