一日一技：8行炫技代碼，知識點多得不得了

作者：kingname 2022-01-27 20:45:08

我們今天來看一段炫技代碼。它可以把任何能接收兩個參數(shù)的函數(shù)定義成一個特殊的運算符。

例如，我們知道Python里面的range函數(shù)，最少可以接收1個參數(shù)，最多能夠接收3個參數(shù)。當只有兩個參數(shù)的時候，格式為range(開始, 結束)，從開始數(shù)字到結束數(shù)字逐次加1，左閉右開。使用今天的方法，可以把它的寫法改為開始 |到| 結束，如下圖所示：

又比如，urllib.parse.urljoin可以把域名和一個相對路徑拼接起來: urljoin(域名, 相對路徑)，我們也可以改寫成如下圖所示的格式：

這種炫技有余，實用不足的功能是怎么實現(xiàn)的呢?其實原理非常簡單，只有8行代碼：

class Test:
    def __init__(self, num):
        self.num = num
    def __or__(self, other):
        print(f'我右邊有一個東西，它是：{other}')

x = Test(100)
x | 55

這里就涉及到一個盲點和兩個真正的知識點。這個盲點就是，你可能以為 |到|是一個字符，但是它是3個字符;你可能會把|拼接|看做一個整體，但是它實際上是3個部分：左邊的|、拼接和右邊的|。

我們把空格加上，就很明顯了：

兩個真正的知識點，就是__or__和__ror__這兩個魔術方法和偏函數(shù)partial。而Change本身就是一個普通的類而已，__or__和__ror__定義了這個類的實例在左側遇到|時，右側遇到|時的具體行為。

我們一個一個來講。首先是__or__。它定義了實例的右側遇到|時的具體行為。例如，我們用一個簡單的代碼來進行測試：

class Test:
    def __init__(self, num):
        self.num = num
    def __or__(self, other):
        print(f'我右邊有一個東西，它是：{other}')

x = Test(100)
x | 55

運行效果如下圖所示：

但如果你把豎線放在左邊，他就會報錯，如下圖所示：

而__ror__就是用來定義|在實例左邊的時候，它的行為：

所以，我們最開始的例子中，2 |到| 10，實際上應該理解為：

到是Change(range)返回的實例
2 | 到生成一個中間對象，我們假設它是x
x | 10 生成結果

在我們演示的例子中，2 | 到首先進入了Change類的__ror__方法中：

  def __ror__(self, other):
        self.func = partial(self.func, other)
        return self

其中，一開始的self.func就是我們在初始化實例Change(range)時傳入的參數(shù)range。所以partial(self.func, other)等價于partial(range, 2)。關于偏函數(shù)partial，大家可以看我這篇文章：偏函數(shù)：在Python中設定默認參數(shù)的另一種辦法。簡單來說，使用偏函數(shù)，可以給一個真正的函數(shù)傳一部分參數(shù)，過一會再補剩下的參數(shù)。

可能大家在日常的開發(fā)者，很少會讓一個實例方法返回self。關于這個寫法，大家可以看我的這一篇文章：一日一技：在Python里面實現(xiàn)鏈式調(diào)用。也就是說，1 | 到返回的，依然是Change類的一個實例，我們簡稱它為x。這個實例的屬性self.func的值是partial(range, 2)。

接下來，x | 10，調(diào)用的是__or__方法，于是，此時執(zhí)行的是partial(range, 2)(10)。偏函數(shù)的參數(shù)補全了，于是它里面的range真正運行了起來，成為了range(2, 10)。

至此，這個Change類我們就解析透了。大家知道，在Python里面，魔術方法是有很多的，如果你不想用|，你還可以用其它的，例如：