checkenv.py

# Check that the packages are installed.
import os
import sys
from pkgutil import iter_modules


def check_import(packagename):
    if packagename in (name for _, name, _ in iter_modules()):
        return True
    else:
        return False


# If there are new packages that you can import, add them to the list.
package_names = [
    "networkx",
    "numpy",
    "matplotlib",
    "hiveplot",
    "pandas",
    "jupyter",
    "nxviz",
    "tqdm",
]
packages = {n: n for n in package_names}
# Only add the packages whose import names are different from the
# package name (what we `pip install` or `conda install`).
packages["community"] = "python-louvain"


assert (
    sys.version_info.major >= 3 and sys.version_info.minor >= 6
), "Please install Python 3.6!"


def print_error(p, i):
    """
    Returns the error message for package installation issues.

    :param str p: The name of the package to install.
    :param str i: The name of the package when imported.
    """
    error_message = f"""
    {i} not present. Please do the installation using either:

    - pip install {p}
    - conda install -c conda-forge {p}
    """
    return error_message


for p, i in packages.items():
    assert check_import(p), print_error(i, p)


# os.system returns 0 if command passed
assert not os.system("command -v ffmpeg"), "please install ffmpeg"


# Credit: @zmilicc for requesting this.
print("All checks passed. Your environment is good to go!")