from enum import Enum, auto from typing import Optional PYGMENTS_BRIGHT_BLACK = 'ansibrightblack' AUTO_STYLE = 'auto' # Follows terminal ANSI color styles class Styles(Enum): PIE = auto() ANSI = auto() class PieStyle(str, Enum): UNIVERSAL = 'pie' DARK = 'pie-dark' LIGHT = 'pie-light' PIE_STYLE_TO_SHADE = { PieStyle.DARK: '500', PieStyle.UNIVERSAL: '600', PieStyle.LIGHT: '700', } SHADE_TO_PIE_STYLE = { shade: style for style, shade in PIE_STYLE_TO_SHADE.items() } class ColorString(str): def __or__(self, other: str) -> 'ColorString': """Combine a style with a property. E.g: PieColor.BLUE | BOLD | ITALIC """ return ColorString(self + ' ' + other) class PieColor(ColorString, Enum): """Styles that are available only in Pie themes.""" PRIMARY = 'primary' SECONDARY = 'secondary' WHITE = 'white' BLACK = 'black' GREY = 'grey' AQUA = 'aqua' PURPLE = 'purple' ORANGE = 'orange' RED = 'red' BLUE = 'blue' PINK = 'pink' GREEN = 'green' YELLOW = 'yellow' class GenericColor(Enum): """Generic colors that are safe to use everywhere.""" # WHITE = {Styles.PIE: PieColor.WHITE, Styles.ANSI: 'white'} BLACK = {Styles.PIE: PieColor.BLACK, Styles.ANSI: 'black'} GREEN = {Styles.PIE: PieColor.GREEN, Styles.ANSI: 'green'} ORANGE = {Styles.PIE: PieColor.ORANGE, Styles.ANSI: 'yellow'} YELLOW = {Styles.PIE: PieColor.YELLOW, Styles.ANSI: 'bright_yellow'} BLUE = {Styles.PIE: PieColor.BLUE, Styles.ANSI: 'blue'} PINK = {Styles.PIE: PieColor.PINK, Styles.ANSI: 'bright_magenta'} PURPLE = {Styles.PIE: PieColor.PURPLE, Styles.ANSI: 'magenta'} RED = {Styles.PIE: PieColor.RED, Styles.ANSI: 'red'} AQUA = {Styles.PIE: PieColor.AQUA, Styles.ANSI: 'cyan'} GREY = {Styles.PIE: PieColor.GREY, Styles.ANSI: 'bright_black'} def apply_style( self, style: Styles, *, style_name: Optional[str] = None ) -> str: """Apply the given style to a particular value.""" exposed_color = self.value[style] if style is Styles.PIE: assert style_name is not None shade = PIE_STYLE_TO_SHADE[PieStyle(style_name)] return get_color(exposed_color, shade) else: return exposed_color # noinspection PyDictCreation COLOR_PALETTE = { # Copy the brand palette PieColor.WHITE: '#F5F5F0', PieColor.BLACK: '#1C1818', PieColor.GREY: { '50': '#F5F5F0', '100': '#EDEDEB', '200': '#D1D1CF', '300': '#B5B5B2', '400': '#999999', '500': '#7D7D7D', '600': '#666663', '700': '#4F4D4D', '800': '#363636', '900': '#1C1818', 'DEFAULT': '#7D7D7D', }, PieColor.AQUA: { '50': '#E8F0F5', '100': '#D6E3ED', '200': '#C4D9E5', '300': '#B0CCDE', '400': '#9EBFD6', '500': '#8CB4CD', '600': '#7A9EB5', '700': '#698799', '800': '#597082', '900': '#455966', 'DEFAULT': '#8CB4CD', }, PieColor.PURPLE: { '50': '#F0E0FC', '100': '#E3C7FA', '200': '#D9ADF7', '300': '#CC96F5', '400': '#BF7DF2', '500': '#B464F0', '600': '#9E54D6', '700': '#8745BA', '800': '#70389E', '900': '#5C2982', 'DEFAULT': '#B464F0', }, PieColor.ORANGE: { '50': '#FFEDDB', '100': '#FFDEBF', '200': '#FFCFA3', '300': '#FFBF87', '400': '#FFB06B', '500': '#FFA24E', '600': '#F2913D', '700': '#E3822B', '800': '#D6701C', '900': '#C75E0A', 'DEFAULT': '#FFA24E', }, PieColor.RED: { '50': '#FFE0DE', '100': '#FFC7C4', '200': '#FFB0AB', '300': '#FF968F', '400': '#FF8075', '500': '#FF665B', '600': '#E34F45', '700': '#C7382E', '800': '#AD2117', '900': '#910A00', 'DEFAULT': '#FF665B', }, PieColor.BLUE: { '50': '#DBE3FA', '100': '#BFCFF5', '200': '#A1B8F2', '300': '#85A3ED', '400': '#698FEB', '500': '#4B78E6', '600': '#426BD1', '700': '#3B5EBA', '800': '#3354A6', '900': '#2B478F', 'DEFAULT': '#4B78E6', }, PieColor.PINK: { '50': '#FFEBFF', '100': '#FCDBFC', '200': '#FCCCFC', '300': '#FCBAFC', '400': '#FAABFA', '500': '#FA9BFA', '600': '#DE85DE', '700': '#C26EC2', '800': '#A854A6', '900': '#8C3D8A', 'DEFAULT': '#FA9BFA', }, PieColor.GREEN: { '50': '#E3F7E8', '100': '#CCF2D6', '200': '#B5EDC4', '300': '#A1E8B0', '400': '#8AE09E', '500': '#73DC8C', '600': '#63C27A', '700': '#52AB66', '800': '#429154', '900': '#307842', 'DEFAULT': '#73DC8C', }, PieColor.YELLOW: { '50': '#F7F7DB', '100': '#F2F2BF', '200': '#EDEDA6', '300': '#E5E88A', '400': '#E0E36E', '500': '#DBDE52', '600': '#CCCC3D', '700': '#BABA29', '800': '#ABA614', '900': '#999400', 'DEFAULT': '#DBDE52', }, } COLOR_PALETTE.update( { # Terminal-specific palette customizations. PieColor.GREY: { # Grey is the same no matter shade for the colors shade: COLOR_PALETTE[PieColor.GREY]['500'] for shade in COLOR_PALETTE[PieColor.GREY].keys() }, PieColor.PRIMARY: { '700': COLOR_PALETTE[PieColor.BLACK], '600': PYGMENTS_BRIGHT_BLACK, '500': COLOR_PALETTE[PieColor.WHITE], }, PieColor.SECONDARY: { '700': '#37523C', '600': '#6c6969', '500': '#6c6969', }, } ) def boldify(color: PieColor) -> str: return f'bold {color}' # noinspection PyDefaultArgument def get_color( color: PieColor, shade: str, *, palette=COLOR_PALETTE ) -> Optional[str]: if color not in palette: return None color_code = palette[color] if isinstance(color_code, dict) and shade in color_code: return color_code[shade] else: return color_code