901f3dce6e
MicroPython's original implementation of __aiter__ was correct for an earlier (provisional) version of PEP492 (CPython 3.5), where __aiter__ was an async-def function. But that changed in the final version of PEP492 (in CPython 3.5.2) where the function was changed to a normal one. See https://www.python.org/dev/peps/pep-0492/#why-aiter-does-not-return-an-awaitable See also the note at the end of this subsection in the docs: https://docs.python.org/3.5/reference/datamodel.html#asynchronous-iterators And for completeness the BPO: https://bugs.python.org/issue27243 To be consistent with the Python spec as it stands today (and now that PEP492 is final) this commit changes MicroPython's behaviour to match CPython: __aiter__ should return an async-iterable object, but is not itself awaitable. The relevant tests are updated to match. See #6267.
48 lines
992 B
Python
48 lines
992 B
Python
# test waiting within "async for" __anext__ function
|
|
|
|
import sys
|
|
if sys.implementation.name in ('micropython', 'circuitpython'):
|
|
# uPy 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 ARange:
|
|
def __init__(self, high):
|
|
print('init')
|
|
self.cur = 0
|
|
self.high = high
|
|
|
|
def __aiter__(self):
|
|
print('aiter')
|
|
return self
|
|
|
|
async def __anext__(self):
|
|
print('anext')
|
|
print('f returned:', await f(20))
|
|
if self.cur < self.high:
|
|
val = self.cur
|
|
self.cur += 1
|
|
return val
|
|
else:
|
|
raise StopAsyncIteration
|
|
|
|
async def coro():
|
|
async for x in ARange(4):
|
|
print('x', x)
|
|
|
|
o = coro()
|
|
try:
|
|
while True:
|
|
print('coro yielded:', o.send(None))
|
|
except StopIteration:
|
|
print('finished')
|