check_install.py

import importlib
import sys
from warnings import warn

onpy2 = False

try:
    assert sys.version_info >= (3,0)
    import importlib.util
except AssertionError:
    warn('This tutorial is written for Python 3.  Legacy Python is not explicitly supported.')
    onpy2 = True

def tuple_version(version):
    return tuple(int(x) for x in version.strip('<>+-=.').split('.'))

def check_versions():
    version_trouble=False
    numba = importlib.import_module('numba')
    numba_version = tuple_version(numba.__version__)
    if numba_version < (0, 26, 0):
        print('Please update Numba to version 0.26.0')
        version_trouble=True

    mpl = importlib.import_module('matplotlib')
    mpl_version = tuple_version(mpl.__version__)
    if mpl_version < (1, 5, 0):
        print('Please update matplotlib to version 1.5.0 or higher')
        version_trouble=True

    return version_trouble

def main():
    required_modules = ['numpy', 'matplotlib', 'jupyter',
                        'numba', 'llvmlite', 'line_profiler', 'IPython',]
    missing_modules = []
    for mod in required_modules:
        if not onpy2:
            spec = importlib.util.find_spec(mod)
            if spec is None:
                missing_modules.append(mod)
        else:
            try:
                importlib.import_module(mod)
            except ImportError:
                missing_modules.append(mod)

    if missing_modules:
        print('The following modules are required but not installed:')
        print('    {}'.format(', '.join(missing_modules)))
        print('\nYou can install them using conda by running:')
        print('\n    conda install {}'.format(' '.join(missing_modules)))
        print('\nOr you can install them using pip by running:')
        print('\n    pip install {}'.format(' '.join(missing_modules)))
    else:
        if check_versions():
            print('All packages are installed but at least one needs updating')
        else:
            print('Everything looks good!')

if __name__ == '__main__':
    main()