mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-04-24 11:21:25 -07:00
cc343a4376
* Using docstrings conflicts with the standard use of docstrings * PYTHON_OPTIMIZE=2 will omit docstrings. Using docstrings makes future changes to the plugin and module code subject to the requirement that we ensure it won't be run with optimization.
272 lines
9.7 KiB
Python
272 lines
9.7 KiB
Python
# (c) Fastly, inc 2016
|
|
# (c) 2017 Ansible Project
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
DOCUMENTATION = """
|
|
callback: selective
|
|
callback_type: stdout
|
|
requirements:
|
|
- set as main display callback
|
|
short_description: only print certain tasks
|
|
version_added: "2.4"
|
|
description:
|
|
- This callback only prints tasks that have been tagged with `print_action` or that have failed.
|
|
This allows operators to focus on the tasks that provide value only.
|
|
- Tasks that are not printed are placed with a '.'.
|
|
- If you increase verbosity all tasks are printed.
|
|
options:
|
|
nocolor:
|
|
default: False
|
|
description: This setting allows suppressing colorizing output
|
|
env:
|
|
- name: ANSIBLE_NOCOLOR
|
|
- name: ANSIBLE_SELECTIVE_DONT_COLORIZE
|
|
ini:
|
|
- section: defaults
|
|
- key: nocolor
|
|
type: boolean
|
|
"""
|
|
|
|
EXAMPLES = """
|
|
- debug: msg="This will not be printed"
|
|
- debug: msg="But this will"
|
|
tags: [print_action]
|
|
"""
|
|
|
|
import difflib
|
|
|
|
from ansible import constants as C
|
|
from ansible.plugins.callback import CallbackBase
|
|
from ansible.module_utils._text import to_text
|
|
|
|
|
|
DONT_COLORIZE = False
|
|
COLORS = {
|
|
'normal': '\033[0m',
|
|
'ok': C.COLOR_OK,
|
|
'bold': '\033[1m',
|
|
'not_so_bold': '\033[1m\033[34m',
|
|
'changed': C.COLOR_CHANGED,
|
|
'failed': C.COLOR_ERROR,
|
|
'endc': '\033[0m',
|
|
'skipped': C.COLOR_SKIP,
|
|
}
|
|
|
|
|
|
def dict_diff(prv, nxt):
|
|
"""Return a dict of keys that differ with another config object."""
|
|
keys = set(prv.keys() + nxt.keys())
|
|
result = {}
|
|
for k in keys:
|
|
if prv.get(k) != nxt.get(k):
|
|
result[k] = (prv.get(k), nxt.get(k))
|
|
return result
|
|
|
|
|
|
def colorize(msg, color):
|
|
"""Given a string add necessary codes to format the string."""
|
|
if DONT_COLORIZE:
|
|
return msg
|
|
else:
|
|
return '{}{}{}'.format(COLORS[color], msg, COLORS['endc'])
|
|
|
|
|
|
class CallbackModule(CallbackBase):
|
|
"""selective.py callback plugin."""
|
|
|
|
CALLBACK_VERSION = 2.0
|
|
CALLBACK_TYPE = 'stdout'
|
|
CALLBACK_NAME = 'selective'
|
|
|
|
def __init__(self, display=None):
|
|
"""selective.py callback plugin."""
|
|
super(CallbackModule, self).__init__(display)
|
|
self.last_skipped = False
|
|
self.last_task_name = None
|
|
self.printed_last_task = False
|
|
|
|
def set_options(self, options):
|
|
|
|
super(CallbackModule, self).set_options(options)
|
|
|
|
global DONT_COLORIZE
|
|
DONT_COLORIZE = self._plugin_options['nocolor']
|
|
|
|
def _print_task(self, task_name=None):
|
|
if task_name is None:
|
|
task_name = self.last_task_name
|
|
|
|
if not self.printed_last_task:
|
|
self.printed_last_task = True
|
|
line_length = 120
|
|
if self.last_skipped:
|
|
print()
|
|
msg = colorize("# {} {}".format(task_name,
|
|
'*' * (line_length - len(task_name))), 'bold')
|
|
print(msg)
|
|
|
|
def _indent_text(self, text, indent_level):
|
|
lines = text.splitlines()
|
|
result_lines = []
|
|
for l in lines:
|
|
result_lines.append("{}{}".format(' ' * indent_level, l))
|
|
return '\n'.join(result_lines)
|
|
|
|
def _print_diff(self, diff, indent_level):
|
|
if isinstance(diff, dict):
|
|
try:
|
|
diff = '\n'.join(difflib.unified_diff(diff['before'].splitlines(),
|
|
diff['after'].splitlines(),
|
|
fromfile=diff.get('before_header',
|
|
'new_file'),
|
|
tofile=diff['after_header']))
|
|
except AttributeError:
|
|
diff = dict_diff(diff['before'], diff['after'])
|
|
if diff:
|
|
diff = colorize(str(diff), 'changed')
|
|
print(self._indent_text(diff, indent_level + 4))
|
|
|
|
def _print_host_or_item(self, host_or_item, changed, msg, diff, is_host, error, stdout, stderr):
|
|
if is_host:
|
|
indent_level = 0
|
|
name = colorize(host_or_item.name, 'not_so_bold')
|
|
else:
|
|
indent_level = 4
|
|
if isinstance(host_or_item, dict):
|
|
if 'key' in host_or_item.keys():
|
|
host_or_item = host_or_item['key']
|
|
name = colorize(to_text(host_or_item), 'bold')
|
|
|
|
if error:
|
|
color = 'failed'
|
|
change_string = colorize('FAILED!!!', color)
|
|
else:
|
|
color = 'changed' if changed else 'ok'
|
|
change_string = colorize("changed={}".format(changed), color)
|
|
|
|
msg = colorize(msg, color)
|
|
|
|
line_length = 120
|
|
spaces = ' ' * (40 - len(name) - indent_level)
|
|
line = "{} * {}{}- {}".format(' ' * indent_level, name, spaces, change_string)
|
|
|
|
if len(msg) < 50:
|
|
line += ' -- {}'.format(msg)
|
|
print("{} {}---------".format(line, '-' * (line_length - len(line))))
|
|
else:
|
|
print("{} {}".format(line, '-' * (line_length - len(line))))
|
|
print(self._indent_text(msg, indent_level + 4))
|
|
|
|
if diff:
|
|
self._print_diff(diff, indent_level)
|
|
if stdout:
|
|
stdout = colorize(stdout, 'failed')
|
|
print(self._indent_text(stdout, indent_level + 4))
|
|
if stderr:
|
|
stderr = colorize(stderr, 'failed')
|
|
print(self._indent_text(stderr, indent_level + 4))
|
|
|
|
def v2_playbook_on_play_start(self, play):
|
|
"""Run on start of the play."""
|
|
pass
|
|
|
|
def v2_playbook_on_task_start(self, task, **kwargs):
|
|
"""Run when a task starts."""
|
|
self.last_task_name = task.get_name()
|
|
self.printed_last_task = False
|
|
|
|
def v2_runner_on_ok(self, result, **kwargs):
|
|
"""Run when a task finishes correctly."""
|
|
failed = result._result.get("failed")
|
|
unreachable = result._result.get("unreachable")
|
|
|
|
if 'print_action' in result._task.tags or failed or unreachable or \
|
|
self._display.verbosity > 1:
|
|
self._print_task()
|
|
self.last_skipped = False
|
|
msg = to_text(result._result.get('msg', '')) or\
|
|
to_text(result._result.get('reason', ''))
|
|
|
|
stderr = [result._result.get('exception', None),
|
|
result._result.get('module_stderr', None)]
|
|
stderr = "\n".join([e for e in stderr if e]).strip()
|
|
|
|
self._print_host_or_item(result._host,
|
|
result._result.get('changed', False),
|
|
msg,
|
|
result._result.get('diff', None),
|
|
is_host=True,
|
|
error=failed or unreachable,
|
|
stdout=result._result.get('module_stdout', None),
|
|
stderr=stderr.strip(),
|
|
)
|
|
if 'results' in result._result:
|
|
for r in result._result['results']:
|
|
failed = 'failed' in r
|
|
|
|
stderr = [r.get('exception', None), r.get('module_stderr', None)]
|
|
stderr = "\n".join([e for e in stderr if e]).strip()
|
|
|
|
self._print_host_or_item(r['item'],
|
|
r.get('changed', False),
|
|
to_text(r.get('msg', '')),
|
|
r.get('diff', None),
|
|
is_host=False,
|
|
error=failed,
|
|
stdout=r.get('module_stdout', None),
|
|
stderr=stderr.strip(),
|
|
)
|
|
else:
|
|
self.last_skipped = True
|
|
print('.', end="")
|
|
|
|
def v2_playbook_on_stats(self, stats):
|
|
"""Display info about playbook statistics."""
|
|
print()
|
|
self.printed_last_task = False
|
|
self._print_task('STATS')
|
|
|
|
hosts = sorted(stats.processed.keys())
|
|
for host in hosts:
|
|
s = stats.summarize(host)
|
|
|
|
if s['failures'] or s['unreachable']:
|
|
color = 'failed'
|
|
elif s['changed']:
|
|
color = 'changed'
|
|
else:
|
|
color = 'ok'
|
|
|
|
msg = '{} : ok={}\tchanged={}\tfailed={}\tunreachable={}'.format(
|
|
host, s['ok'], s['changed'], s['failures'], s['unreachable'])
|
|
print(colorize(msg, color))
|
|
|
|
def v2_runner_on_skipped(self, result, **kwargs):
|
|
"""Run when a task is skipped."""
|
|
if self._display.verbosity > 1:
|
|
self._print_task()
|
|
self.last_skipped = False
|
|
|
|
line_length = 120
|
|
spaces = ' ' * (31 - len(result._host.name) - 4)
|
|
|
|
line = " * {}{}- {}".format(colorize(result._host.name, 'not_so_bold'),
|
|
spaces,
|
|
colorize("skipped", 'skipped'),)
|
|
|
|
reason = result._result.get('skipped_reason', '') or \
|
|
result._result.get('skip_reason', '')
|
|
if len(reason) < 50:
|
|
line += ' -- {}'.format(reason)
|
|
print("{} {}---------".format(line, '-' * (line_length - len(line))))
|
|
else:
|
|
print("{} {}".format(line, '-' * (line_length - len(line))))
|
|
print(self._indent_text(reason, 8))
|
|
print(reason)
|
|
|
|
v2_playbook_on_handler_task_start = v2_playbook_on_task_start
|
|
v2_runner_on_failed = v2_runner_on_ok
|
|
v2_runner_on_unreachable = v2_runner_on_ok
|