Python隨手筆記(九)--------面向對象高級編程(2)

使用@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
測試通過!
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章