mirror of
https://github.com/checktheroads/hyperglass
synced 2024-05-11 05:55:08 +00:00
132 lines
3.4 KiB
Python
132 lines
3.4 KiB
Python
"""Helper functions for CLI message printing."""
|
|
# Standard Library
|
|
import re
|
|
|
|
# Third Party
|
|
from click import echo, style
|
|
|
|
# Project
|
|
from hyperglass.cli.static import CMD_HELP, Message
|
|
from hyperglass.cli.exceptions import CliError
|
|
|
|
|
|
def cmd_help(emoji="", help_text="", supports_color=False):
|
|
"""Print formatted command help."""
|
|
if supports_color:
|
|
help_str = emoji + style(help_text, **CMD_HELP)
|
|
else:
|
|
help_str = help_text
|
|
return help_str
|
|
|
|
|
|
def _base_formatter(state, text, callback, **kwargs):
|
|
"""Format text block, replace template strings with keyword arguments.
|
|
|
|
Arguments:
|
|
state {dict} -- Text format attributes
|
|
label {dict} -- Keyword format attributes
|
|
text {[type]} -- Text to format
|
|
callback {function} -- Callback function
|
|
|
|
Returns:
|
|
{str|ClickException} -- Formatted output
|
|
"""
|
|
fmt = Message(state)
|
|
|
|
if callback is None:
|
|
callback = style
|
|
|
|
for k, v in kwargs.items():
|
|
if not isinstance(v, str):
|
|
v = str(v)
|
|
kwargs[k] = style(v, **fmt.kw)
|
|
|
|
text_all = re.split(r"(\{\w+\})", text)
|
|
text_all = [style(i, **fmt.msg) for i in text_all]
|
|
text_all = [i.format(**kwargs) for i in text_all]
|
|
|
|
if fmt.emoji:
|
|
text_all.insert(0, fmt.emoji)
|
|
|
|
text_fmt = "".join(text_all)
|
|
|
|
return callback(text_fmt)
|
|
|
|
|
|
def info(text, callback=echo, **kwargs):
|
|
"""Generate formatted informational text.
|
|
|
|
Arguments:
|
|
text {str} -- Text to format
|
|
callback {callable} -- Callback function (default: {echo})
|
|
|
|
Returns:
|
|
{str} -- Informational output
|
|
"""
|
|
return _base_formatter(state="info", text=text, callback=callback, **kwargs)
|
|
|
|
|
|
def error(text, callback=CliError, **kwargs):
|
|
"""Generate formatted exception.
|
|
|
|
Arguments:
|
|
text {str} -- Text to format
|
|
callback {callable} -- Callback function (default: {echo})
|
|
|
|
Raises:
|
|
ClickException: Raised after formatting
|
|
"""
|
|
raise _base_formatter(state="error", text=text, callback=callback, **kwargs)
|
|
|
|
|
|
def success(text, callback=echo, **kwargs):
|
|
"""Generate formatted success text.
|
|
|
|
Arguments:
|
|
text {str} -- Text to format
|
|
callback {callable} -- Callback function (default: {echo})
|
|
|
|
Returns:
|
|
{str} -- Success output
|
|
"""
|
|
return _base_formatter(state="success", text=text, callback=callback, **kwargs)
|
|
|
|
|
|
def warning(text, callback=echo, **kwargs):
|
|
"""Generate formatted warning text.
|
|
|
|
Arguments:
|
|
text {str} -- Text to format
|
|
callback {callable} -- Callback function (default: {echo})
|
|
|
|
Returns:
|
|
{str} -- Warning output
|
|
"""
|
|
return _base_formatter(state="warning", text=text, callback=callback, **kwargs)
|
|
|
|
|
|
def label(text, callback=echo, **kwargs):
|
|
"""Generate formatted info text with accented labels.
|
|
|
|
Arguments:
|
|
text {str} -- Text to format
|
|
callback {callable} -- Callback function (default: {echo})
|
|
|
|
Returns:
|
|
{str} -- Label output
|
|
"""
|
|
return _base_formatter(state="label", text=text, callback=callback, **kwargs)
|
|
|
|
|
|
def status(text, callback=echo, **kwargs):
|
|
"""Generate formatted status text.
|
|
|
|
Arguments:
|
|
text {str} -- Text to format
|
|
callback {callable} -- Callback function (default: {echo})
|
|
|
|
Returns:
|
|
{str} -- Status output
|
|
"""
|
|
return _base_formatter(state="status", text=text, callback=callback, **kwargs)
|