diff --git a/hooks/pre_gen_project.py b/hooks/pre_gen_project.py index c73f003028ac7c9d11167cea2f7e5b627c88b540..c4b16a3949b4e2318e5b31ea8cf42b8a00daf536 100644 --- a/hooks/pre_gen_project.py +++ b/hooks/pre_gen_project.py @@ -6,6 +6,15 @@ logging.basicConfig() LOGGER = logging.getLogger(__name__) MODULE_REGEX = r"^[_a-zA-Z][_a-zA-Z0-9]+$" +class bcolors: + WARNING = "\033[93m" + ENDC = "\033[0m" + BOLD = "\033[1m" + + +def colorize(escape_code, text): + code = getattr(bcolors, escape_code) + return f"{code}{text}{bcolors.ENDC}" def log_module_name_warning(module_name, logger):