diff --git a/src/music_kraken/cli/main_downloader.py b/src/music_kraken/cli/main_downloader.py index 52d9847..c18aa8f 100644 --- a/src/music_kraken/cli/main_downloader.py +++ b/src/music_kraken/cli/main_downloader.py @@ -11,6 +11,7 @@ from ..utils.string_processing import fit_to_file_system from ..utils.support_classes.query import Query from ..utils.support_classes.download_result import DownloadResult from ..utils.exception.download import UrlNotFoundException +from ..utils.enums.colors import BColors from ..download.results import Results, Option, PageResults from ..download.page_attributes import Pages from ..pages import Page @@ -176,7 +177,7 @@ class Downloader: else: prefix = ALPHABET[page_count % len(ALPHABET)] print( - f"({prefix}) ------------------------{option.__name__:{PAGE_NAME_FILL}<{MAX_PAGE_LEN}}------------") + f"{BColors.HEADER}({prefix}) ------------------------{option.__name__:{PAGE_NAME_FILL}<{MAX_PAGE_LEN}}------------{BColors.ENDC}") self.page_dict[prefix] = option self.page_dict[option.__name__] = option diff --git a/src/music_kraken/utils/enums/colors.py b/src/music_kraken/utils/enums/colors.py new file mode 100644 index 0000000..61fe1e4 --- /dev/null +++ b/src/music_kraken/utils/enums/colors.py @@ -0,0 +1,14 @@ +from enum import Enum + + +class BColors: + # https://stackoverflow.com/a/287944 + HEADER = "\033[95m" + OKBLUE = "\033[94m" + OKCYAN = "\033[96m" + OKGREEN = "\033[92m" + WARNING = "\033[93m" + FAIL = "\033[91m" + ENDC = "\033[0m" + BOLD = "\033[1m" + UNDERLINE = "\033[4m" \ No newline at end of file