Source code for ochanticipy.utils.check_extra_imports

"""Check imports that are in extras_require."""
import importlib.util




def check_extra_imports(libraries: list, subpackage: str):
    """Check that libraries are installed and available.

    Parameters
    ----------
    libraries : str
        List of libraries to check.
    subpackage : str
        String of subpackage defined for extra_requires
        that import should warn to install from.
    """
    missing_pkgs = []
    for library in libraries:
        if importlib.util.find_spec(library) is None:
            missing_pkgs += [library]

    if missing_pkgs:
        raise ModuleNotFoundError(
            f"{', '.join(missing_pkgs)} were not found. Install the "
            f"missing packages with `pip install ochanticipy[{subpackage}]`."
        )