python - 如何在可等待的子类中使用 super()？

Question

我想通过子类化现有的可等待类来为其添加新功能。

让我们从一个非常简单的基类开始创建对象，这些对象在短暂休眠后异步返回 99。子类应该只向结果添加 +1。

我找不到使用 super() 来引用基类的正确方法。

import asyncio

class R99:
    def __await__(self):
        loop = asyncio.get_event_loop()
        fut = loop.create_future()
        loop.call_later(0.5, fut.set_result, 99)
        return fut.__await__()

class R100(R99):
    async def add1(self):
        v = await R99()
        #v = await super().__await__()   # <== error
        return v + 1

    def __await__(self):
        return self.add1().__await__()

async def test():
    print(await R99())
    print(await R100())

asyncio.get_event_loop().run_until_complete(test())

Answer 1

await method必须返回一个迭代器，所以你可以使它成为一个生成器并使用 yield from语法:

class R100(R99):

    def __await__(self):
        v = yield from super().__await__()
        return v + 1