异步上下文管理器

Question

我有一个异步API，用于连接邮件并将其发送到SMTP服务器，该服务器具有一些设置并已删除。因此，它非常适合使用contextmanagerPython
3的contextlib。

虽然，我不知道是否可以写，因为它们都使用生成器语法来写。

这可能证明了这个问题（包含yield-base和async-await语法的混合，以演示异步调用和上下文管理器的yield之间的区别）。

@contextmanager
async def smtp_connection():
    client = SMTPAsync()
    ...

    try:
        await client.connect(smtp_url, smtp_port)
        await client.starttls()
        await client.login(smtp_username, smtp_password)
        yield client
    finally:
        await client.quit()

目前在python中可能发生这种情况吗？以及如何使用该with as语句？如果没有，我可以实现这一目标的另一种方法-也许使用旧样式的上下文管理器？

Answer 1

在Python 3.7中，您将能够编写：

from contextlib import asynccontextmanager

@asynccontextmanager
async def smtp_connection():
    client = SMTPAsync()
    ...

    try:
        await client.connect(smtp_url, smtp_port)
        await client.starttls()
        await client.login(smtp_username, smtp_password)
        yield client
    finally:
        await client.quit()

在3.7发布之前，您可以使用此async_generator软件包。在3.6上，您可以编写：

# This import changed, everything else is the same
from async_generator import asynccontextmanager

@asynccontextmanager
async def smtp_connection():
    client = SMTPAsync()
    ...

    try:
        await client.connect(smtp_url, smtp_port)
        await client.starttls()
        await client.login(smtp_username, smtp_password)
        yield client
    finally:
        await client.quit()

如果您想一直回到3.5，可以这样写：

# This import changed again:
from async_generator import asynccontextmanager, async_generator, yield_

@asynccontextmanager
@async_generator      # <-- added this
async def smtp_connection():
    client = SMTPAsync()
    ...

    try:
        await client.connect(smtp_url, smtp_port)
        await client.starttls()
        await client.login(smtp_username, smtp_password)
        await yield_(client)    # <-- this line changed
    finally:
        await client.quit()