Function collect_all_stubs

scripts/gen-py-stdlib.py:201–249

Gather every stub in the allowlist plus any re-export targets they pull in transitively (e.g. os.path -> posixpath -> genericpath). Returns (modules, star_imports_per_module).

(stdlib_root: Path)

def collect_all_stubs(stdlib_root: Path) -> tuple[dict[str, ModuleStubs], dict[str, list[StarReExport]]]:
    """Gather every stub in the allowlist plus any re-export targets they
    pull in transitively (e.g. os.path -> posixpath -> genericpath).
    Returns (modules, star_imports_per_module). """
    modules: dict[str, ModuleStubs] = {}
    star_imports: dict[str, list[StarReExport]] = {}

    # First pass: walk allowlist
    queue: list[tuple[Path, str]] = []
    for path in sorted(stdlib_root.rglob("*.pyi")):
        rel = path.relative_to(stdlib_root)
        parts = list(rel.parts)
        if parts[-1] == "__init__.pyi":
            parts.pop()
        else:
            parts[-1] = parts[-1][:-len(".pyi")]
        if not parts:
            continue
        top = parts[0]
        if top.startswith("_"):
            continue
        if top not in ALLOWED_MODULES:
            continue
        module_qn = ".".join(parts)
        queue.append((path, module_qn))

    seen_targets: set[str] = set()
    while queue:
        path, mod_qn = queue.pop()
        if mod_qn in modules:
            continue
        ms, stars = parse_module(path, mod_qn)
        modules[mod_qn] = ms
        star_imports[mod_qn] = stars
        # Enqueue re-export targets (e.g. posixpath, ntpath, genericpath)
        for star in stars:
            tgt = star.target_module
            if tgt in seen_targets:
                continue
            seen_targets.add(tgt)
            tgt_path = stdlib_root / (tgt.replace(".", "/") + ".pyi")
            if tgt_path.is_file():
                queue.append((tgt_path, tgt))
            else:
                tgt_init = stdlib_root / tgt.replace(".", "/") / "__init__.pyi"
                if tgt_init.is_file():
                    queue.append((tgt_init, tgt))

    return modules, star_imports


def resolve_reexports(modules: dict[str, ModuleStubs],