mirror of https://github.com/tqdm/tqdm.git
examples: add async_coroutines
This commit is contained in:
parent
7289d88cd8
commit
13f847eabc
|
@ -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())
|
Loading…
Reference in New Issue