Added comments to explain the playbook code.
This commit is contained in:
parent
17a6ddb7ee
commit
c0ac0e9b62
1 changed files with 54 additions and 9 deletions
|
@ -49,8 +49,11 @@ class PlayBook(object):
|
||||||
remote_pass =C.DEFAULT_REMOTE_PASS,
|
remote_pass =C.DEFAULT_REMOTE_PASS,
|
||||||
verbose=False):
|
verbose=False):
|
||||||
|
|
||||||
# runner is reused between calls
|
# TODO, once ansible-playbook is it's own script this will
|
||||||
|
# have much LESS parameters to the constructor and will
|
||||||
|
# read most everything per pattern from the playbook
|
||||||
|
# and this will be greatly simplified
|
||||||
|
|
||||||
self.host_list = host_list
|
self.host_list = host_list
|
||||||
self.module_path = module_path
|
self.module_path = module_path
|
||||||
self.forks = forks
|
self.forks = forks
|
||||||
|
@ -59,25 +62,32 @@ class PlayBook(object):
|
||||||
self.remote_pass = remote_pass
|
self.remote_pass = remote_pass
|
||||||
self.verbose = verbose
|
self.verbose = verbose
|
||||||
|
|
||||||
# list of changes/invocations/failure counts per host
|
# store the list of changes/invocations/failure counts
|
||||||
|
# as a dictionary of integers keyed off the hostname
|
||||||
|
|
||||||
self.processed = {}
|
self.processed = {}
|
||||||
self.dark = {}
|
self.dark = {}
|
||||||
self.changed = {}
|
self.changed = {}
|
||||||
self.invocations = {}
|
self.invocations = {}
|
||||||
self.failures = {}
|
self.failures = {}
|
||||||
|
|
||||||
|
# playbook file can be passed in as a path or
|
||||||
|
# as file contents (to support API usage)
|
||||||
|
|
||||||
if type(playbook) == str:
|
if type(playbook) == str:
|
||||||
playbook = yaml.load(file(playbook).read())
|
playbook = yaml.load(file(playbook).read())
|
||||||
self.playbook = playbook
|
self.playbook = playbook
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
''' run against all patterns in the playbook '''
|
''' run all patterns in the playbook '''
|
||||||
|
|
||||||
|
# loop through all patterns and run them
|
||||||
for pattern in self.playbook:
|
for pattern in self.playbook:
|
||||||
self._run_pattern(pattern)
|
self._run_pattern(pattern)
|
||||||
if self.verbose:
|
if self.verbose:
|
||||||
print "\n"
|
print "\n"
|
||||||
|
|
||||||
|
# summarize the results
|
||||||
results = {}
|
results = {}
|
||||||
for host in self.processed.keys():
|
for host in self.processed.keys():
|
||||||
results[host] = {
|
results[host] = {
|
||||||
|
@ -96,20 +106,33 @@ class PlayBook(object):
|
||||||
'''
|
'''
|
||||||
|
|
||||||
if host_list is None:
|
if host_list is None:
|
||||||
|
# pruned host lists occur when running triggered
|
||||||
|
# actions where not all hosts have changed
|
||||||
|
# though top-level tasks will pass in "None" here
|
||||||
host_list = self.host_list
|
host_list = self.host_list
|
||||||
|
|
||||||
|
# load the module name and parameters from the task
|
||||||
|
# entry
|
||||||
instructions = task['do']
|
instructions = task['do']
|
||||||
(comment, module_details) = instructions
|
(comment, module_details) = instructions
|
||||||
tokens = shlex.split(module_details)
|
tokens = shlex.split(module_details)
|
||||||
module_name = tokens[0]
|
module_name = tokens[0]
|
||||||
module_args = tokens[1:]
|
module_args = tokens[1:]
|
||||||
|
|
||||||
|
# tasks can be direct (run on all nodes matching
|
||||||
|
# the pattern) or conditional, where they ran
|
||||||
|
# as the result of a change handler on a subset
|
||||||
|
# of all of the hosts
|
||||||
|
|
||||||
if self.verbose:
|
if self.verbose:
|
||||||
if not conditional:
|
if not conditional:
|
||||||
print "\nTASK [%s]" % (comment)
|
print "\nTASK [%s]" % (comment)
|
||||||
else:
|
else:
|
||||||
print "\nNOTIFIED [%s]" % (comment)
|
print "\nNOTIFIED [%s]" % (comment)
|
||||||
|
|
||||||
|
# load up an appropriate ansible runner to
|
||||||
|
# run the task in parallel
|
||||||
|
|
||||||
runner = ansible.runner.Runner(
|
runner = ansible.runner.Runner(
|
||||||
pattern=pattern,
|
pattern=pattern,
|
||||||
module_name=module_name,
|
module_name=module_name,
|
||||||
|
@ -123,6 +146,10 @@ class PlayBook(object):
|
||||||
)
|
)
|
||||||
results = runner.run()
|
results = runner.run()
|
||||||
|
|
||||||
|
# walk through the results and build up
|
||||||
|
# summary information about successes and
|
||||||
|
# failures. TODO: split into subfunction
|
||||||
|
|
||||||
dark = results.get("dark", [])
|
dark = results.get("dark", [])
|
||||||
contacted = results.get("contacted", [])
|
contacted = results.get("contacted", [])
|
||||||
ok_hosts = contacted.keys()
|
ok_hosts = contacted.keys()
|
||||||
|
@ -165,8 +192,11 @@ class PlayBook(object):
|
||||||
else:
|
else:
|
||||||
self.changed[host] = self.changed[host] + 1
|
self.changed[host] = self.changed[host] + 1
|
||||||
|
|
||||||
|
|
||||||
# flag which notify handlers need to be run
|
# flag which notify handlers need to be run
|
||||||
|
# this will be on a SUBSET of the actual host list. For instance
|
||||||
|
# a file might need to be written on only half of the nodes so
|
||||||
|
# we would only trigger restarting Apache on half of the nodes
|
||||||
|
|
||||||
subtasks = task.get('notify', [])
|
subtasks = task.get('notify', [])
|
||||||
if len(subtasks) > 0:
|
if len(subtasks) > 0:
|
||||||
for host, results in contacted.items():
|
for host, results in contacted.items():
|
||||||
|
@ -174,19 +204,22 @@ class PlayBook(object):
|
||||||
for subtask in subtasks:
|
for subtask in subtasks:
|
||||||
self._flag_handler(handlers, subtask, host)
|
self._flag_handler(handlers, subtask, host)
|
||||||
|
|
||||||
# TODO: if a host fails in any task, remove it from
|
|
||||||
# the host list immediately
|
|
||||||
|
|
||||||
def _flag_handler(self, handlers, match_name, host):
|
def _flag_handler(self, handlers, match_name, host):
|
||||||
'''
|
'''
|
||||||
if a task has any notify elements, flag handlers for run
|
if a task has any notify elements, flag handlers for run
|
||||||
at end of execution cycle for hosts that have indicated
|
at end of execution cycle for hosts that have indicated
|
||||||
changes have been made
|
changes have been made
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
# for all registered handlers in the ansible playbook
|
||||||
|
# for this particular pattern group
|
||||||
|
|
||||||
for x in handlers:
|
for x in handlers:
|
||||||
attribs = x["do"]
|
attribs = x["do"]
|
||||||
name = attribs[0]
|
name = attribs[0]
|
||||||
if match_name == name:
|
if match_name == name:
|
||||||
|
# flag the handler with the list of hosts
|
||||||
|
# it needs to be run on, it will be run later
|
||||||
if not x.has_key("run"):
|
if not x.has_key("run"):
|
||||||
x['run'] = []
|
x['run'] = []
|
||||||
x['run'].append(host)
|
x['run'].append(host)
|
||||||
|
@ -196,22 +229,31 @@ class PlayBook(object):
|
||||||
run a list of tasks for a given pattern, in order
|
run a list of tasks for a given pattern, in order
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
# get configuration information about the pattern
|
||||||
pattern = pg['pattern']
|
pattern = pg['pattern']
|
||||||
tasks = pg['tasks']
|
tasks = pg['tasks']
|
||||||
handlers = pg['handlers']
|
handlers = pg['handlers']
|
||||||
user = pg.get('user', C.DEFAULT_REMOTE_USER)
|
user = pg.get('user', C.DEFAULT_REMOTE_USER)
|
||||||
|
|
||||||
self.host_list = pg.get('hosts', '/etc/ansible/hosts')
|
self.host_list = pg.get('hosts', '/etc/ansible/hosts')
|
||||||
|
|
||||||
if self.verbose:
|
if self.verbose:
|
||||||
print "PLAY: [%s] from [%s] ********** " % (pattern, self.host_list)
|
print "PLAY: [%s] from [%s] ********** " % (pattern, self.host_list)
|
||||||
|
|
||||||
|
# run all the top level tasks, these get run on every node
|
||||||
|
|
||||||
for task in tasks:
|
for task in tasks:
|
||||||
self._run_task(
|
self._run_task(
|
||||||
pattern=pattern,
|
pattern=pattern,
|
||||||
task=task,
|
task=task,
|
||||||
handlers=handlers,
|
handlers=handlers,
|
||||||
remote_user=user)
|
remote_user=user)
|
||||||
|
|
||||||
|
# handlers only run on certain nodes, they are flagged by _flag_handlers
|
||||||
|
# above. They only run on nodes when things mark them as changed, and
|
||||||
|
# handlers only get run once. For instance, the system is designed
|
||||||
|
# such that multiple config files if changed can ask for an Apache restart
|
||||||
|
# but Apache will only be restarted once (at the end).
|
||||||
|
|
||||||
for task in handlers:
|
for task in handlers:
|
||||||
if type(task.get("run", None)) == list:
|
if type(task.get("run", None)) == list:
|
||||||
self._run_task(
|
self._run_task(
|
||||||
|
@ -223,5 +265,8 @@ class PlayBook(object):
|
||||||
remote_user=user
|
remote_user=user
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# end of execution for this particular pattern. Multiple patterns
|
||||||
|
# can be in a single playbook file
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue