Python SQLAlchemy如何判断查询返回的空结果并处理

核心问题修正点 #

• 判断查询结果是否为空不要直接判断execute返回的结果对象，调用scalar_one_or_none()方法即可：匹配到记录返回对应ORM对象，无匹配直接返回None，直接做布尔判断就行，不需要写冗余的遍历、异常捕获逻辑。
• 你之前写的exists逻辑死循环是因为子句查询目标错了：你写的exists是判断Software表的name是否存在，和你要校验的Status表software_id重复的逻辑无关，不需要硬套exists写法，单条记录取数判空是最容易上手的方案。
• 不要用裸except吞异常，会把所有代码报错都隐藏，出问题很难排查。

可直接运行的修正代码 #

def getStatusData():
    session = Session(engine)
    locale.setlocale(locale.LC_ALL, 'en_US')  # 解析'1 Jan 2020'格式日期
    statusList = getStatusList()
    for content in statusList:
        prev_datestring = f"{content['date']} {content['time']}"  # 拼接日期时间字段
        prev_date = datetime.strptime(prev_datestring, '%d %b %Y %H:%M:%S') # 转为Python datetime格式
        
        # 获取对应software的id，无匹配则跳过当前循环
        software_id = session.execute(
            select(db.Software.id).where(db.Software.name == content['name'])
        ).scalar_one_or_none()
        if not software_id:
            continue

        # 查询status表是否存在对应software_id的条目
        exist_status = session.execute(
            select(db.Status).where(db.Status.software_id == software_id)
        ).scalar_one_or_none()

        if exist_status:
            # 条目已存在，执行更新，替换成你需要更新的字段逻辑
            exist_status.record_time = prev_date
        else:
            # 条目不存在，新建记录
            new_status = db.Status(
                software_id=software_id,
                record_time=prev_date
                # 按业务需求补充其他字段
            )
            session.add(new_status)
    
    # 所有操作完成后统一提交事务，循环内不要反复提交
    session.commit()
    session.close()

更高效的UPSERT方案 #

如果不想写先查再判断的逻辑，可以直接用SQLite原生支持的INSERT ON CONFLICT语法（即UPSERT），单条语句完成存在则更新、不存在则插入，执行效率更高，不会出现查询和写入之间的逻辑冲突：

# 顶部先导入SQLite专属的insert方法
from sqlalchemy.dialects.sqlite import insert

# 把上面循环里的查-判-写逻辑替换成下面的代码即可
upsert_stmt = insert(db.Status).values(
    software_id=software_id,
    record_time=prev_date
    # 补充其他需要写入的字段
).on_conflict_do_update(
    # 指定判断重复的唯一键字段，即你的software_id字段
    index_elements=['software_id'],
    # 指定冲突时需要更新的字段值
    set_=dict(record_time=prev_date)
)
session.execute(upsert_stmt)

注意：用UPSERT方案需要确保status表的software_id字段设置了唯一约束，否则无法触发冲突判断逻辑。

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