今天介绍一下python的装饰器,先看一个例子
import traceback
class Screen(object):
@property
def width(self):
return self._width
@width.setter
def width(self, width):
if isinstance(width, str):
raise ValueError('width can not be string!')
self._width = width
@property
def height(self):
return self._height
@width.setter
def height(self, height):
if isinstance(height, str):
raise ValueError('height can not be string!')
self._height = height
@property
def resolution(self):
self._resolution = self._width * self._height
return self._resolution
s = Screen()
s.width = 75
s.height = 20
print(s.resolution)
>>>1500
再看下面:
try:
s.width = 'abc'
except ValueError:
traceback.print_exc()
try:
s.resolution = 1500
except AttributeError:
traceback.print_exc()
>>>Traceback (most recent call last):
File "<ipython-input-17-9c5a1f270fcd>", line 2, in <module>
s.width = 'abc'
File "<ipython-input-16-8690b78dfded>", line 11, in width
raise ValueError('width can not be string!')
ValueError: width can not be string!
>>>Traceback (most recent call last):
File "<ipython-input-17-9c5a1f270fcd>", line 6, in <module>
s.resolution = 1500
AttributeError: can't set attribute
在绑定属性时,如果我们直接把属性暴露出去,虽然写起来很简单,但是,没办法检查参数,导致可以随便改:
有没有既能检查参数,又可以用类似属性这样简单的方式来访问类的变量呢?
装饰器(decorator)可以给函数动态加上功能,对于类的方法,装饰器一样起作用。Python内置的@property装饰器就是负责把一个方法变成属性调用的:
注意到这个神奇的@property,我们在对实例属性操作的时候,就知道该属性很可能不是直接暴露的,而是通过getter和setter方法来实现的。
@property的实现比较复杂,我们先考察如何使用。把一个getter方法变成属性,只需要加上@property就可以了,此时,@property本身又创建了另一个装饰器@width.setter负责把一个setter方法变成属性赋值,于是,我们就拥有一个可控的属性操作:
还可以定义只读属性,只定义getter方法,不定义setter方法就是一个只读属性
上例中我们利用@property给一个Screen对象加上width和height属性,以及一个只读属性resolution
@property广泛应用在类的定义中,可以让调用者写出简短的代码,同时保证对参数进行必要的检查,这样,程序运行时就减少了出错的可能性。