Поиск статических атрибутов класса в Python

Question

Это необычный вопрос, но я хотел бы динамически генерировать атрибут __slots__ класса на основе любых атрибутов, которые я случайно добавил в класс.

Например, если у меня есть класс:

class A(object): 
    one = 1 
    two = 2 

    __slots__ = ['one', 'two'] 

я хотел бы сделать это динамически, а не задавая параметры вручную, как бы я это сделать?

Answer 1

В момент, когда вы пытаетесь определить слоты, класс еще не построен, поэтому вы не можете определить его динамически из класса A.

Чтобы получить нужное поведение, используйте метакласс, чтобы исследовать определение A и добавить атрибут слотов.

class MakeSlots(type): 

    def __new__(cls, name, bases, attrs): 
     attrs['__slots__'] = attrs.keys() 

     return super(MakeSlots, cls).__new__(cls, name, bases, attrs) 

class A(object): 
    one = 1 
    two = 2 

    __metaclass__ = MakeSlots 

Answer 2

Одна очень важная вещь, чтобы быть в курсе - если эти атрибуты остаются в классе, __slots__ поколение будет бесполезно ... ну, может быть, не бесполезно - это сделает атрибуты класса чтения -только; вероятно, не то, что вы хотите.

Легкий способ сказать: «Хорошо, я инициализирую их никому, а затем пусть они исчезнут». Отлично! Вот один из способов сделать это:

class B(object): 
    three = None 
    four = None 

    temp = vars()     # get the local namespace as a dict() 
    __slots__ = temp.keys()   # put their names into __slots__ 
    __slots__.remove('temp')   # remove non-__slots__ names 
    __slots__.remove('__module__') # now remove the names from the local 
    for name in __slots__:   # namespace so we don't get read-only 
     del temp[name]    # class attributes 
    del temp       # and get rid of temp 

Если вы хотите сохранить эти начальные значения она занимает немного больше работы ... вот один из возможных решений:

class B(object): 
    three = 3 
    four = 4 

    def __init__(self): 
     for key, value in self.__init__.defaults.items(): 
      setattr(self, key, value) 

    temp = vars() 
    __slots__ = temp.keys() 
    __slots__.remove('temp') 
    __slots__.remove('__module__') 
    __slots__.remove('__init__') 
    __init__.defaults = dict() 
    for name in __slots__: 
     __init__.defaults[name] = temp[name] 
     del temp[name] 
    del temp 

Как вы можете видеть, это возможно сделать это без метакласса - но кто хочет все, что шаблон? Метакласс определенно может помочь нам очистить это:

class MakeSlots(type): 
    def __new__(cls, name, bases, attrs): 
     new_attrs = {} 
     new_attrs['__slots__'] = slots = attrs.keys() 
     slots.remove('__module__') 
     slots.remove('__metaclass__') 
     new_attrs['__weakref__'] = None 
     new_attrs['__init__'] = init = new_init 
     init.defaults = dict() 
     for name in slots: 
      init.defaults[name] = attrs[name] 
     return super(MakeSlots, cls).__new__(cls, name, bases, new_attrs) 

def new_init(self): 
    for key, value in self.__init__.defaults.items(): 
     setattr(self, key, value) 

class A(object): 
    __metaclass__ = MakeSlots 

    one = 1 
    two = 2 


class B(object): 
    __metaclass__ = MakeSlots 

    three = 3 
    four = 4 

Теперь все занудство хранится в метаклассе, и фактический класс легко читать и (надеюсь!) Понять.

Если у вас есть что-то еще в этих классах помимо атрибутов, я настоятельно рекомендую вам поставить все, что есть в классе mixin - их непосредственное включение в конечный класс еще более усложнит метаклассу.