Source code for bokeh.util.dependencies
#-----------------------------------------------------------------------------
# Copyright (c) Anaconda, Inc., and Bokeh Contributors.
# All rights reserved.
#
# The full license is in the file LICENSE.txt, distributed with this software.
#-----------------------------------------------------------------------------
''' Utilities for checking dependencies
'''
#-----------------------------------------------------------------------------
# Boilerplate
#-----------------------------------------------------------------------------
from __future__ import annotations
import logging # isort:skip
log = logging.getLogger(__name__)
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
# Standard library imports
from importlib import import_module
from types import ModuleType
from typing import Any
#-----------------------------------------------------------------------------
# Globals and constants
#-----------------------------------------------------------------------------
__all__ = (
'import_optional',
'import_required',
'uses_pandas',
)
#-----------------------------------------------------------------------------
# General API
#-----------------------------------------------------------------------------
[docs]
def import_optional(mod_name: str) -> ModuleType | None:
''' Attempt to import an optional dependency.
Silently returns None if the requested module is not available.
Args:
mod_name (str) : name of the optional module to try to import
Returns:
imported module or None, if import fails
'''
try:
return import_module(mod_name)
except ImportError:
pass
except Exception:
msg = f"Failed to import optional module `{mod_name}`"
log.exception(msg)
return None
[docs]
def import_required(mod_name: str, error_msg: str) -> ModuleType:
''' Attempt to import a required dependency.
Raises a RuntimeError if the requested module is not available.
Args:
mod_name (str) : name of the required module to try to import
error_msg (str) : error message to raise when the module is missing
Returns:
imported module
Raises:
RuntimeError
'''
try:
return import_module(mod_name)
except ImportError as e:
raise RuntimeError(error_msg) from e
[docs]
def uses_pandas(obj: Any) -> bool:
"""
Checks if an object is a ``pandas`` object.
Use this before conditional ``import pandas as pd``.
"""
module = type(obj).__module__
return module is not None and module.startswith("pandas.")
#-----------------------------------------------------------------------------
# Dev API
#-----------------------------------------------------------------------------
#-----------------------------------------------------------------------------
# Private API
#-----------------------------------------------------------------------------
#-----------------------------------------------------------------------------
# Code
#-----------------------------------------------------------------------------