def _plot_with_legend()

in chartify/_core/plot.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

    def _plot_with_legend(method, **kwargs):
        """Call plotting method with the associated kwargs.

        Removes the legend parameter if it is set to None because
        Bokeh breaks if None is passed to a legend parameter

        """
        legend_label = kwargs.pop("legend_label", None)
        legend_group = kwargs.pop("legend_group", None)

        if legend_label is not None:
            return method(**kwargs, legend_label=legend_label)
        elif legend_group is not None:
            return method(**kwargs, legend_group=legend_group)
        else:
            return method(**kwargs)