TR updates, first round

Author: lpozo

python-asyncio/as_completed.py

@@ -8,8 +8,8 @@ async def coro(numbers):
 
 
 async def main():
-    task1 = asyncio.create_task(coro([3, 2, 1]))
-    task2 = asyncio.create_task(coro([10, 5, 2]))
+    task1 = asyncio.create_task(coro([10, 5, 2]))
+    task2 = asyncio.create_task(coro([3, 2, 1]))
     print("Start:", time.strftime("%X"))
     for task in asyncio.as_completed([task1, task2]):
         result = await task

python-asyncio/chained.py

@@ -3,6 +3,21 @@
 import time
 
 
+async def main():
+    user_ids = [1, 2, 3]
+    start = time.perf_counter()
+    await asyncio.gather(
+        *(get_user_with_posts(user_id) for user_id in user_ids)
+    )
+    end = time.perf_counter()
+    print(f"\n==> Total time: {end - start:.2f} seconds")
+
+
+async def get_user_with_posts(user_id):
+    user = await fetch_user(user_id)
+    await fetch_posts(user)
+
+
 async def fetch_user(user_id):
     delay = random.uniform(0.5, 2.0)
     print(f"User coro: fetching user by {user_id=}...")
@@ -25,21 +40,6 @@ async def fetch_posts(user):
         print(f" - {post}")
 
 
-async def get_user_with_posts(user_id):
-    user = await fetch_user(user_id)
-    await fetch_posts(user)
-
-
-async def main():
-    user_ids = [1, 2, 3]
-    start = time.perf_counter()
-    await asyncio.gather(
-        *(get_user_with_posts(user_id) for user_id in user_ids)
-    )
-    end = time.perf_counter()
-    print(f"\n==> Total time: {end - start:.2f} seconds")
-
-
 if __name__ == "__main__":
     random.seed(444)
     asyncio.run(main())

python-asyncio/except_group.py

@@ -0,0 +1,35 @@
+import asyncio
+
+
+async def coro_a():
+    await asyncio.sleep(1)
+    raise ValueError("Error in coro A")
+
+
+async def coro_b():
+    await asyncio.sleep(2)
+    raise TypeError("Error in coro B")
+
+
+async def coro_c():
+    await asyncio.sleep(0.5)
+    raise IndexError("Error in coro C")
+
+
+async def main():
+    results = await asyncio.gather(
+        coro_a(), coro_b(), coro_c(), return_exceptions=True
+    )
+    exceptions = [e for e in results if isinstance(e, Exception)]
+    if exceptions:
+        raise ExceptionGroup("Errors", exceptions)
+
+
+try:
+    asyncio.run(main())
+except* ValueError as ve_group:
+    print(f"[ValueError handled] {ve_group.exceptions}")
+except* TypeError as te_group:
+    print(f"[TypeError handled] {te_group.exceptions}")
+except* IndexError as ie_group:
+    print(f"[IndexError handled] {ie_group.exceptions}")

python-asyncio/gathers.py

@@ -8,8 +8,8 @@ async def coro(numbers):
 
 
 async def main():
-    task1 = asyncio.create_task(coro([3, 2, 1]))
-    task2 = asyncio.create_task(coro([10, 5, 2]))
+    task1 = asyncio.create_task(coro([10, 5, 2]))
+    task2 = asyncio.create_task(coro([3, 2, 1]))
     print("Start:", time.strftime("%X"))
     result = await asyncio.gather(task1, task2)
     print("End:", time.strftime("%X"))

python-asyncio/queued.py

@@ -3,6 +3,19 @@
 import time
 
 
+async def main():
+    queue = asyncio.Queue()
+    user_ids = [1, 2, 3]
+
+    start = time.perf_counter()
+    await asyncio.gather(
+        producer(queue, user_ids),
+        *(consumer(queue) for _ in user_ids),
+    )
+    end = time.perf_counter()
+    print(f"\n==> Total time: {end - start:.2f} seconds")
+
+
 async def producer(queue, user_ids):
     async def fetch_user(user_id):
         delay = random.uniform(0.5, 2.0)
@@ -34,19 +47,6 @@ async def consumer(queue):
             print(f"  - {post}")
 
 
-async def main():
-    queue = asyncio.Queue()
-    user_ids = [1, 2, 3]
-
-    start = time.perf_counter()
-    await asyncio.gather(
-        producer(queue, user_ids),
-        *(consumer(queue) for _ in user_ids),
-    )
-    end = time.perf_counter()
-    print(f"\n==> Total time: {end - start:.2f} seconds")
-
-
 if __name__ == "__main__":
     random.seed(444)
     asyncio.run(main())

python-asyncio/rand.py

@@ -9,6 +9,14 @@
 )
 
 
+async def main():
+    return await asyncio.gather(
+        makerandom(1, 9),
+        makerandom(2, 8),
+        makerandom(3, 8),
+    )
+
+
 async def makerandom(delay, threshold=6):
     color = COLORS[delay]
     print(f"{color}Initiated makerandom({delay}).")
@@ -19,15 +27,6 @@ async def makerandom(delay, threshold=6):
     return number
 
 
-async def main():
-    result = await asyncio.gather(
-        makerandom(1, 9),
-        makerandom(2, 8),
-        makerandom(3, 8),
-    )
-    return result
-
-
 if __name__ == "__main__":
     random.seed(444)
     r1, r2, r3 = asyncio.run(main())

python-asyncio/tasks.py

@@ -8,8 +8,8 @@ async def coro(numbers):
 
 async def main():
     task = asyncio.create_task(coro([3, 2, 1]))
-    print(f"task: type -> {type(task)}")
-    print(f"task: done -> {task.done()}")
+    print(f"{type(task) = }")
+    print(f"{task.done() = }")
     return await task