如何将Sqlalchemy查询结果存入Python变量以消除代码冗余？

错误根因 #

你给query_variable设置的初始值是整数0，如果所有if/elif分支的条件都未命中，该变量不会被重新赋值，遍历整数时就会触发'int' object is not iterable的类型错误。即使你的业务逻辑保证一定会命中其中一个分支，Python的类型校验也会判定该变量存在为整数的可能性，仍可能触发报错。

基础修复方法 #

将query_variable的初始值改为空列表即可，Sqlalchemy的.all()方法本身返回的就是查询结果组成的列表，类型完全对齐：

# 初始值设为空列表，和查询返回类型一致
query_variable = []

if condition1:
    query_variable = s.query(Invoice).filter(Invoice.trading_partner_name==trading_partner_name).all()
elif condition2:
    query_variable = s.query(Invoice).filter(...).all()
elif condition3:
    query_variable = s.query(Invoice).filter(...).all()
# 其余分支依次补充

# 统一执行后续处理逻辑，无匹配分支时遍历空列表不会报错
for data in query_variable:
    # 你的长处理逻辑
    ...

可选优化方案（简化12个if/elif分支） #

你可以把条件和对应的过滤规则提前定义为有序映射列表，避免堆砌大量分支代码，后续维护也更方便：

# 映射列表顺序和原有if/elif的判断优先级保持一致
query_rules = [
    (condition1, lambda: Invoice.trading_partner_name == trading_partner_name),
    (condition2, lambda: ...),
    (condition3, lambda: ...),
    # 其余规则按顺序添加即可
]

query_variable = []
for cond, filter_exp in query_rules:
    if cond:
        query_variable = s.query(Invoice).filter(filter_exp()).all()
        break

# 后续处理逻辑保持不变
for data in query_variable:
    ...

额外说明 #

Sqlalchemy的查询结果本身就是普通的Python可迭代对象，直接赋值给变量没有特殊要求，只要保证变量的类型统一即可。如果你的业务逻辑要求必须命中一个查询条件，可以把初始值设为None，遍历前增加校验逻辑：

query_variable = None
# 分支赋值逻辑不变
if query_variable is None:
    raise ValueError("未匹配到任何有效查询条件")
for data in query_variable:
    ...

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