以下实例均为 Python 3.5 版本,但同样适用于 Python 2.x 和 Python 3.x 版本。

重写类中的 __getattr__ 魔术方法是 Python 中实现动态属性的很普通的方法。试想有这样一个数据词典 AttrDict,它允许类似属性的方式访问其存储的键值对:

class AttrDict(dict):

def __getattr__(self, item):

return self[item]

这个简化的 AttrDict 类允许以类似属性的方式获取数据字典的值,同时,它允许一种非常简单的方式来设置键值对的值。无论哪种情况下,它都可以这样做:

>>> attrd = AttrDict()

... attrd["key"] = "value"

... print(attrd.key)

value

重写 __getattr__ 方法(和 __setattr__ 方法)非常有用——这能够让你的构建远程过程调用(RPCs)单元的代码更具可读性。然而,动态属性也有令人沮丧的地方——它们在使用之前是不可见的!

动态属性在交互的 shell 下有两大使用问题。第一个问题就是当用户使用 dir 方法去检查对象的 API 的时候,它们并不会出现:

>>> dir(attrd) # 我想知道如何使用 attrd

['__class__', '__contains__', ... 'keys', 'values']

>>> # 没有动态属性 [傲娇脸]

第二个问题是自动完成——如果我们照常设置 normal_attribute 属性,大部分主流的 shell 环境 1 下都能自动完成。

但是以字典键值对的方式设置 dynamic_attribute 的时候并没有自动完成功能:

然而,你可以在实现动态属性的时候锦上添花地实现 __dir__ 方法,这样不但能提高用户体验还能一石二鸟地解决以上两个问题。详见 文章:

如果对象内有名为 __dir__()的方法,方法被调用时必须返回属性列表。这使得对象可以实现一个自定义的__getattr__() 或者 __getattribute__() 方法,来自定义 dir() 方法输出属性的方式。

实现 __dir__ 方法非常简单,只需返回对象的属性键名列表:

class AttrDict(dict):

def __getattr__(self, item):

return self[item]

def __dir__(self):

return super().__dir__() + [str(k) for k in self.keys()]

这样 dir(attrd) 将会返回动态属性和普通属性。有趣的是,shell 环境将会使用 __dir__ 方法来进行自动完成提示!因此我们毫不费力就实现了自动完成 2功能:

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐