如何在SQLAlchemy中实现带可选参数的多条件SELECT查询？

在开发基础联系人管理应用时，使用SQLAlchemy查询联系人遇到问题：希望用户输入部分信息（比如仅姓氏）就能匹配所有对应联系人，但当前查询逻辑会把为None的参数（如名字、年龄）当作查询条件，要求对应字段为None，导致无法返回预期结果。比如传入last_name='Vincent'时，因为first_name和age为None，查询会要求这两个字段的值也是None，所以查不到数据库中已有的3条Vincent姓氏的联系人。

修改方案 #

核心思路是仅当参数不为None时，才将该字段的匹配条件加入查询，通过动态构建查询条件列表实现：

修改select_contact函数如下：

def select_contact(last_name=None, first_name=None, age=None):
    with Session(engine) as session:
        # 初始化空条件列表
        conditions = []
        # 仅在参数非空时添加对应查询条件
        if last_name is not None:
            conditions.append(Contact.last_name == last_name)
        if first_name is not None:
            conditions.append(Contact.first_name == first_name)
        if age is not None:
            conditions.append(Contact.age == age)
        
        # 构建查询，无有效条件时返回所有联系人
        stmt = select(Contact).where(*conditions)
        
        for contact in session.scalars(stmt):
            print(contact)

关键修改说明 #

• 抛弃原有的in_([参数, None])写法：这种写法会强制匹配字段值为参数或None，不符合“忽略空参数”的需求
• 动态收集有效条件：通过判断参数是否为None，只将有值的查询条件加入列表
• 展开条件列表：使用*conditions将列表中的条件传入where方法，SQLAlchemy会自动用AND连接所有有效条件
• 处理全空参数：若所有参数都是None，查询会返回数据库中所有联系人

另外注意修正原代码的笔误：最后调用的函数名select_contract应改为select_contact，否则会触发未定义函数错误。

内容的提问来源于stack exchange，提问作者asakha_dev