python - 获取派生类的所有 __slots__

Question

我需要用 None 初始化一个实例的所有插槽。如何获取派生类的所有插槽？

示例(不起作用):

class A(object):
    __slots__ = "a"

    def __init__(self):
        # this does not work for inherited classes
        for slot in type(self).__slots__:
            setattr(self, slot, None)

class B(A):
    __slots__ = "b"

我可以使用一个额外的类属性来保存所有类的插槽(包括继承的)，比如

class A(object):
    __slots__ = "a"
    all_slots = "a"

    def __init__(self):
        # this does not work for inherited classes
        for slot in type(self).all_slots:
            setattr(self, slot, None)

class B(A):
    __slots__ = "b"
    all_slots = ["a", "b"]

但这似乎不是最理想的。

如有任何意见，我们将不胜感激!

干杯，

一月

Answer 1

首先是

class A(object):
    __slots__ = ('a',)
class B(A):
    __slots__ =  ('b',)

制作一个包含 B 或其任何父类的 __slots__ 所包含的所有元素的列表将是:

from itertools import chain
slots = chain.from_iterable(getattr(cls, '__slots__', []) for cls in B.__mro__)