Incorporate @somacdavid's edits

Author: bsolomon1124

asyncio-walkthrough/asyncq.py

@@ -8,26 +8,26 @@
 import time
 
 
-async def makeitem(size=5):
+async def makeitem(size: int = 5) -> str:
     return os.urandom(size).hex()
 
 
-async def seconds():
+async def seconds() -> float:
     return time.perf_counter()
 
 
-async def randint(a, b):
+async def randint(a: int, b" int) -> int:
     return random.randint(a, b)
 
 
-async def randsleep(a=1, b=5, caller=None):
+async def randsleep(a: int = 1, b: int = 5, caller=None) -> None:
     i = await randint(a, b)
     if caller:
         print(f"{caller} sleeping for {i} seconds.")
     await asyncio.sleep(i)
 
 
-async def produce(name: int, q: asyncio.Queue):
+async def produce(name: int, q: asyncio.Queue) -> None:
     n = await randint(1, 5)
     for _ in it.repeat(None, n):  # Synchronous
         await randsleep(caller=f"Producer {name}")
@@ -37,7 +37,7 @@ async def produce(name: int, q: asyncio.Queue):
         print(f"Producer {name} added <{i}> to queue.")
 
 
-async def consume(name: int, q: asyncio.Queue):
+async def consume(name: int, q: asyncio.Queue) -> None:
     while True:
         await randsleep(caller=f"Consumer {name}")
         i, t = await q.get()
@@ -48,7 +48,7 @@ async def consume(name: int, q: asyncio.Queue):
         q.task_done()
 
 
-async def main(nprod, ncon):
+async def main(nprod: int, ncon: int):
     q = asyncio.Queue()
     producers = [asyncio.create_task(produce(n, q)) for n in range(nprod)]
     consumers = [asyncio.create_task(consume(n, q)) for n in range(ncon)]

asyncio-walkthrough/chained.py

@@ -6,11 +6,11 @@
 import time
 
 
-async def randint(a, b):
+async def randint(a: int, b: int) -> int:
     return random.randint(a, b)
 
 
-async def part1(n):
+async def part1(n: int) -> str:
     i = await randint(0, 10)
     print(f"part1({n}) sleeping for {i} seconds.")
     await asyncio.sleep(i)
@@ -19,7 +19,7 @@ async def part1(n):
     return result
 
 
-async def part2(n, arg):
+async def part2(n: int, arg: str) -> str:
     i = await randint(0, 10)
     print(f"part2{n, arg} sleeping for {i} seconds.")
     await asyncio.sleep(i)
@@ -28,24 +28,24 @@ async def part2(n, arg):
     return result
 
 
-async def chain(n):
+async def chain(n: int) -> None:
     start = time.perf_counter()
     p1 = await part1(n)
     p2 = await part2(n, p1)
     end = time.perf_counter() - start
     print(f"-->Chained result{n} => {p2} (took {end:0.2f} seconds).")
 
 
-async def main(*ns):
-    await asyncio.gather(*(chain(n) for n in ns))
+async def main(*args):
+    await asyncio.gather(*(chain(n) for n in args))
 
 
 if __name__ == "__main__":
     import sys
 
     random.seed(444)
-    ns = [1, 2, 3] if len(sys.argv) == 1 else map(int, sys.argv[1:])
+    args = [1, 2, 3] if len(sys.argv) == 1 else map(int, sys.argv[1:])
     start = time.perf_counter()
-    asyncio.run(main(*ns))
+    asyncio.run(main(*args))
     end = time.perf_counter() - start
     print(f"Program finished in {end:0.2f} seconds.")

asyncio-walkthrough/rand.py

@@ -13,11 +13,11 @@
 )
 
 
-async def randint(a, b):
+async def randint(a: int, b: int) -> int:
     return random.randint(a, b)
 
 
-async def makerandom(idx, threshold=6):
+async def makerandom(idx: int, threshold: int = 6) -> int:
     print(c[idx + 1] + f"Initiated makerandom({idx}).")
     i = await randint(0, 10)
     while i <= threshold: