From 224750c0d65a45bc3fd7fbb4c0179701c241977c Mon Sep 17 00:00:00 2001 From: Ryan Petrello Date: Thu, 15 Aug 2019 09:41:44 -0400 Subject: [PATCH] replace the termcolor dependency w/ a simple function --- awxkit/awxkit/cli/utils.py | 37 ++++++++++++++++++++++++++----------- awxkit/requirements.txt | 1 - 2 files changed, 26 insertions(+), 12 deletions(-) diff --git a/awxkit/awxkit/cli/utils.py b/awxkit/awxkit/cli/utils.py index 71513210bc..1ea7fcff9d 100644 --- a/awxkit/awxkit/cli/utils.py +++ b/awxkit/awxkit/cli/utils.py @@ -1,8 +1,7 @@ from argparse import ArgumentParser -import threading +import os import sys - -import termcolor +import threading _color = threading.local() _color.enabled = True @@ -62,15 +61,31 @@ def disable_color(): _color.enabled = False -def colored(value, color): - if _color.enabled: - return termcolor.colored(value, color) - else: - return value +COLORS = dict( + list( + zip( + [ + 'grey', 'red', 'green', 'yellow', 'blue', 'magenta', 'cyan', + 'white', + ], + list(range(30, 38)) + ) + ) +) -def cprint(value, color, **kwargs): +def colored(text, color=None): + '''Colorize text w/ ANSI color sequences''' + if _color.enabled and os.getenv('ANSI_COLORS_DISABLED') is None: + fmt_str = '\033[%dm%s' + if color is not None: + text = fmt_str % (COLORS[color], text) + text += '\033[0m' + return text + + +def cprint(text, color, **kwargs): if _color.enabled: - termcolor.cprint(value, color, **kwargs) + print(colored(text, color), **kwargs) else: - print(value, **kwargs) + print(text, **kwargs) diff --git a/awxkit/requirements.txt b/awxkit/requirements.txt index b662004a18..a73058e771 100644 --- a/awxkit/requirements.txt +++ b/awxkit/requirements.txt @@ -1,4 +1,3 @@ PyYAML>=5.1 python-dateutil requests -termcolor