Question

class A(object):
    _all = set()

    def __new__(cls):
        obj = super(A,cls).__new__(cls)
        cls._all.add(obj)
        return obj

class B(A):
    pass

class C(A):
    pass

b = B()
c = C()

这将创建所有实例并将其放入 A._all 。我想要的是将 C 实例放入 C._all (和C的类似)。

我可以这样定义 >B 和 C :

class B(A):
    _all = set()

class C(A):
    _all = set()

哪个会有效。但是, 这感觉就像重复代码, 如果我忘了在下面的子类 < code> A 或 misspell < alll 我可能很难调试错误时添加 < code> all 。

我该如何做好呢?

Answer 1

此处为它提供一个元级 :

class make_all_meta(type):
   def __init__(cls, name, bases, dic):
       type.__init__(cls, name, bases, dic)
       cls._all = set()

class A(object):
    __metaclass__ = make_all_meta

在 Python 3 中, < code> A 的定义必须是:

class A(metaclass=make_all_meta):
    pass

Answer 2

设置 all 到 noone 并创建您在中的集, 如果仍然 noone , 给每个类别它自己的副本 :

class A(object):
    _all = None

    def __new__(cls):
        obj = super(A,cls).__new__(cls)
        if cls._all is None:
            cls._all = set()
        cls._all.add(cls)
        return obj

友情链接