renamed 22-async/

22-async/domains/curio/domainlib.py

@@ -0,0 +1,25 @@
+from curio import TaskGroup
+from curio.socket import getaddrinfo, gaierror
+from collections.abc import Iterable, AsyncIterator
+from typing import NamedTuple
+
+
+class Result(NamedTuple):
+    domain: str
+    found: bool
+
+
+async def probe(domain: str) -> Result:
+    try:
+        await getaddrinfo(domain, None)
+    except gaierror:
+        return Result(domain, False)
+    return Result(domain, True)
+
+
+async def multi_probe(domains: Iterable[str]) -> AsyncIterator[Result]:
+    async with TaskGroup() as group:
+        for domain in domains:
+            await group.spawn(probe, domain)
+        async for task in group:
+            yield task.result