Function get_all_imports

scripts/update_lib/deps.py:967–985

Get all imports from a library file. Args: name: Module name cpython_prefix: CPython directory prefix Returns: Frozenset of all imported module names

(name: str, cpython_prefix: str)

@functools.cache
def get_all_imports(name: str, cpython_prefix: str) -> frozenset[str]:
    """Get all imports from a library file.

    Args:
        name: Module name
        cpython_prefix: CPython directory prefix

    Returns:
        Frozenset of all imported module names
    """
    all_imports = set()
    for lib_path in get_lib_paths(name, cpython_prefix):
        if lib_path.exists():
            for _, content in read_python_files(lib_path):
                all_imports.update(parse_lib_imports(content))

    # Remove self
    all_imports.discard(name)
    return frozenset(all_imports)


@functools.cache