FastAPI SQLAlchemy 会话返回空对象：如何解决？

mysql

2024-03-22 14:42:46

FastAPI SQLAlchemy 会话返回空对象：解决之道

导言

在使用 FastAPI 和 SQLAlchemy 时，从会话中查询数据后，可能会遇到返回对象为空的问题。虽然打印该对象时显示实际数据，但在会话之外访问该对象时却为空。本文将深入探讨导致此问题的原因，并提供行之有效的解决方案。

问题分析

根源：上下文管理器

问题的根源在于使用上下文管理器 session_scope() 来创建会话。虽然在上下文管理器内查询数据并正确赋值，但在上下文管理器外访问时，实际访问的是一个不同的会话，其中没有数据。这是因为上下文管理器创建了一个新的会话作用域。

解决方案：expire_on_commit 标志

为了解决这个问题，需要在上下文管理器之外将数据从查询的会话移动到主会话。一种方法是使用 expire_on_commit 标志。

步骤：

在上下文管理器内，使用 sess.query(...) 查询数据并将其分配给变量。
设置变量的 expire_on_commit 标志为 False。
提交会话。

代码示例：

@router.get("/me")
def getusers(dependencies=Depends(JWTBearer())):
    user = None
    with session_scope() as sess:
        x = sess.query(TokenTable).where(TokenTable.access_toke == dependencies).first()
        user = sess.query(User).where(User.id == x.user_id).first()
        user.expire_on_commit = False
        sess.commit()
    return user