| Dir : /proc/self/root/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/ |
| Server: Linux ngx353.inmotionhosting.com 4.18.0-553.22.1.lve.1.el8.x86_64 #1 SMP Tue Oct 8 15:52:54 UTC 2024 x86_64 IP: 209.182.202.254 |
| Dir : //proc/self/root/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/color.py |
"""
Functions used for CLI color themes.
"""
import logging
import os
from salt.utils.textformat import TextFormat
log = logging.getLogger(__name__)
def get_color_theme(theme):
"""
Return the color theme to use
"""
# Keep the heavy lifting out of the module space
import salt.utils.data
import salt.utils.files
import salt.utils.yaml
if not os.path.isfile(theme):
log.warning("The named theme %s if not available", theme)
try:
with salt.utils.files.fopen(theme, "rb") as fp_:
colors = salt.utils.data.decode(salt.utils.yaml.safe_load(fp_))
ret = {}
for color in colors:
ret[color] = f"\033[{colors[color]}m"
if not isinstance(colors, dict):
log.warning("The theme file %s is not a dict", theme)
return {}
return ret
except Exception: # pylint: disable=broad-except
log.warning("Failed to read the color theme %s", theme)
return {}
def get_colors(use=True, theme=None):
"""
Return the colors as an easy to use dict. Pass `False` to deactivate all
colors by setting them to empty strings. Pass a string containing only the
name of a single color to be used in place of all colors. Examples:
.. code-block:: python
colors = get_colors() # enable all colors
no_colors = get_colors(False) # disable all colors
red_colors = get_colors('RED') # set all colors to red
"""
colors = {
"BLACK": TextFormat("black"),
"DARK_GRAY": TextFormat("bold", "black"),
"RED": TextFormat("red"),
"LIGHT_RED": TextFormat("bold", "red"),
"GREEN": TextFormat("green"),
"LIGHT_GREEN": TextFormat("bold", "green"),
"YELLOW": TextFormat("yellow"),
"LIGHT_YELLOW": TextFormat("bold", "yellow"),
"BLUE": TextFormat("blue"),
"LIGHT_BLUE": TextFormat("bold", "blue"),
"MAGENTA": TextFormat("magenta"),
"LIGHT_MAGENTA": TextFormat("bold", "magenta"),
"CYAN": TextFormat("cyan"),
"LIGHT_CYAN": TextFormat("bold", "cyan"),
"LIGHT_GRAY": TextFormat("white"),
"WHITE": TextFormat("bold", "white"),
"DEFAULT_COLOR": TextFormat("default"),
"ENDC": TextFormat("reset"),
}
if theme:
colors.update(get_color_theme(theme))
if not use:
for color in colors:
colors[color] = ""
if isinstance(use, str):
# Try to set all of the colors to the passed color
if use in colors:
for color in colors:
# except for color reset
if color == "ENDC":
continue
colors[color] = colors[use]
return colors