Python 3.9：在 __anext__ 中使用 Yield 时，“async_generator 不能在 'await' 表达式中使用”

Question

我正在尝试以下操作：

class Payload_Session_Generator:
    def __init__(self):
        pass

    async def __anext__(self):
        async for payload in generate_fb_payload():
            if type(payload) != str:
                yield payload
            else:
                StopAsyncIteration

    def __aiter__(self):
        return self

然后将其作为实例传递给不同的函数，并显式调用

__aiter__

方法，并且

_iter

是上述类的对象：

chunk = await self._iter.__anext__()

这会产生以下错误：

TypeError：对象 async_generator 不能在“await”表达式中使用

Answer 1

更改您的代码，例如

class PayloadSessionGenerator:
    def __init__(self):
        pass
    
    async def __aiter__(self):
        async for payload in generate_fb_payload():
            if type(payload) != str:
                yield payload
            else:
                StopAsyncIteration

尝试下一个通话

x = Payload_Session_Generator()
chunk = await x.__anext__()

Answer 2

异步生成器与常规生成器类似：

__anext__

每次调用时都必须提供一个元素。由于（异步）生成器是（异步）迭代器，因此它必须返回对象本身。

因此，

__anext__

应该使用

return

代替

yield

：

class Payload_Session_Generator:
    async def __anext__(self):
        async for payload in generate_fb_payload():
            if type(payload) != str:
                return payload            # return one element
            else:
                raise StopAsyncIteration  # raise at end of iteration

    def __aiter__(self):
        return self

请注意，如果一个类仅使用

__anext__

和

__aiter__

，则编写适当的异步生成器会更简单。异步生成器在主体中使用

async def

和

yield

定义：

async def payload_session_generator():
    async for payload in generate_fb_payload():
        if type(payload) != str:
            yield payload  # yield each element
        else:
            break          # exiting ends iteration