3ec8b9a77c
With the aim of getting consistency, and removing the need to learn an additional term, replace uses of uPy/uPython with MPy/MicroPython. Rule of thumb was to use "MPy" abbreviation where "CPy" is used nearby, but the full word MicroPython otherwise. This work was funded through GitHub Sponsors. Signed-off-by: Angus Gratton <angus@redyak.com.au>
38 lines
862 B
Python
38 lines
862 B
Python
# test waiting within async with enter/exit functions
|
|
|
|
import sys
|
|
if sys.implementation.name == 'micropython':
|
|
# MicroPython allows normal generators to be awaitables
|
|
coroutine = lambda f: f
|
|
else:
|
|
import types
|
|
coroutine = types.coroutine
|
|
|
|
@coroutine
|
|
def f(x):
|
|
print('f start:', x)
|
|
yield x + 1
|
|
yield x + 2
|
|
return x + 3
|
|
|
|
class AContext:
|
|
async def __aenter__(self):
|
|
print('enter')
|
|
print('f returned:', await f(10))
|
|
async def __aexit__(self, exc_type, exc, tb):
|
|
print('exit', exc_type, exc)
|
|
print('f returned:', await f(20))
|
|
|
|
async def coro():
|
|
async with AContext():
|
|
print('body start')
|
|
print('body f returned:', await f(30))
|
|
print('body end')
|
|
|
|
o = coro()
|
|
try:
|
|
while True:
|
|
print('coro yielded:', o.send(None))
|
|
except StopIteration:
|
|
print('finished')
|