Source code for pyfstat.utils.importing

import inspect
import logging
import os
from functools import wraps

logger = logging.getLogger(__name__)


def initializer(func):
    """Decorator to automatically assign the parameters of a class instantiation to self."""
    argspec = inspect.getfullargspec(func)

    @wraps(func)
    def wrapper(self, *args, **kargs):
        for name, arg in list(zip(argspec.args[1:], args)) + list(kargs.items()):
            setattr(self, name, arg)

        for name, default in zip(reversed(argspec.args), reversed(argspec.defaults)):
            if not hasattr(self, name):
                setattr(self, name, default)

        func(self, *args, **kargs)

    return wrapper


def safe_X_less_plt():

    if "DISPLAY" in os.environ:
        import matplotlib.pyplot as plt

        logger.debug("$DISPLAY environment variable found," " using it for matplotlib.")
    else:
        logger.info(
            "No $DISPLAY environment variable found, so importing"
            " matplotlib.pyplot with non-interactive 'Agg' backend."
        )
        import matplotlib

        matplotlib.use("agg")
        import matplotlib.pyplot as plt

    plt.rcParams["axes.formatter.useoffset"] = False
    return plt