def _get_palette_class()

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

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

    def _get_palette_class(cls, chart, palette_type="categorical", palette=None, accent_values=None):
        if palette_type == "categorical":
            if palette is None:
                palette_name = options.get_option("style.color_palette_categorical")
                palette = colors.color_palettes[palette_name]
            return CategoricalPalette(chart, palette)
        elif palette_type == "sequential":
            if palette is None:
                palette_name = options.get_option("style.color_palette_sequential")
                palette = colors.color_palettes[palette_name]
            return OrdinalPalette(chart, palette)
        elif palette_type == "diverging":
            if palette is None:
                palette_name = options.get_option("style.color_palette_diverging")
                palette = colors.color_palettes[palette_name]
            return OrdinalPalette(chart, palette)
        elif palette_type == "accent":
            if palette is None:
                palette_name = options.get_option("style.color_palette_accent")
                palette = colors.color_palettes[palette_name]
            return AccentPalette(chart, palette, accent_values)
        else:
            raise ValueError(
                """Type must be one of: ('categorical', 'sequential',
                                         'diverging', 'accent')."""
            )