aboutsummaryrefslogtreecommitdiff
path: root/color.py
blob: ce1b1829a6bec77030bc64784b2690701ae22931 (plain)
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
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
91
92
from typing import Literal

class ColorCode:
    """A superclass to signal that color codes are strings"""

    BLACK: Literal[str]
    RED: Literal[str]
    GREEN: Literal[str]
    YELLOW: Literal[str]
    BLUE: Literal[str]
    MAGENTA: Literal[str]
    CYAN: Literal[str]
    WHITE: Literal[str]
    RESET: Literal[str]

    BLACK_BG: Literal[str]
    RED_BG: Literal[str]
    GREEN_BG: Literal[str]
    YELLOW_BG: Literal[str]
    BLUE_BG: Literal[str]
    MAGENTA_BG: Literal[str]
    CYAN_BG: Literal[str]
    WHITE_BG: Literal[str]

    BOLD: Literal[str]
    UNDERLINE: Literal[str]
    BLINK: Literal[str]


class VT100ColorCode(ColorCode):
    """A class containing VT100 color codes"""

    # Define the color codes
    BLACK = '\033[30m'
    RED = '\033[31m'
    GREEN = '\033[32m'
    YELLOW = '\033[33m'
    BLUE = '\033[34m'
    MAGENTA = '\033[35m'
    CYAN = '\033[36m'
    WHITE = '\033[37m'
    RESET = '\033[0m'

    # Define the background color codes
    BLACK_BG = '\033[40m'
    RED_BG = '\033[41m'
    GREEN_BG = '\033[42m'
    YELLOW_BG = '\033[43m'
    BLUE_BG = '\033[44m'
    MAGENTA_BG = '\033[45m'
    CYAN_BG = '\033[46m'
    WHITE_BG = '\033[47m'

    # Define the bold, underline and blink codes
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'
    BLINK = '\033[5m'

class NoColorColorCode(ColorCode):
    """A class nullifying color codes to disable color"""

    # Define the color codes
    BLACK = ''
    RED = ''
    GREEN = ''
    YELLOW = ''
    BLUE = ''
    MAGENTA = ''
    CYAN = ''
    WHITE = ''
    RESET = ''

    # Define the background color codes
    BLACK_BG = ''
    RED_BG = ''
    GREEN_BG = ''
    YELLOW_BG = ''
    BLUE_BG = ''
    MAGENTA_BG = ''
    CYAN_BG = ''
    WHITE_BG = ''

    # Define the bold, underline and blink codes
    BOLD = ''
    UNDERLINE = ''
    BLINK = ''

def get_color_codes(no_color=False) -> ColorCode:
    if no_color:
        return NoColorColorCode
    else:
        return VT100ColorCode