将 PonyOrm 与 FastApi 一起使用的正确方法是什么？

Question

对于个人项目，我将 PonyOrm 与 FastApi 一起使用；有没有一种经典的方法可以在端点的整个异步生命周期调用中保持 db_session ？

PonyOrm 的文档中谈到了使用装饰器和产量；但它对我不起作用，所以在查看了其他 Github 项目后，我发现这个解决方法效果很好。

但我真的不知道幕后发生了什么，以及为什么 Pony 的文档对异步主题不准确。

def _enter_session():
    session = db_session(sql_debug=True)
    Request.pony_session = session
    session.__enter__()


def _exit_session():
    session = getattr(Request, 'pony_session', None)
    if session is not None:
        session.__exit__()


@app.middleware("http")
async def add_pony(request: Request, call_next):
    _enter_session()
    response = await call_next(request)
    _exit_session()
    return response

然后在一个依赖中，例如：

async def current_user(
        username: str = Depends(current_user_from_token)) -> User:

    with Request.pony_session:
        # db actions 

在端点调用中：

@router.post("/token", response_model=Token)
async def login_for_access_token(
        request: Request,
        user_agent: Optional[str] = Header(None),
        form_data: OAuth2PasswordRequestForm = Depends()):

    status: bool = authenticate_user(
        form_data.username,
        form_data.password,
        request.client.host,
        user_agent)


@db_session
def authenticate_user(
        username: str,
        password: str,
        client_ip: str = 'Undefined',
        client_app: str = 'Undefined'):
    user: User = User.get(email=username)

如果你们有更好的方法或好的解释，我很想听听 :)

Answer 1

我有点像 PonyORM 开发人员和 FastAPI 用户。

async 和 Pony 的问题在于 Pony 使用我们理解的原子事务。此外，如果上下文将切换到另一个协程，我们还使用可以在另一个会话中使用的线程本地缓存。 我同意我们应该在文档中添加有关它的信息。

为确保一切正常，您应该使用 db_session 作为上下文管理器，并确保在此代码块中没有异步调用。

如果您的端点不是异步的，您也可以为它们使用db_session 装饰器。

在 Pony 中，我们同意在某些情况下使用 ContextVar 代替 Local 应该会有所帮助。

一句话的答案是：使用短暂的会话，不要用异步来打断它们。

Answer 2

尝试使用标准的 fastapi 依赖：

from fastapi import Depends

async def get_pony():
    with db_session(sql_debug=True) as session:
        yield session


async def current_user(
    username: str = Depends(current_user_from_token),
    pony_session = Depends(get_pony)) -> User:

    with pony_session:
        # db actions