头图

使用Python做web开发,OR Mapping的部分,通常会用到Sqlalchemy,它是一个相对重量级的数据库封装层,提供了丰富的调用接口。

用Sqlalchemy实现查询是相当简单的,比如下面的SQL:

select * from table_a where col1=3 and col2='abc'

写成代码就是:

TableA.query.filter(TableA.col1==3).filter(TableA.col2=='abc')

但有的时候,我们会发现代码中经常有两条或多条查询,它们的条件是一模一样的,只有select后面有差别,比如我还想取个总数:

select count(*) from table_a where col1=3 and col2='abc'
db.session.query(func.count('*').filter(TableA.col1==3).filter(TableA.col2=='abc')

这就很别扭了,同样的代码被重复编写,如果这个条件复杂,需要经过各种逻辑判断拼接的话,就更加难受:每条拼接语句都要copy一次,代码的坏味道快溢出屏幕了~~~

有没有办法让它们共享同一个filter呢?还有有的,可以用list嵌套表达式,先把filter保存起来,最后再一起传给query对象:

all_filters = [
    TableA.col1==3,
    TableA.col2=='abc'
]

records = TableA.query.filter(*all_filters).all()

count = db.session.query(
    func.count('*')
).filter(
    *all_filters
).first()

网上描述这个问题的文章不多,google了半天,才找到一个答案:Apply condition based multiple filters in SQLAlchemy query

问题是它还有个错误,把*all_filters写成了**all_filters,害的我差点放弃。


songofhawk
303 声望24 粉丝