python-asyncio - 将 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 使用我们理解的原子事务。此外，如果上下文将切换到另一个协程，我们还使用可以在另一个 session 中使用的线程本地缓存。
我同意我们应该在文档中添加有关它的信息。
为确保一切正常，您应该使用 db_session作为上下文管理器，并确保您在此代码块中没有异步调用。
如果您的端点不是异步的，您也可以使用 db_session他们的装饰师。
在 Pony 中，我们同意使用 ContextVar而不是 Local应该对某些情况有所帮助。
一句话的答案是:使用小的短期 session ，不要用异步中断它们。