-
Notifications
You must be signed in to change notification settings - Fork 0
/
Colorizer.py
49 lines (38 loc) · 1.24 KB
/
Colorizer.py
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
from colorama import Fore, Style, init
class ColorizeType:
INFO = 'info'
WARNING = 'warning'
ERROR = 'error'
SUCCESS = 'success'
HIGHLIGHT = 'highlight'
class Colorizer:
"""
Colorize text output with colorama
"""
init()
INFO = Fore.BLUE
WARNING = Fore.YELLOW
ERROR = Fore.RED
SUCCESS = Fore.GREEN
HIGHLIGHT = Fore.CYAN
@staticmethod
def colorize(text, color_type):
# Отримуємо відповідний код кольору
color_code = getattr(Colorizer, color_type.upper(), Fore.RESET)
# Повертаємо текст з встановленим кольором та скиданням стилю
return f"{color_code}{text}{Style.RESET_ALL}"
@staticmethod
def info(text):
return Colorizer.colorize(text, ColorizeType.INFO)
@staticmethod
def warn(text):
return Colorizer.colorize(text, ColorizeType.WARNING)
@staticmethod
def error(text):
return Colorizer.colorize(text, ColorizeType.ERROR)
@staticmethod
def success(text):
return Colorizer.colorize(text, ColorizeType.SUCCESS)
@staticmethod
def highlight(text):
return Colorizer.colorize(text, ColorizeType.HIGHLIGHT)