Obtener todos los __slots__ de la clase derivada

Question

Necesito inicializar todas las ranuras de una instancia con None. ¿Cómo obtengo todos los espacios de una clase derivada?

Ejemplo (que no funciona):

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" 

que podría utilizar un atributo clase adicional que contiene las ranuras (incluyendo el heredada) para todas las clases, como

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

pero que parece subóptimo.

¡Cualquier comentario es apreciado!

Cheers,

Ene

Answer 1

En primer lugar, es

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

Hacer una lista que contiene todos los elementos contenidos por __slots__ de B o cualquiera de sus clases padre sería:

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

Answer 2

Quiere iterar a través de cada clase en el 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) 

Se puede ver que esto funciona como se esperaba en la clase derivada:

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

>>> b = B() 

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