如何在Python里面實現(xiàn)鏈?zhǔn)秸{(diào)用

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

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

在這種寫法里面，query對象有一個filter方法，這個方法的返回數(shù)據(jù)還可以繼續(xù)調(diào)用filter方法，可以這樣無限制地調(diào)用下去。

這種寫法是怎么實現(xiàn)的呢？

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

class Query:
    def filter(self):
        pass

query = Query()
query.filter().filter()

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

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

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

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

class Query:
    def filter(self):
        return self

query = Query()
query.filter().filter()

從圖中可以看出，現(xiàn)在已經(jīng)不會報錯了。那么回到最開始的問題，Django里面的鏈?zhǔn)秸{(diào)用傳入查詢參數(shù)是如何實現(xiàn)的呢？

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

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

所以我們這里要模擬這個環(huán)境，把查詢條件緩存起來。

那么為了獲取調(diào)用方法時傳入的參數(shù)名，我們就要使用**kwargs參數(shù)。這個參數(shù)可以接受所有的key=value形式的參數(shù)：

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)

運行效果如下圖所示：

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

推薦閱讀

困擾 Chrome 用戶多年的大問題，終于要解決了！

Pandas中的這3個函數(shù)，沒想到竟成了我數(shù)據(jù)處理的主力