examples: add async_coroutines

This commit is contained in:
Casper da Costa-Luis 2020-07-16 20:37:00 +01:00
parent 7289d88cd8
commit 13f847eabc
No known key found for this signature in database
GPG Key ID: 986B408043AE090D
1 changed files with 37 additions and 0 deletions

View File

@ -0,0 +1,37 @@
"""
Asynchronous examples using `asyncio`, `async` and `await` on `python>=3.7`.
"""
import asyncio
from tqdm.asyncio import tqdm, trange
def count(start=0, step=1):
i = start
while True:
new_start = yield i
if new_start is None:
i += step
else:
i = new_start
async def main():
N = int(1e6)
async for row in tqdm(trange(N, desc="inner"), desc="outer"):
if row >= N:
break
with tqdm(count(), desc="coroutine", total=N + 2) as pbar:
async for row in pbar:
if row == N:
pbar.send(-10)
elif row < 0:
assert row == -9
break
# should be under 10 seconds
for i in tqdm.as_completed(list(map(asyncio.sleep, [1] * 10)),
desc="as_completed"):
await i
if __name__ == "__main__":
asyncio.run(main())