Skip to content

Color Scheme

src.drawpyo.utils.color_scheme.ColorScheme

Represents a set of colors used for an object's fill, stroke and font.

A color can be

• None • A hex string • A DefaultColor

Source code in src/drawpyo/utils/color_scheme.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
class ColorScheme:
    """
    Represents a set of colors used for an object's fill, stroke and font.

    A color can be:
        • None
        • A hex string
        • A DefaultColor
    """

    HEX_PATTERN = re.compile(r"^#[0-9A-Fa-f]{6}$")

    ###########################################################
    # Initialization
    ###########################################################

    def __init__(
        self,
        fill_color: ColorType = None,
        stroke_color: ColorType = None,
        font_color: ColorType = None,
    ) -> None:
        self.fill_color: ColorType = self._validated(fill_color)
        self.stroke_color: ColorType = self._validated(stroke_color)
        self.font_color: ColorType = self._validated(font_color)
        logger.info(f"🎨 ColorScheme created: {self.__repr__()}")

    ###########################################################
    # Public Setters
    ###########################################################

    def set_fill_color(self, color: ColorType) -> None:
        self.fill_color = self._validated(color)

    def set_stroke_color(self, color: ColorType) -> None:
        self.stroke_color = self._validated(color)

    def set_font_color(self, color: ColorType) -> None:
        self.font_color = self._validated(color)

    ###########################################################
    # Validation
    ###########################################################

    def _validated(self, color: ColorType) -> ColorType:
        """Validate hex strings or DefaultColor enums."""
        if color is None:
            return None

        if isinstance(color, StandardColor):
            return color.value

        if isinstance(color, str):
            if not self.is_valid_hex(color):
                raise ValueError(
                    f"Invalid color '{color}'. "
                    f"Expected '#RRGGBB' (example: #A1B2C3)."
                )
            return color.upper()

        raise TypeError(
            f"Color must be a hex string like '#AABBCC', None, or a DefaultColor enum. "
            f"Received: {type(color)}"
        )

    @classmethod
    def is_valid_hex(cls, value: str) -> bool:
        """Return True if string is a valid #RRGGBB hex color."""
        return bool(cls.HEX_PATTERN.match(value))

    ###########################################################
    # Utility
    ###########################################################

    def __repr__(self) -> str:
        return (
            f"fill: {self.fill_color} "
            f"| stroke: {self.stroke_color} "
            f"| font: {self.font_color}"
        )

is_valid_hex(value) classmethod

Return True if string is a valid #RRGGBB hex color.

Source code in src/drawpyo/utils/color_scheme.py 
76
77
78
79
@classmethod
def is_valid_hex(cls, value: str) -> bool:
    """Return True if string is a valid #RRGGBB hex color."""
    return bool(cls.HEX_PATTERN.match(value))