python - 在 Python 中 : How to remove an object from a list if it is only referenced in that list?

Question

我想跟踪当前正在使用的某种类型的对象。例如:跟踪一个类的所有实例或由元类创建的所有类。

很容易跟踪这样的实例:

class A():
    instances = []
    def __init__(self):
        self.instances.append(self)

但是，如果一个实例在该列表之外的任何地方都没有被引用，那么它就不再需要了，我不想在一个可能耗时的循环中处理该实例。

我尝试使用 sys.getrefcount 删除仅在列表中引用的对象。

for i in A.instances:
    if sys.getrefcount(i) <=3: # in the list, in the loop and in getrefcount
        # collect and remove after the loop

我遇到的问题是引用计数非常模糊。 打开一个新的 shell 并创建一个没有内容的虚拟类返回 5 for

sys.getrefcount(DummyClass)

另一个想法是复制对象，然后删除列表并检查哪些对象已安排进行垃圾收集，并在最后一步删除这些对象。像这样的东西:

Copy = copy(A.instances)
del A.instances
A.instances = [i for i in Copy if not copy_of_i_is_in_GC(i)]

当引用计数变为 0 时，不必立即删除对象。我只是不想在不再使用的对象上浪费太多资源。

Answer 1

这个答案与 Kevin 的答案相同，但我正在编写一个带有弱引用的示例实现，并将其发布在这里。使用弱引用解决了对象被 self.instance 列表引用的问题，因此它永远不会被删除。

为对象创建弱引用的其中一件事是您可以在删除对象时包含回调。存在诸如程序退出时未发生回调等问题...但这可能正是您想要的。

import threading
import weakref

class A(object):
    instances = []
    lock = threading.RLock()

    @classmethod
    def _cleanup_ref(cls, ref):
        print('cleanup') # debug
        with cls.lock:
            try:
                cls.instances.remove(ref)
            except ValueError:
                pass

    def __init__(self):
        with self.lock:
            self.instances.append(weakref.ref(self, self._cleanup_ref))

# test
test = [A() for _ in range(3)]
for i in range(3,-1,-1):
    assert len(A.instances) == i
    if test:
        test.pop()

print("see if 3 are removed at exit")
test = [A() for _ in range(3)]