# test waiting within async with enter/exit functions

# uPy allows normal generators to be awaitables.
# CircuitPython does not.
# In CircuitPython you need to have an __await__ method on an awaitable like in CPython;
#  and like in CPython, generators do not have __await__.

class Awaitable:
    def __init__(self, x):
        self.x = x

    def __await__(self):
        print('f start:', self.x)
        yield self.x + 1
        yield self.x + 2
        return self.x + 3

class AContext:
    async def __aenter__(self):
        print('enter')
        print('f returned:', await Awaitable(10))
    async def __aexit__(self, exc_type, exc, tb):
        print('exit', exc_type, exc)
        print('f returned:', await Awaitable(20))

async def coro():
    async with AContext():
        print('body start')
        print('body f returned:', await Awaitable(30))
        print('body end')

o = coro()
try:
    while True:
        print('coro yielded:', o.send(None))
except StopIteration:
    print('finished')