Python словарь из полей объекта - PullRequest
283 голосов
/ 14 сентября 2008

Знаете ли вы, есть ли встроенная функция для построения словаря из произвольного объекта? Я хотел бы сделать что-то вроде этого:

>>> class Foo:
...     bar = 'hello'
...     baz = 'world'
...
>>> f = Foo()
>>> props(f)
{ 'bar' : 'hello', 'baz' : 'world' }

ПРИМЕЧАНИЕ: Не должно включать методы. Только поля.

Ответы [ 11 ]

353 голосов
/ 15 сентября 2008

Обратите внимание, что в Python 2.7 рекомендуется использовать классы в новом стиле (не требуется для Python 3), т.е.

class Foo(object):
   ...

Кроме того, существует разница между «объектом» и «классом». Чтобы построить словарь из произвольного объекта , достаточно использовать __dict__. Обычно вы объявляете свои методы на уровне класса и свои атрибуты на уровне экземпляра, поэтому __dict__ должно быть в порядке. Например:

>>> class A(object):
...   def __init__(self):
...     self.b = 1
...     self.c = 2
...   def do_nothing(self):
...     pass
...
>>> a = A()
>>> a.__dict__
{'c': 2, 'b': 1}

Лучшим подходом (предложенным Роберт в комментариях) является встроенная vars функция:

>>> vars(a)
{'c': 2, 'b': 1}

В качестве альтернативы, в зависимости от того, что вы хотите сделать, было бы неплохо унаследовать от dict. Тогда ваш класс уже словарь, и если вы хотите, вы можете переопределить getattr и / или setattr для вызова и установить диктовку. Например:

class Foo(dict):
    def __init__(self):
        pass
    def __getattr__(self, attr):
        return self[attr]

    # etc...
108 голосов
/ 02 августа 2015

Вместо x.__dict__ на самом деле более питонно использовать vars(x).

55 голосов
/ 14 сентября 2008

Встроенный dir предоставит вам все атрибуты объекта, включая специальные методы, такие как __str__, __dict__ и целый ряд других, которые вы, вероятно, не хотите. Но вы можете сделать что-то вроде:

>>> class Foo(object):
...     bar = 'hello'
...     baz = 'world'
...
>>> f = Foo()
>>> [name for name in dir(f) if not name.startswith('__')]
[ 'bar', 'baz' ]
>>> dict((name, getattr(f, name)) for name in dir(f) if not name.startswith('__')) 
{ 'bar': 'hello', 'baz': 'world' }

Таким образом, можно расширить это, чтобы возвращать только атрибуты данных, а не методы, определив вашу функцию props следующим образом:

import inspect

def props(obj):
    pr = {}
    for name in dir(obj):
        value = getattr(obj, name)
        if not name.startswith('__') and not inspect.ismethod(value):
            pr[name] = value
    return pr
24 голосов
/ 14 сентября 2008

Я согласился с комбинацией обоих ответов:

dict((key, value) for key, value in f.__dict__.iteritems() 
    if not callable(value) and not key.startswith('__'))
15 голосов
/ 15 сентября 2008

Чтобы построить словарь из произвольного объекта , достаточно использовать __dict__.

Это пропускает атрибуты, которые объект наследует от своего класса. Например,

class c(object):
    x = 3
a = c()

hasattr (a, 'x') имеет значение true, но 'x' не появляется в .__ dict __

13 голосов
/ 29 марта 2015

Я подумал, что мне понадобится время, чтобы показать вам, как вы можете перевести объект в диктовку через dict(obj).

class A(object):
    d = '4'
    e = '5'
    f = '6'

    def __init__(self):
        self.a = '1'
        self.b = '2'
        self.c = '3'

    def __iter__(self):
        # first start by grabbing the Class items
        iters = dict((x,y) for x,y in A.__dict__.items() if x[:2] != '__')

        # then update the class items with the instance items
        iters.update(self.__dict__)

        # now 'yield' through the items
        for x,y in iters.items():
            yield x,y

a = A()
print(dict(a)) 
# prints "{'a': '1', 'c': '3', 'b': '2', 'e': '5', 'd': '4', 'f': '6'}"

Ключевым разделом этого кода является функция __iter__.

Как поясняется в комментариях, в первую очередь мы берем предметы класса и предотвращаем все, что начинается с '__'.

Как только вы создали это dict, вы можете использовать диктовку update и передать экземпляр __dict__.

Это даст вам полный словарь класса + экземпляр класса. Теперь осталось только перебрать их и получить возврат.

Кроме того, если вы планируете много использовать это, вы можете создать декоратор класса @iterable.

def iterable(cls):
    def iterfn(self):
        iters = dict((x,y) for x,y in cls.__dict__.items() if x[:2] != '__')
        iters.update(self.__dict__)

        for x,y in iters.items():
            yield x,y

    cls.__iter__ = iterfn
    return cls

@iterable
class B(object):
    d = 'd'
    e = 'e'
    f = 'f'

    def __init__(self):
        self.a = 'a'
        self.b = 'b'
        self.c = 'c'

b = B()
print(dict(b))
7 голосов
/ 04 июля 2013

Поздний ответ, но с учетом полноты и пользы googlers:

def props(x):
    return dict((key, getattr(x, key)) for key in dir(x) if key not in dir(x.__class__))

Здесь не будут отображаться методы, определенные в классе, но все равно будут отображаться поля, в том числе те, которые назначены лямбда-выражениям, или те, которые начинаются с двойного подчеркивания.

4 голосов
/ 29 мая 2014

Я думаю, что самый простой способ - создать атрибут getitem для класса. Если вам нужно записать объект, вы можете создать пользовательский setattr . Вот пример для getitem :

class A(object):
    def __init__(self):
        self.b = 1
        self.c = 2
    def __getitem__(self, item):
        return self.__dict__[item]

# Usage: 
a = A()
a.__getitem__('b')  # Outputs 1
a.__dict__  # Outputs {'c': 2, 'b': 1}
vars(a)  # Outputs {'c': 2, 'b': 1}

dict генерирует атрибуты объектов в словарь, и объект словаря можно использовать для получения необходимого элемента.

3 голосов
/ 18 декабря 2018

Недостатком использования __dict__ является то, что он неглубокий; он не преобразует подклассы в словари.

Если вы используете Python3.5 или выше, вы можете использовать jsons:

>>> import jsons
>>> jsons.dump(f)
{'bar': 'hello', 'baz': 'world'}
3 голосов
/ 07 января 2016

Если вы хотите перечислить часть своих атрибутов, переопределите __dict__:

def __dict__(self):
    d = {
    'attr_1' : self.attr_1,
    ...
    }
    return d

# Call __dict__
d = instance.__dict__()

Это очень помогает, если ваш instance получает большие данные блока и вы хотите отправить d в очередь сообщений, как Redis

...