如何在Python里面实现链式调用
共 2023字,需浏览 5分钟
·
2021-03-17 10:46
我们在使用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()
![](https://filescdn.proginn.com/17458668edb3b29f20a01ee2f105cadf/b11faa687c4f2607703e4442545aec22.webp)
直接对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()
![](https://filescdn.proginn.com/c47c46abd99db9a134e4ab02c72beb30/bc847de8bf5ffacfeb5b4ae3d54be2c5.webp)
从图中可以看出,现在已经不会报错了。那么回到最开始的问题,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)
运行效果如下图所示:
![](https://filescdn.proginn.com/a80f654cb6baa3cb25385a8b71d73fcf/2cc0884c3ea15259bc7d3fcb625b3fa9.webp)
在真正需要输出结果的时候,再使用这些缓存的条件,去数据库中查询结果即可。
推荐阅读
![](https://filescdn.proginn.com/428a236e17f81cda4311b8724faf3663/910941f92db400d91684dae442fd082a.webp)
困扰 Chrome 用户多年的大问题,终于要解决了!
![](https://filescdn.proginn.com/b655818e45e5814da2a10098926aca59/6123b03a4c2d78fa9b8ac8ed80508a29.webp)
Pandas中的这3个函数,没想到竟成了我数据处理的主力