本文字数:2075 字
阅读本文大概需要:6 分钟
写在之前
我们在昨天的文章(Python 黑魔法之内存优化)讲了一种黑魔法,今天我们来讲另一种:「属性拦截」。当我们访问某个类或者是实例属性的时候,如果它不存在的话,就会出现异常。对于异常,我们总是要处理的,接下来就让我们来看,是怎么处理的。
属性拦截
在 Python 中,就有一些具有拦截功能的方法,让我们先来看一下:
1.__setattr__(self,name,value):如果想要给 name 赋值的话,就需要调用这个方法。
2.__getattr__(self,name):如果 name 被访问且它又不存在,那么这个方法将被调用。
3.__delattr__(self,name):如果要删除 name 的话,这个方法就要被调用了。
下面我们用例子来演示一下:
>>> classSample:... def__getattr__(self,name):... print('hello getattr')... def__setattr__(self,name,value):... print('hello setattr')... self.__dict__[name] = value...
上面的例子中类 Sample 只有两个方法,下面让我们实例化一下:
>>> s = Sample()>>> s.xhello getattr
s.x 这个实例属性本来是不存在的,但是由于类中有了 __getattr__(self,name) 方法,当发现属性 x 不存在于对象的 __dict__ 中时,就调用了 __getattr__,也就是所谓的「拦截成员」。
>>> s.x = 7hello setattr
同理,给对象的属性赋值的时候,调用了 __setattr__(self,name,value) 方法,这个方法中有 self.__dict__[name] = value,通过这个就将属性和数据存到了对象 __dict__ 中。如果再调用这个属性的话,会成为下面这样:
>>> s.x7
出现这种结果的原因是它已经存在于对象的 __dict__ 中了。
看了上面的两个,你是不是觉得上面的方法很有魔力呢?这就是「黑魔法」,但是它具体有什么应用呢?我们接着来看:
classRectangle:""" the width and length of Rectangle """def__init__(self): self.width = 0 self.length = 0defsetSize(self,size): self.width, self.length = sizedefgetSize(self):return self.width, self.lengthif __name__ == "__main__": r = Rectangle() r.width = 3 r.length = 4 print(r.getSize()) print(r.setSize((30,40))) print(r.width) print(r.length)
上面是我根据一个很有名的例子改编的,你可以先想一下结果,想完以后可以接着往下看:
(3, 4)3040
这段代码运行的结果如上面所示,作为一个强迫证的码农,对于这种可以改进的程序当然不能容忍。我们在上面介绍的特殊方法,我们一定要学以致用,虽然重新写的不一定比原来的好,但我们还是要尝试去用:
classNewRectangle:""" the width and length of Rectangle """def__init__(self): self.width = 0 self.length = 0def__setattr__(self, name, value):if name == 'size': self.width, self.length = valueelse: self.__dict__[name] = valuedef__getattr__(self, name):if name == 'size':return self.width, self.lengthelse:return AttributeErrorif __name__ == "__main__": r = NewRectangle() r.width = 3 r.length = 4 print(r.size) r.size = 30,40 print(r.width) print(r.length)
我们来看一下运行的结果:
(3, 4)3040
我们可以看到,除了类的写法变了以外,调用的方式没有变,结果也没有变。
写在之后
黑魔法我暂时先说这两个,其实还有别的,如果你感兴趣的话,可以去 Google 一些文章来看,相信自己的探索会让自己收获更多。
至此,不知道你发现了没有,我们使用了很多以双下划线开头和结尾的属性和方法,比如 __init__ 等。在 Python 中,以双下划线开头和结尾的都是「特殊」的,当然这个只是一个惯例,目的是为了确保这些特殊的名字不会和你定义的名字发生冲突。如果你看着它们不顺眼,想要把它们重写当然也是可以的。
如果你觉得本篇文章对你有所帮助的话,欢迎点赞转发,让更多的人看到,期待和你的交流。
The end。