From cdfb52a569cd4a54046bcc7ad57bee1c7835dcbf Mon Sep 17 00:00:00 2001 From: "Salvador E. Tropea" Date: Fri, 23 Oct 2020 19:30:55 -0300 Subject: [PATCH] Added a fallback for the case that colorama isn't available --- kibot/mcpyrate/colorizer.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/kibot/mcpyrate/colorizer.py b/kibot/mcpyrate/colorizer.py index 433cb278..6efd709f 100644 --- a/kibot/mcpyrate/colorizer.py +++ b/kibot/mcpyrate/colorizer.py @@ -4,10 +4,20 @@ __all__ = ["setcolor", "colorize", "ColorScheme", "Fore", "Back", "Style"] -from colorama import init as colorama_init, Fore, Back, Style +no_colorama = False +try: + from colorama import init as colorama_init, Fore, Back, Style +except ImportError: # pragma: no cover + no_colorama = True +if no_colorama: + class Fore(object): + YELLOW = CYAN = LIGHTBLUE_EX = RESET = GREEN = BLUE = '' -colorama_init() + class Style(object): + BRIGHT = DIM = RESET_ALL = NORMAL = '' +else: + colorama_init() def setcolor(*colors):