salt/tasks/utils.py
Pedro Algarvio 6a4703146c
Replace tests/unit/test_doc.py with a pre-commit hook.
We don't need to test docs on all platforms. In turn, we make the check
a pre-commit hook which seems more suited.
2020-04-23 20:01:05 +01:00

61 lines
1.4 KiB
Python

# -*- coding: utf-8 -*-
"""
tasks.utils
~~~~~~~~~~~
Invoke utilities
"""
import sys
try:
from blessings import Terminal
terminal = Terminal()
HAS_BLESSINGS = True
except ImportError:
terminal = None
HAS_BLESSINGS = False
def exit_invoke(exitcode, message=None, *args, **kwargs):
if message is not None:
if exitcode > 0:
warn(message, *args, **kwargs)
else:
info(message, *args, **kwargs)
sys.exit(exitcode)
def info(message, *args, **kwargs):
if not isinstance(message, str):
message = str(message)
message = message.format(*args, **kwargs)
if terminal:
message = terminal.bold(terminal.green(message))
write_message(message)
def warn(message, *args, **kwargs):
if not isinstance(message, str):
message = str(message)
message = message.format(*args, **kwargs)
if terminal:
message = terminal.bold(terminal.yellow(message))
write_message(message)
def error(message, *args, **kwargs):
if not isinstance(message, str):
message = str(message)
message = message.format(*args, **kwargs)
if terminal:
message = terminal.bold(terminal.red(message))
write_message(message)
def write_message(message):
sys.stderr.write(message)
if not message.endswith("\n"):
sys.stderr.write("\n")
sys.stderr.flush()