SQLAlchemy ORM / Python：仅当搜索值不是 None 时才包含查询过滤器的一行（内联 'if'）

Question

我可以在 Python 中实现方法级别的内联 if 吗？

让我们以这种（简化的）情况为例：

# DB-level 'or'
db.session.query(TableClass).filter(or_(TableClass.column_1 == value_1,
                                        TableClass.column_2 == value_2)).first()

# basic Python-level 'if'
if value_1:
    db.session.query...value_1
if value_2:
    db.session.query...value_2

不是像上面那样查询数据库或添加额外的 if 语句，是否有可能实现相同的内联，如：

# desired inline Python 'if' at method level
db.session.query(TableClass).filter({
           if value_1 TableClass.column_1 == value_1 \
           else if value_2 TableClass.column_2 == value_2 else None
    }).first()

# ^-- or something similar at method level

或者，让 SQLAlchemy 过滤器的一部分仅在基础值存在时才处于活动状态。

这里的主要动机是 DRY 并避免数据库搜索 None 值。

尝试了 starred expression with if statement as suggested 的内联组合：

session.query(Foo).filter((*[Foo.attr1==attr1, Foo.attr2==attr2] if attr2 else *[Foo.attr1==attr1]))

但我在 if 得到 SyntaxError: invalid syntax。

更新

现在我将条件定义为可以附加在一起的单独变量：

query = session.query(Table)

conditions = []
if abc:
    conditions.append(Table.field1 == foo)
if def:
    conditions.append(Table.field2 == bar)

query = query.filter(or_(*conditions))

尽管我很欣赏 Pierre V 在下面为加星标的表达式和 or_(TableClass.column_1 == value_1, true() if value_1 is None else false()) 建议修复语法的回答。

Answer 1

您可以通过移动括号来消除语法错误：

session.query(Foo).filter(*([Foo.attr1==attr1, Foo.attr2==attr2] if attr2 else [Foo.attr1==attr1]))

否则，您可能会使用 true() and false() Sqlalchemy 表达式获得所需的结果。

类似于：

db.session.query(TableClass).filter(
    or_(TableClass.column_1 == value_1, true() if value_1 is None else false()),
    or_(TableClass.column_2 == value_2, true() if value_2 is None else false())
).first()

但恕我直言，感觉比

笨重得多

query = db.session.query(TableClass)
if value_1:
    query = query.filter(TableClass.column_1 == value_1)
if value_2:
    query = query.filter(TableClass.column_2 == value_2)

SQLAlchemy ORM / Python：仅当搜索值不是 None 时才包含查询过滤器的一行（内联 'if'）

SQLAlchemy ORM / Python: one-liner to include query filter only if searched value is not None (inline 'if')

python

methods

if-statement

sqlalchemy

inline