Source code for jacinle.utils.vendor

#! /usr/bin/env python3
# -*- coding: utf-8 -*-
# File   : vendor.py
# Author : Jiayuan Mao
# Email  : maojiayuan@gmail.com
# Date   : 05/25/2019
#
# This file is part of Jacinle.
# Distributed under terms of the MIT license.

import importlib
import functools
from jacinle.logging import get_logger

logger = get_logger(__file__)

__all__ = ['has_vendor', 'requires_vendors']




def has_vendor(vendor):
    """Check if the module is available."""
    try:
        importlib.import_module(vendor)
    except ImportError as e:
        return False
    return True





def requires_vendors(*vendors):
    """Decorator to check if the module is available. If not, raises an error when the function is called."""
    def wrapper(func):
        checked_vendors = False

        @functools.wraps(func)
        def wrapped(*args, **kwargs):
            nonlocal checked_vendors
            if not checked_vendors:
                for v in vendors:
                    try:
                        importlib.import_module(v)
                    except ImportError as e:
                        raise ImportError('Cannot import {}. Make sure you have it as a vendor for Jacinle.'.format(v)) from e
                checked_vendors = True

            return func(*args, **kwargs)

        return wrapped
    return wrapper