一日一技：在 Python 裏面實現鏈式調用

攝影：產品經理

薄如蟬翼

我們在使用 Django 的 models 查詢數據庫時，可以看到有這種寫法：

form app.models import XXX 
query = XXX.objects.all() 
query = query.filter(name=123, age=456).filter(salary=999)

在這種寫法裏面，query 對象有一個filter方法，這個方法的返回數據還可以繼續調用filter方法，可以這樣無限制地調用下去。

這種寫法是怎麼實現的呢？

如果我們直接寫一個類的方法，看看能不能這樣調用：

class Query: 
def filter(self): 
pass 
query = Query() 
query.filter().filter()

直接對query.filter()返回的結果再調用一次filter，就會導致報錯了。這是因爲在沒有顯式寫 return 語句的時候，方法會返回None，而 None 對象是沒有所謂的filter方法的。

那麼什麼東西有filter方法呢？顯然我們的 query 對象有filter方法。那麼如何讓這個方法返回自身這個對象呢？

這個時候，我們就要看看我們在定義類方法的時候，總會寫的的第一個參數self了。幾乎每個類方法裏面都會有它。大家只知道在類裏面調用類方法的時候可以用self.xxx()，在調用類屬性的時候可以用self.yy，那麼有沒有思考過，這個東西如果單獨使用會怎麼樣呢？

實際上，self指的就是這個類實例化成一個對象以後，這個對象自身。而這個對象顯然是有filter方法的。所以我們修改一下filter方法，讓它返回self:

class Query: 
    def filter(self): 
        return self 
query = Query() 
query.filter().filter()

從圖中可以看出，現在已經不會報錯了。那麼回到最開始的問題，Django 裏面的鏈式調用傳入查詢參數是如何實現的呢？

實際上這裏涉及到一個惰性查詢的問題。

當我們不停調用.filter()方法的時候，Django 會把這些查詢條件全部緩存起來，只有當我們需要獲取結果，或者查詢滿足條件的數據有多少條時，它纔會真正地連接數據庫去查詢。

所以我們這裏要模擬這個環境，把查詢條件緩存起來。

那麼爲了獲取調用方法時傳入的參數名，我們就要使用**kwargs參數。這個參數可以接受所有的 key=value 形式的參數：

class Query(): 
    def __init__(self): 
        self.query_condition = {} 

    def filter(self, **kwargs): 
        self.query_condition.update(kwargs) 
        return self 

query = Query() 
a = query.filter(name='kingname').filter(age__gt=15, address='yyyyyy').filter(salary=99999) 
print(query.query_condition)

運行效果如下圖所示：

在真正需要輸出結果的時候，再使用這些緩存的條件，去數據庫中查詢結果即可。

本文由 Readfog 進行 AMP 轉碼，版權歸原作者所有。
來源：https://mp.weixin.qq.com/s/Y689WhpOopHdI8hbCGcFMg