Проверить атрибуты класса Python

Ниже приведен трудный путь. Вот простой способ. Не знаю, почему это не пришло мне в голову раньше.

import inspect

def get_user_attributes(cls):
    boring = dir(type('dummy', (object,), {}))
    return [item
            for item in inspect.getmembers(cls)
            if item[0] not in boring]

Вот начало

def get_user_attributes(cls):
    boring = dir(type('dummy', (object,), {}))
    attrs = {}
    bases = reversed(inspect.getmro(cls))   
    for base in bases:
        if hasattr(base, '__dict__'):
            attrs.update(base.__dict__)
        elif hasattr(base, '__slots__'):
            if hasattr(base, base.__slots__[0]): 
                # We're dealing with a non-string sequence or one char string
                for item in base.__slots__:
                    attrs[item] = getattr(base, item)
            else: 
                # We're dealing with a single identifier as a string
                attrs[base.__slots__] = getattr(base, base.__slots__)
    for key in boring:
        del attrs['key']  # we can be sure it will be present so no need to guard this
    return attrs

Это должно быть довольно надежно. По сути, он работает, заставляя игнорировать атрибуты подкласса по умолчанию object. Затем он получает MRO класса, который ему передан, и просматривает его в обратном порядке, чтобы ключи подкласса могли перезаписывать ключи суперкласса. Он возвращает словарь пар ключ-значение. Если вам нужен список кортежей ключей и значений, как в inspect.getmembers, тогда просто верните либо attrs.items(), либо list(attrs.items()) в Python 3.

Если вы на самом деле не хотите проходить mro и просто хотите, чтобы атрибуты определялись непосредственно в подклассе, тогда это проще:

def get_user_attributes(cls):
    boring = dir(type('dummy', (object,), {}))
    if hasattr(cls, '__dict__'):
        attrs = cls.__dict__.copy()
    elif hasattr(cls, '__slots__'):
        if hasattr(base, base.__slots__[0]): 
            # We're dealing with a non-string sequence or one char string
            for item in base.__slots__:
                attrs[item] = getattr(base, item)
            else: 
                # We're dealing with a single identifier as a string
                attrs[base.__slots__] = getattr(base, base.__slots__)
    for key in boring:
        del attrs['key']  # we can be sure it will be present so no need to guard this
    return attrs

Двойное подчеркивание на обоих концах «специальных атрибутов» было частью Python до 2.0. Маловероятно, что они изменят это когда-либо в ближайшем будущем.

class Foo(object):
  a = 1
  b = 2

def get_attrs(klass):
  return [k for k in klass.__dict__.keys()
            if not k.startswith('__')
            and not k.endswith('__')]

print get_attrs(Foo)

['a', 'b']

Спасибо, aaronasterling, вы дали мне нужное выражение :-) Моя последняя функция инспектора атрибутов класса выглядит так:

def get_user_attributes(cls,exclude_methods=True):
  base_attrs = dir(type('dummy', (object,), {}))
  this_cls_attrs = dir(cls)
  res = []
  for attr in this_cls_attrs:
    if base_attrs.count(attr) or (callable(getattr(cls,attr)) and exclude_methods):
      continue
    res += [attr]
  return res

Либо верните только переменные атрибутов класса (exclude_methods = True), либо также получите методы. Мои начальные тесты вышеупомянутой функции поддерживают классы Python как в старом, так и в новом стиле.

/ Якоб

Если вы используете новые классы стилей, не могли бы вы просто вычесть атрибуты родительского класса?

class A(object):
    a = 10
    b = 20
    #...

def get_attrs(Foo):
    return [k for k in dir(Foo) if k not in dir(super(Foo))]

Изменить: Не совсем. _2 _, _ 3_ и __weakref__ появляются при наследовании от объекта, но отсутствуют в самом объекте. Вы можете выделить их в особом случае - я сомневаюсь, что они будут меняться очень часто.

Прошу прощения за некроз нити. Я удивлен, что по состоянию на 2019 год до сих пор нет простой функции (или библиотеки) для обработки такого распространенного использования.

Хочу поблагодарить aaronasterling за идею. На самом деле контейнер set предоставляет более простой способ выразить это:

class dummy:    pass

def abridged_set_of_user_attributes(obj):
    return set(dir(obj))-set(dir(dummy))

def abridged_list_of_user_attributes(obj):
    return list(abridged_set_of_user_attributes(obj))

Исходное решение, использующее понимание списка, на самом деле представляет собой двухуровневый цикл, потому что в нем объединены два in ключевых слова, несмотря на то, что наличие только одного ключевого слова for делало его менее трудоемким, чем есть на самом деле.

Это помогло мне включить определенные пользователем атрибуты с __, которые могут быть найдены в cls.__dict__

import inspect

class A:
    __a = True
    
    def __init__(self, _a, b, c):
        self._a = _a
        self.b = b
        self.c = c 

    def test(self):
        return False

cls = A(1, 2, 3)

members = inspect.getmembers(cls, predicate=lambda x: not inspect.ismethod(x))
attrs = set(dict(members).keys()).intersection(set(cls.__dict__.keys()))
__attrs = {m[0] for m in members if m[0].startswith(f'_{cls.__class__.__name__}')}
attrs.update(__attrs)

Это правильно даст: {'_A__a', '_a', 'b', 'c'}

Вы можете обновить, чтобы очистить cls.__class__.__name__, если хотите