Source code for idmtools.registry.functions

"""
FunctionPluginManager provided definition for the function plugin specification, hooks, and plugin manager.

Copyright 2021, Bill & Melinda Gates Foundation. All rights reserved.
"""
from logging import getLogger, DEBUG

from pluggy import PluginManager
import idmtools.registry.hook_specs as hookspecs
from idmtools.utils.decorators import SingletonMixin

logger = getLogger(__name__)


class FunctionPluginManager(PluginManager, SingletonMixin):
    """
    FunctionPluginManager acts as registry for function based plugins.
    """

    def __init__(self):
        """
        Initialize function plugin manager.
        """
        super(FunctionPluginManager, self).__init__(hookspecs.IDMTOOLS_HOOKS)
        self.add_hookspecs(hookspecs)
        self.load_setuptools_entrypoints(hookspecs.IDMTOOLS_HOOKS.lower())
        if logger.isEnabledFor(DEBUG):
            logger.debug(self.get_plugins())