替換一個(gè)實(shí)例方法,沒(méi)你想的那么簡(jiǎn)單
思路一:簡(jiǎn)單地替換
當(dāng)你想對(duì)類實(shí)例的方法進(jìn)行替換時(shí),你可能想到的是直接對(duì)他進(jìn)行粗暴地替換:
- class People:
- def speak(self):
- print("hello, world")
- def speak(self):
- print("hello, python")
- p = People()
- p.speak = speak
- p.speak()
但當(dāng)你試著執(zhí)行這段代碼的時(shí)候,就會(huì)發(fā)現(xiàn)行不通,它提示我們要傳入 self 參數(shù):
- Traceback (most recent call last):
- File "/Users/MING/Code/Python/demo.py", line 12, in <module>
- p.speak()
- TypeError: speak() missing 1 required positional argument: 'self'
不對(duì)啊~ self 不是實(shí)例本身嗎?函數(shù)不是一直就這么寫(xiě)的?
實(shí)際上你這么替換,speak 就變成了一個(gè) function,而不是一個(gè)和實(shí)例綁定的 method ,你可以把替換前后的 speak 打印出來(lái)
- p = People()
- print(p.speak)
- p.speak = speak
- print(p.speak)
輸出結(jié)果如下,區(qū)別非常明顯
- <bound method People.speak of <__main__.People object at 0x10cfa7fd0>>
- <function speak at 0x10ca10040>
這種方法,只能用在替換不與實(shí)例綁定的靜態(tài)方法上,不然你每次調(diào)用的時(shí)候,就得手動(dòng)傳入實(shí)例本身,但這樣調(diào)用就會(huì)變得非常怪異。
思路二:利用 im_func
有 Python 2 使用經(jīng)驗(yàn)的朋友,可以會(huì)知道類實(shí)例的方法,都有 im_func 和 im_class 屬性,分別指向了該方法的函數(shù)和類。
很抱歉的是,這些在 Python3 中全都取消了,意味你無(wú)法再使用 im_func 和 im_class 。
但即使你身處 Python 2 的環(huán)境下,你想通過(guò) im_func 去直接替換函數(shù),也仍然是有問(wèn)題的。
因?yàn)樵?Python2 中不推薦普通用戶對(duì)類實(shí)例的方法進(jìn)行替換,所以 Python 給類實(shí)例的方法賦予了只讀屬性
思路三:非常危險(xiǎn)的字節(jié)碼替換
表層不行,但這個(gè)方法在字節(jié)碼層面卻是可行的
這種方法,非常的粗暴且危險(xiǎn),他會(huì)直接影響到使用 People 的所有實(shí)例的 speak 方法,因此這種方法千萬(wàn)不要使用。
思路四:利用 types 綁定方法
在 types 中有一個(gè) MethodType,可以將普通方法與實(shí)例進(jìn)行綁定。
綁定后,就可以直接替換掉原實(shí)例的 speak 方法了,完整代碼如下:
- import types
- class People:
- def speak(self):
- print("hello, world")
- def speak(self):
- print("hello, python")
- p = People()
- p.speak = types.MethodType(speak, p)
- p.speak()
這種方法,最為安全,不會(huì)影響其他實(shí)例。并且 Python 2 和 Python 3 都適用,是官方推薦的一種做法。
總結(jié)一下
- 直接替換:只適用于靜態(tài)方法
- 使用 im_func 替換:行不通
- 使用 im_func.func_code 替換字節(jié)碼:非常危險(xiǎn),請(qǐng)不要使用
- 使用 types.MethodType 進(jìn)行方法綁定:安全且有效,推薦使用