Uzyskaj wszystkie __slots__ z klasy pochodnej

Question

Muszę zainicjować wszystkie gniazda instancji za pomocą Brak. Jak uzyskać wszystkie gniazda klasy pochodnej?

Przykład (który nie działa):

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" 

mogę użyć dodatkowego atrybutu class, która posiada otwory (w tym dziedziczone) dla wszystkich klas, takich jak

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"] 

ale wydaje nieoptymalny.

Wszelkie uwagi są mile widziane!

Cheers,

Jan

Answer 1

Przede wszystkim, to

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

Sporządzenie listy, która zawiera wszystkie elementy zawarte przez __slots__ B lub któregokolwiek z jej klas nadrzędnych byłoby:

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

Answer 2

Chcesz powtórzyć każdą klasę w MRO:

class A(object): 
    __slots__ = ('x', 'y') 
    def __init__(self): 
     for slots in [getattr(cls, '__slots__', []) for cls in type(self).__mro__]: 
      for attr in slots: 
       setattr(self, attr, None) 

Widać, że to działa zgodnie z oczekiwaniami w klasie pochodnej:

class B(A): 
    __slots__ = ('z',) 

>>> b = B() 

>>> b.x, b.y, b.z 
<<< (None, None, None)