def orbit_style_decorator()

in orbit/utils/plot.py [0:0]

• 23 lines of code
• 6 McCabe index (conditional complexity)

def orbit_style_decorator(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        # check if arg is specified; if not, set True as default
        if "use_orbit_style" in kwargs.keys():
            use_orbit_style = kwargs["use_orbit_style"]
            del kwargs["use_orbit_style"]
        else:
            use_orbit_style = True

        # use orbit style plot if it is set to be True
        if use_orbit_style:
            orbit_style_path = get_orbit_style()
            try:
                with plt.style.context(orbit_style_path):
                    return func(*args, **kwargs)
            except:
                logging.info(
                    "Cannot find path:{}. Use default plot style.".format(
                        orbit_style_path
                    )
                )
                return func(*args, **kwargs)
        else:
            return func(*args, **kwargs)

    return wrapper