color_picker.py

class ColorPicker:

    def __init__(self, channels: int = 4):
        self._color_by_id = {}
        self._channels = channels

    def get(self, idx):
        if idx not in self._color_by_id:
            self._color_by_id[idx] = self._generate_color(channels=self._channels)
        return self._color_by_id[idx]

    @staticmethod
    def _generate_color(low: int = 0, high: int = 255, channels: int = 4):
        if channels == 4:
            return (random.randint(low, high), random.randint(low, high), random.randint(low, high), 255)
        return (random.randint(low, high), random.randint(low, high), random.randint(low, high))