Coverage for lib/ansible/plugins/callback/profile_roles.py : 31%

# (c) 2017, Tennis Smith, http://github.com/gamename 

# (c) 2017 Ansible Project 

# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) 

# Make coding more python3-ish 

from __future__ import (absolute_import, division, print_function) 

__metaclass__ = type 

DOCUMENTATION = ''' 

callback: profile_roles 

type: aggregate 

short_description: adds timing information to roles 

version_added: "2.4" 

description: 

- This callback module provides profiling for ansible roles. 

requirements: 

- whitelisting in configuration 

''' 

import collections 

import time 

from ansible.plugins.callback import CallbackBase 

from ansible.module_utils.six.moves import reduce 

# define start time 

t0 = tn = time.time() 

def secondsToStr(t): 

# http://bytes.com/topic/python/answers/635958-handy-short-cut-formatting-elapsed-time-floating-point-seconds 

def rediv(ll, b): 

return list(divmod(ll[0], b)) + ll[1:] 

return "%d:%02d:%02d.%03d" % tuple( 

reduce(rediv, [[t * 1000, ], 1000, 60, 60])) 

def filled(msg, fchar="*"): 

if len(msg) == 0: 

width = 79 

else: 

msg = "%s " % msg 

width = 79 - len(msg) 

if width < 3: 

width = 3 

filler = fchar * width 

return "%s%s " % (msg, filler) 

def timestamp(self): 

if self.current is not None: 

self.stats[self.current] = time.time() - self.stats[self.current] 

self.totals[self.current] += self.stats[self.current] 

def tasktime(): 

global tn 

time_current = time.strftime('%A %d %B %Y %H:%M:%S %z') 

time_elapsed = secondsToStr(time.time() - tn) 

time_total_elapsed = secondsToStr(time.time() - t0) 

tn = time.time() 

return filled('%s (%s)%s%s' % 

(time_current, time_elapsed, ' ' * 7, time_total_elapsed)) 

class CallbackModule(CallbackBase): 

""" 

This callback module provides profiling for ansible roles. 

""" 

CALLBACK_VERSION = 2.0 

CALLBACK_TYPE = 'aggregate' 

CALLBACK_NAME = 'profile_roles' 

CALLBACK_NEEDS_WHITELIST = True 

def __init__(self): 

self.stats = collections.Counter() 

self.totals = collections.Counter() 

self.current = None 

super(CallbackModule, self).__init__() 

def _record_task(self, task): 

""" 

Logs the start of each task 

""" 

self._display.display(tasktime()) 

timestamp(self) 

if task._role: 

self.current = task._role._role_name 

else: 

self.current = task.action 

self.stats[self.current] = time.time() 

def v2_playbook_on_task_start(self, task, is_conditional): 

self._record_task(task) 

def v2_playbook_on_handler_task_start(self, task): 

self._record_task(task) 

def playbook_on_setup(self): 

self._display.display(tasktime()) 

def playbook_on_stats(self, stats): 

self._display.display(tasktime()) 

self._display.display(filled("", fchar="=")) 

timestamp(self) 

total_time = sum(self.totals.values()) 

# Print the timings starting with the largest one 

for result in self.totals.most_common(): 

msg = u"{0:-<70}{1:->9}".format(result[0] + u' ', u' {0:.02f}s'.format(result[1])) 

self._display.display(msg) 

msg_total = u"{0:-<70}{1:->9}".format(u'total ', u' {0:.02f}s'.format(total_time)) 

self._display.display(filled("", fchar="~")) 

self._display.display(msg_total)