在字典理解中使用__slots__
属性plusgetattr()
:
{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'