SpotifyCodeCollageMaker/color_factory.py at main · wiskiw/SpotifyCodeCollageMaker · GitHub

1
2
3
4
5
6
7
8
9
10
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
import random

from spotify_utils import SpotifyCodeColor, SpotifyBarColor, DefaultSpotifyBackgroundColor


class ColorFactory:

    def get_color(self, code_index: int, codes_count: int) -> SpotifyCodeColor:
        pass


class SpotifyRandomColorFactory(ColorFactory):
    _bar_color: SpotifyBarColor

    def __init__(self, bar_color: SpotifyBarColor):
        self._bar_color = bar_color

    def get_color(self, code_index: int, codes_count: int) -> SpotifyCodeColor:
        bg_color = random.choice(list(DefaultSpotifyBackgroundColor))

        return SpotifyCodeColor(bar=self._bar_color, background_hex=bg_color.value)


class SpotifyOrderedColorFactory(ColorFactory):
    _bar_color: SpotifyBarColor

    def __init__(self, bar_color: SpotifyBarColor):
        self._bar_color = bar_color

    def get_color(self, code_index: int, codes_count: int) -> SpotifyCodeColor:
        default_colors_count = len(list(DefaultSpotifyBackgroundColor))

        color_index = code_index % default_colors_count
        bg_color = list(DefaultSpotifyBackgroundColor)[color_index]

        return SpotifyCodeColor(bar=self._bar_color, background_hex=bg_color.value)


class SingleColorFactory(ColorFactory):
    _code_color: SpotifyCodeColor

    def __init__(self, code_color: SpotifyCodeColor):
        self._code_color = code_color

    def get_color(self, code_index: int, codes_count: int) -> SpotifyCodeColor:
        return SpotifyCodeColor(bar=self._code_color.bar, background_hex=self._code_color.background_hex)