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())
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')
How can I solve it?
Use direct __await__()
call:
async def new_sleep(): await asyncio.sleep(2) class Waiting: def __await__(self): return new_sleep().__await__()
The solution was recommended by Yury Selivanov (the author of PEP 492) for aioodbc library
Short version: await foo
can be replaced by yield from foo.__await__()
Combining all the ideas from the other answers -
in the simplest case, just delegating to another awaitable works:
def __await__(self): return new_sleep().__await__()
This works because the __await__
method returns an iterator (see PEP 492), so returning another __await__
's iterator is fine.
This means, of course, that we can't change the suspension behavior of the original awaitable at all. The more general approach is to mirror the await
keyword and use yield from
- this lets us combine multiple awaitables' iterators into one:
def __await__(self): # theoretically possible, but not useful for my example: #yield from something_else_first().__await__() yield from new_sleep().__await__()
Here's the catch: this is not doing exactly the same thing as the first variant! yield from
is an expression, so to do exactly the same as before, we need to also return that value:
def __await__(self): return (yield from new_sleep().__await__())
This directly mirrors how we would write proper delegation using the await
syntax:
return await new_sleep()
extra bit - what's the difference between these two?
def __await__(self): do_something_synchronously() return new_sleep().__await__() def __await__(self): do_something_synchronously() return (yield from new_sleep().__await__())
The first variant is a plain function: when you call it, do_...
is executed and an iterator returned. The second is a generator function; calling it doesn't execute any of our code at all! Only when the returned iterator is yielded for the first time will do_...
be executed. This makes a difference in the following, a little contrived situation:
def foo(): tmp = Waiting.__await__() do_something() yield from tmp
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With