一日一技:8行炫技代碼,知識(shí)點(diǎn)多得不得了
我們今天來看一段炫技代碼。它可以把任何能接收兩個(gè)參數(shù)的函數(shù)定義成一個(gè)特殊的運(yùn)算符。
例如,我們知道Python里面的range函數(shù),最少可以接收1個(gè)參數(shù),最多能夠接收3個(gè)參數(shù)。當(dāng)只有兩個(gè)參數(shù)的時(shí)候,格式為range(開始, 結(jié)束),從開始數(shù)字到結(jié)束數(shù)字逐次加1,左閉右開。使用今天的方法,可以把它的寫法改為開始 |到| 結(jié)束,如下圖所示:
又比如,urllib.parse.urljoin可以把域名和一個(gè)相對(duì)路徑拼接起來: urljoin(域名, 相對(duì)路徑),我們也可以改寫成如下圖所示的格式:
這種炫技有余,實(shí)用不足的功能是怎么實(shí)現(xiàn)的呢?其實(shí)原理非常簡單,只有8行代碼:
class Test:
def __init__(self, num):
self.num = num
def __or__(self, other):
print(f'我右邊有一個(gè)東西,它是:{other}')
x = Test(100)
x | 55
這里就涉及到一個(gè)盲點(diǎn)和兩個(gè)真正的知識(shí)點(diǎn)。這個(gè)盲點(diǎn)就是,你可能以為 |到|是一個(gè)字符,但是它是3個(gè)字符;你可能會(huì)把|拼接|看做一個(gè)整體,但是它實(shí)際上是3個(gè)部分:左邊的|、拼接和右邊的|。
我們把空格加上,就很明顯了:
兩個(gè)真正的知識(shí)點(diǎn),就是__or__和__ror__這兩個(gè)魔術(shù)方法和偏函數(shù)partial。而Change本身就是一個(gè)普通的類而已,__or__和__ror__定義了這個(gè)類的實(shí)例在左側(cè)遇到|時(shí),右側(cè)遇到|時(shí)的具體行為。
我們一個(gè)一個(gè)來講。首先是__or__。它定義了實(shí)例的右側(cè)遇到|時(shí)的具體行為。例如,我們用一個(gè)簡單的代碼來進(jìn)行測試:
class Test:
def __init__(self, num):
self.num = num
def __or__(self, other):
print(f'我右邊有一個(gè)東西,它是:{other}')
x = Test(100)
x | 55
運(yùn)行效果如下圖所示:
但如果你把豎線放在左邊,他就會(huì)報(bào)錯(cuò),如下圖所示:
而__ror__就是用來定義|在實(shí)例左邊的時(shí)候,它的行為:
所以,我們最開始的例子中,2 |到| 10,實(shí)際上應(yīng)該理解為:
- 到是Change(range)返回的實(shí)例
- 2 | 到 生成一個(gè)中間對(duì)象,我們假設(shè)它是x
- x | 10 生成結(jié)果
在我們演示的例子中,2 | 到首先進(jìn)入了Change類的__ror__方法中:
def __ror__(self, other):
self.func = partial(self.func, other)
return self
其中,一開始的self.func就是我們在初始化實(shí)例Change(range)時(shí)傳入的參數(shù)range。所以partial(self.func, other)等價(jià)于partial(range, 2)。關(guān)于偏函數(shù)partial,大家可以看我這篇文章:偏函數(shù):在Python中設(shè)定默認(rèn)參數(shù)的另一種辦法。簡單來說,使用偏函數(shù),可以給一個(gè)真正的函數(shù)傳一部分參數(shù),過一會(huì)再補(bǔ)剩下的參數(shù)。
可能大家在日常的開發(fā)者,很少會(huì)讓一個(gè)實(shí)例方法返回self。關(guān)于這個(gè)寫法,大家可以看我的這一篇文章:一日一技:在Python里面實(shí)現(xiàn)鏈?zhǔn)秸{(diào)用。也就是說,1 | 到返回的,依然是Change類的一個(gè)實(shí)例,我們簡稱它為x。這個(gè)實(shí)例的屬性self.func的值是partial(range, 2)。
接下來,x | 10,調(diào)用的是__or__方法,于是,此時(shí)執(zhí)行的是partial(range, 2)(10)。偏函數(shù)的參數(shù)補(bǔ)全了,于是它里面的range真正運(yùn)行了起來,成為了range(2, 10)。
至此,這個(gè)Change類我們就解析透了。大家知道,在Python里面,魔術(shù)方法是有很多的,如果你不想用|,你還可以用其它的,例如:
或者:
或者:
同時(shí),這個(gè)Change類,你甚至可以直接當(dāng)做裝飾器來使用。任何能夠接收兩個(gè)參數(shù)的函數(shù),都能使用這個(gè)裝飾器。例如:
最后總結(jié)一下。大家都知道,我是非常反對(duì)在工作代碼中炫技的,因?yàn)殪偶嫉膶懛ê茈y讀,很難維護(hù)。今天這個(gè)炫技的方法,雖然我也不推薦大家用在工作中,但是它短短8行代碼里面,包含了很多個(gè)知識(shí)點(diǎn),這就值得大家玩一玩了。