我怎样才能在类似未来的对象的 __await__ 中等待?

问题描述

PEP 0492 添加了新的 __await__ 魔法方法.实现此方法的对象将成为类未来对象，并且可以使用 await 等待.很明显:

PEP 0492 adds new __await__ magic method. Object that implements this method becomes future-like object and can be awaited using await. It's clear:

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 表达式:

Ok, but what if I want to call some async def defined function instead of asyncio.sleep? I can't use await because __await__ is not async function, I can't use yield from because native coroutines requires await expression:

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')

我该如何解决?

推荐答案

使用直接__await__()调用:

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

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

该解决方案由 Yury Selivanov(PEP 492 的作者)推荐用于 aioodbc 库

The solution was recommended by Yury Selivanov (the author of PEP 492) for aioodbc library

这篇关于我怎样才能在类似未来的对象的 __await__ 中等待?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！