python - 如何在类似 future 的对象的 __await__ 中等待？

Question

PEP 0492添加新的 __await__ 魔术方法。实现这个方法的对象变成了future-like object，可以使用await来等待。很清楚:

import asyncio


class Waiting:
    def __await__(self):
        yield from asyncio.sleep(2)
        print('ok')

async def main():
    await Waiting()

if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())

好的，但是如果我想调用一些 async def 定义的函数而不是 asyncio.sleep 怎么办？我不能使用 await 因为 __await__ 不是 async 函数，我不能使用 yield from 因为原生协程需要 await 表达式:

async def new_sleep():
    await asyncio.sleep(2)

class Waiting:
    def __await__(self):
        yield from new_sleep()  # this is TypeError
        await new_sleep()  # this is SyntaxError
        print('ok')

我该如何解决？

Answer 1

使用直接 __await__() 调用:

async def new_sleep():
    await asyncio.sleep(2)

class Waiting:
    def __await__(self):
        return new_sleep().__await__()

Yury Selivanov(PEP 492 的作者)为 aioodbc library 推荐了该解决方案。