From 6bb3e54d3983bb1ec3b3c97693caf3b4b6a7bebc Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Wed, 12 Jul 2023 22:44:51 +0200 Subject: [PATCH] added functions that wrap their argument string in a color/reset pair --- slic/utils/cprint.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/slic/utils/cprint.py b/slic/utils/cprint.py index 35a06171..aa08a86f 100644 --- a/slic/utils/cprint.py +++ b/slic/utils/cprint.py @@ -40,3 +40,17 @@ def _print(color, text, sep, kwargs): +# the following creates functions from the COLORS dict which wrap their argument string in a color/reset pair + +def _mk_color_func(color): + def func(text): + return color + text + Fore.RESET + return func + +for name, color in COLORS.items(): + if name is None: + continue + locals()[name] = _mk_color_func(color) + + +