使用@property
在綁定屬性時,如果我們直接把屬性暴露出去,雖然寫起來很簡單,但是,沒辦法檢查參數,導致可以把成績隨便改:
>>> an = Animal()
>>> an.age = 2000
這顯然不合邏輯。爲了限制age的範圍,可以通過一個set_age()方法來設置成績,再通過一個get_age()來獲取成績,這樣,在set_age()方法裏,就可以檢查參數:
>>> class Animal(object):
def __init__(self, age):
self.age = age
def get_age(self):
return self.age
def set_age(self, value):
if not isinstance(value, int):
raise ValueError('age must be an integer!')
if value < 0 or value > 100:
raise ValueError('age must between 0 ~ 100!')
self.age = value
現在,對任意的Student實例進行操作,就不能隨心所欲地設置score了:
>>> an = Animal(10)
>>> an.set_age(60)
>>> an.get_age()
60
>>> an.set_age(1000)
Traceback (most recent call last):
File "<pyshell#22>", line 1, in <module>
an.set_age(1000)
File "<pyshell#15>", line 12, in set_age
raise ValueError('age must between 0 ~ 100!')
ValueError: age must between 0 ~ 100!
>>> an.set_age('25')
Traceback (most recent call last):
File "<pyshell#23>", line 1, in <module>
an.set_age('25')
File "<pyshell#15>", line 10, in set_age
raise ValueError('age must be an integer!')
ValueError: age must be an integer!
但是,上面的調用方法又略顯複雜,沒有直接用屬性這麼直接簡單。
有沒有既能檢查參數,又可以用類似屬性這樣簡單的方式來訪問類的變量呢?
還記得裝飾器(decorator)可以給函數動態加上功能嗎?對於類的方法,裝飾器一樣起作用。Python內置的@property裝飾器就是負責把一個方法變成屬性調用的:
class Animal(object):
@property
def age(self):
return self._age
@age.setter
def age(self, value):
if not isinstance(value, int):
raise ValueError('age must be an integer!')
if value < 0 or value > 100:
raise ValueError('age must between 0 ~ 100!')
self.age = value
@property的實現比較複雜,把一個getter方法變成屬性,只需要加上@property就可以了,此時,@property本身又創建了另一個裝飾器@score.setter,負責把一個setter方法變成屬性賦值,於是,我們就擁有一個可控的屬性操作:
>>> an = Animal()
>>> an.age = 1000
Traceback (most recent call last):
File "<pyshell#37>", line 1, in <module>
an.age = 1000
File "<pyshell#35>", line 11, in age
raise ValueError('age must between 0 ~ 100!')
ValueError: age must between 0 ~ 100!
注意到這個神奇的@property,我們在對實例屬性操作的時候,就知道該屬性很可能不是直接暴露的,而是通過getter和setter方法來實現的。
還可以定義只讀屬性,只定義getter方法,不定義setter方法就是一個只讀屬性:
class Animal(object):
@property
def birth(self):
return self._birth
@birth.setter
def birth(self,value):
self._birth = value
練習:請利用@property給一個Screen對象加上width和height屬性,以及一個只讀屬性resolution:
class Screen(object):
class Screen(object):
@property
def width(self):
return self._width
@width.setter
def width(self, value):
self._width = value
@property
def height(self):
return self._height
@height.setter
def height(self, value2):
self._height = value2
@property
def resolution(self):
return self._width * self._height
s = Screen()
s.width = 1024
s.height = 768
print('resolution =', s.resolution)
if s.resolution == 786432:
print('測試通過!')
else:
print('測試失敗!')
結果輸出:
resolution = 786432
測試通過!