add new

wang-engineer · wang-engineer · commit 19e7b2c8eead · 2023-03-11T20:54:57.000-05:00
diff --git a/19-concurrency/spinner_thread.py b/19-concurrency/spinner_thread.py
@@ -12,13 +12,17 @@
 def spin(msg: str, done: Event) -> None:  # <1>
     for char in itertools.cycle(r'\|/-'):  # <2>
         status = f'\r{char} {msg}'  # <3>
-        print(status, end='', flush=True)
-        if done.wait(.1):  # <4>
+        print(status,
+              end='',
+              flush=True
+              )
+        if done.wait(.5):  # <4>
             break  # <5>
     blanks = ' ' * len(status)
     print(f'\r{blanks}\r', end='')  # <6>
 
 def slow() -> int:
+    # print("sleep for 3 seconds")
     time.sleep(3)  # <7>
     return 42
 # end::SPINNER_THREAD_TOP[]
diff --git a/20-executors/demo_executor_map.py b/20-executors/demo_executor_map.py
@@ -20,10 +20,10 @@ def loiter(n):  # <2>
 def main():
     display('Script starting.')
     executor = futures.ThreadPoolExecutor(max_workers=3)  # <4>
-    results = executor.map(loiter, range(5))  # <5>
+    results = executor.map(loiter, range(5, 0, -1))  # <5>
     display('results:', results)  # <6>
     display('Waiting for individual results:')
-    for i, result in enumerate(results):  # <7>
+    for i, result in enumerate(results,1):  # <7>
         display(f'result {i}: {result}')
 
 if __name__ == '__main__':
diff --git a/20-executors/primes/proc_pool1.py b/20-executors/primes/proc_pool1.py
@@ -0,0 +1,48 @@
+
+import sys
+from concurrent import futures
+from time import perf_counter
+from typing import NamedTuple
+
+from primes import is_prime, NUMBERS
+
+
+class PrimeResult(NamedTuple):
+    n: int
+    flag: bool
+    elapsed: float
+
+
+def check(n: int) -> PrimeResult:
+    t0 = perf_counter()
+    res = is_prime(n)
+    return PrimeResult(n, res, perf_counter()-t0)
+
+
+def main() -> None:
+    if len(sys.argv) < 2:
+        workers = None
+    else:
+        workers = int(sys.argv[1])
+
+    executor = futures.ProcessPoolExecutor(max_workers=workers)
+    actual_workers = executor._max_workers
+
+    print(f'Checking{len(NUMBERS)} numbers with {actual_workers} processes:')
+
+    t0 = perf_counter()
+    numbers = sorted(NUMBERS, reverse=True)
+    with executor:
+        for n, prime, elapsed in executor.map(check, numbers):
+            label = 'P' if prime else ' '
+            print(f'{n: 16} {label} {elapsed:9.6f}s')
+    time = perf_counter() - t0
+    print(f'Total time: {time:.2f}s')
+
+
+if __name__ == '__main__':
+    main()
+
+
+
+
diff --git a/21-async/domains/asyncio/blogdom1.py b/21-async/domains/asyncio/blogdom1.py
@@ -0,0 +1,31 @@
+import asyncio
+import socket
+
+from keyword import kwlist
+
+MAX_KEYWORD_LEN = 4
+
+
+async def probe(domain: str) -> tuple[str: bool]:
+    loop = asyncio.get_running_loop()
+    try:
+        await loop.getaddrinfo(domain, None)
+    except socket.gaierror:
+        return (domain, False)
+    return (domain, True)
+
+
+async def main() -> None:
+    names = (kw for kw in kwlist if len(kw) <= MAX_KEYWORD_LEN)
+    domains = (f"{name}.dev".lower() for name in names)
+    coros = [probe(domain) for domain in domains]
+    # for coro in coros:
+    for coro in asyncio.as_completed(coros):
+        domain, found = await coro
+        mark = "+" if found else " "
+        print(f"{mark} {domain}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+
