如何将带有插槽的python类转换为字典?


问题内容

我正在使用带有插槽的类来减少实例将占用的内存。现在,如何将插槽实例转换为字典?

插槽类如下所示:

class Foo(object):
       __slots__ = ['x','y','z']
       def __init__(self):
           self.x = 1
           self.y = 2 
           self.z = 3

我期望这样的事情:

y = Foo()
y.__dict__
{'x': 1, 'y': 2, 'z': 3}

问题答案:

在字典理解中使用__slots__属性plus
getattr()

{s: getattr(obj, s) for s in obj.__slots__ if hasattr(obj, s)}

跳过所有未设置的属性。

或者,将缺少的属性设置为None

{s: getattr(obj, s, None) for s in obj.__slots__}

演示:

>>> class Foo(object):
...     __slots__ = ('bar', 'spam')
... 
>>> obj = Foo()
>>> obj.bar = 42
>>> {s: getattr(obj, s) for s in obj.__slots__ if hasattr(obj, s)}
{'bar': 42}
>>> {s: getattr(obj, s, None) for s in obj.__slots__}
{'spam': None, 'bar': 42}

您甚至可以将其设为该类的属性,vars()并将使用它:

>>> class Foo(object):
...     __slots__ = ('bar', 'spam')
...     @property
...     def __dict__(self):
...         return {s: getattr(self, s) for s in self.__slots__ if hasattr(self, s)}
... 
>>> f = Foo()
>>> f.bar = 42
>>> f.__dict__
{'bar': 42}
>>> f.spam = 'eggs'
>>> f.__dict__
{'spam': 'eggs', 'bar': 42}
>>> vars(f)
{'spam': 'eggs', 'bar': 42}
>>> f.hello = 'world'
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: 'Foo' object has no attribute 'hello'