Source code for niftynet.utilities.util_import

# -*- coding: utf-8 -*-
""" check module to be imported"""
import importlib

import tensorflow as tf


def check_module(name, min_version=None):
    """
    Check if the module exists, and
    satisfies the minimum version requirement.

    Raises ImportError and AssertionError.

    :param name:
    :param min_version:
    :return:
    """

    name = '{}'.format(name)
    try:
        the_module = importlib.import_module(name)
    except ImportError:
        tf.logging.info(
            'Optional Python module %s not found, '
            'please install %s and retry if the application fails.',
            name, name)
        raise

    try:
        if min_version is not None:
            assert the_module.__version__ >= '{}'.format(min_version)
    except AttributeError:
        pass
    except AssertionError:
        tf.logging.info(
            'Optional Python module %s version %s not found, '
            'please install %s-%s and retry if the application fails.',
            name, min_version, name, min_version)
        raise