mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-11-23 13:20:41 +00:00
339 lines
12 KiB
Python
339 lines
12 KiB
Python
|
# (c) 2012-2014, Michael DeHaan <michael.dehaan@gmail.com>
|
||
|
|
#
|
||
|
|
# This file is part of Ansible
|
||
|
|
#
|
||
|
|
# Ansible is free software: you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU General Public License as published by
|
||
|
|
# the Free Software Foundation, either version 3 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
#
|
||
|
|
# Ansible is distributed in the hope that it will be useful,
|
||
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
# GNU General Public License for more details.
|
||
|
|
#
|
||
|
|
# You should have received a copy of the GNU General Public License
|
||
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
# Make coding more python3-ish
|
||
|
|
from __future__ import (absolute_import, division, print_function)
|
||
|
|
__metaclass__ = type
|
||
|
|
|
||
|
|
from ansible import constants as C
|
||
|
|
from ansible.errors import AnsibleParserError, AnsibleAssertionError
|
||
|
|
from ansible.module_utils.six import string_types
|
||
|
|
from ansible.playbook.attribute import FieldAttribute
|
||
|
|
from ansible.playbook.base import Base
|
||
|
|
from ansible.playbook.become import Become
|
||
|
|
from ansible.playbook.block import Block
|
||
|
|
from ansible.playbook.helpers import load_list_of_blocks, load_list_of_roles
|
||
|
|
from ansible.playbook.role import Role
|
||
|
|
from ansible.playbook.taggable import Taggable
|
||
|
|
from ansible.vars.manager import preprocess_vars
|
||
|
|
|
||
|
|
try:
|
||
|
|
from __main__ import display
|
||
|
|
except ImportError:
|
||
|
|
from ansible.utils.display import Display
|
||
|
|
display = Display()
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ['Play']
|
||
|
|
|
||
|
|
|
||
|
|
class Play(Base, Taggable, Become):
|
||
|
|
|
||
|
|
"""
|
||
|
|
A play is a language feature that represents a list of roles and/or
|
||
|
|
task/handler blocks to execute on a given set of hosts.
|
||
|
|
|
||
|
|
Usage:
|
||
|
|
|
||
|
|
Play.load(datastructure) -> Play
|
||
|
|
Play.something(...)
|
||
|
|
"""
|
||
|
|
|
||
|
|
# =================================================================================
|
||
|
|
_hosts = FieldAttribute(isa='list', required=True, listof=string_types, always_post_validate=True)
|
||
|
|
|
||
|
|
# Facts
|
||
|
|
_fact_path = FieldAttribute(isa='string', default=None)
|
||
|
|
_gather_facts = FieldAttribute(isa='bool', default=None, always_post_validate=True)
|
||
|
|
_gather_subset = FieldAttribute(isa='list', default=None, listof=string_types, always_post_validate=True)
|
||
|
|
_gather_timeout = FieldAttribute(isa='int', default=None, always_post_validate=True)
|
||
|
|
|
||
|
|
# Variable Attributes
|
||
|
|
_vars_files = FieldAttribute(isa='list', default=list, priority=99)
|
||
|
|
_vars_prompt = FieldAttribute(isa='list', default=list, always_post_validate=False)
|
||
|
|
|
||
|
|
# Role Attributes
|
||
|
|
_roles = FieldAttribute(isa='list', default=list, priority=90)
|
||
|
|
|
||
|
|
# Block (Task) Lists Attributes
|
||
|
|
_handlers = FieldAttribute(isa='list', default=list)
|
||
|
|
_pre_tasks = FieldAttribute(isa='list', default=list)
|
||
|
|
_post_tasks = FieldAttribute(isa='list', default=list)
|
||
|
|
_tasks = FieldAttribute(isa='list', default=list)
|
||
|
|
|
||
|
|
# Flag/Setting Attributes
|
||
|
|
_force_handlers = FieldAttribute(isa='bool', always_post_validate=True)
|
||
|
|
_max_fail_percentage = FieldAttribute(isa='percent', always_post_validate=True)
|
||
|
|
_serial = FieldAttribute(isa='list', default=list, always_post_validate=True)
|
||
|
|
_strategy = FieldAttribute(isa='string', default=C.DEFAULT_STRATEGY, always_post_validate=True)
|
||
|
|
_order = FieldAttribute(isa='string', always_post_validate=True)
|
||
|
|
|
||
|
|
# =================================================================================
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
super(Play, self).__init__()
|
||
|
|
|
||
|
|
self._included_conditional = None
|
||
|
|
self._included_path = None
|
||
|
|
self._removed_hosts = []
|
||
|
|
self.ROLE_CACHE = {}
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return self.get_name()
|
||
|
|
|
||
|
|
def get_name(self):
|
||
|
|
''' return the name of the Play '''
|
||
|
|
return self._attributes.get('name')
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def load(data, variable_manager=None, loader=None, vars=None):
|
||
|
|
if ('name' not in data or data['name'] is None) and 'hosts' in data:
|
||
|
|
if isinstance(data['hosts'], list):
|
||
|
|
data['name'] = ','.join(data['hosts'])
|
||
|
|
else:
|
||
|
|
data['name'] = data['hosts']
|
||
|
|
p = Play()
|
||
|
|
if vars:
|
||
|
|
p.vars = vars.copy()
|
||
|
|
return p.load_data(data, variable_manager=variable_manager, loader=loader)
|
||
|
|
|
||
|
|
def preprocess_data(self, ds):
|
||
|
|
'''
|
||
|
|
Adjusts play datastructure to cleanup old/legacy items
|
||
|
|
'''
|
||
|
|
|
||
|
|
if not isinstance(ds, dict):
|
||
|
|
raise AnsibleAssertionError('while preprocessing data (%s), ds should be a dict but was a %s' % (ds, type(ds)))
|
||
|
|
|
||
|
|
# The use of 'user' in the Play datastructure was deprecated to
|
||
|
|
# line up with the same change for Tasks, due to the fact that
|
||
|
|
# 'user' conflicted with the user module.
|
||
|
|
if 'user' in ds:
|
||
|
|
# this should never happen, but error out with a helpful message
|
||
|
|
# to the user if it does...
|
||
|
|
if 'remote_user' in ds:
|
||
|
|
raise AnsibleParserError("both 'user' and 'remote_user' are set for %s. "
|
||
|
|
"The use of 'user' is deprecated, and should be removed" % self.get_name(), obj=ds)
|
||
|
|
|
||
|
|
ds['remote_user'] = ds['user']
|
||
|
|
del ds['user']
|
||
|
|
|
||
|
|
return super(Play, self).preprocess_data(ds)
|
||
|
|
|
||
|
|
def _load_tasks(self, attr, ds):
|
||
|
|
'''
|
||
|
|
Loads a list of blocks from a list which may be mixed tasks/blocks.
|
||
|
|
Bare tasks outside of a block are given an implicit block.
|
||
|
|
'''
|
||
|
|
try:
|
||
|
|
return load_list_of_blocks(ds=ds, play=self, variable_manager=self._variable_manager, loader=self._loader)
|
||
|
|
except AssertionError as e:
|
||
|
|
raise AnsibleParserError("A malformed block was encountered while loading tasks", obj=self._ds, orig_exc=e)
|
||
|
|
|
||
|
|
def _load_pre_tasks(self, attr, ds):
|
||
|
|
'''
|
||
|
|
Loads a list of blocks from a list which may be mixed tasks/blocks.
|
||
|
|
Bare tasks outside of a block are given an implicit block.
|
||
|
|
'''
|
||
|
|
try:
|
||
|
|
return load_list_of_blocks(ds=ds, play=self, variable_manager=self._variable_manager, loader=self._loader)
|
||
|
|
except AssertionError as e:
|
||
|
|
raise AnsibleParserError("A malformed block was encountered while loading pre_tasks", obj=self._ds, orig_exc=e)
|
||
|
|
|
||
|
|
def _load_post_tasks(self, attr, ds):
|
||
|
|
'''
|
||
|
|
Loads a list of blocks from a list which may be mixed tasks/blocks.
|
||
|
|
Bare tasks outside of a block are given an implicit block.
|
||
|
|
'''
|
||
|
|
try:
|
||
|
|
return load_list_of_blocks(ds=ds, play=self, variable_manager=self._variable_manager, loader=self._loader)
|
||
|
|
except AssertionError as e:
|
||
|
|
raise AnsibleParserError("A malformed block was encountered while loading post_tasks", obj=self._ds, orig_exc=e)
|
||
|
|
|
||
|
|
def _load_handlers(self, attr, ds):
|
||
|
|
'''
|
||
|
|
Loads a list of blocks from a list which may be mixed handlers/blocks.
|
||
|
|
Bare handlers outside of a block are given an implicit block.
|
||
|
|
'''
|
||
|
|
try:
|
||
|
|
return self._extend_value(
|
||
|
|
self.handlers,
|
||
|
|
load_list_of_blocks(ds=ds, play=self, use_handlers=True, variable_manager=self._variable_manager, loader=self._loader),
|
||
|
|
prepend=True
|
||
|
|
)
|
||
|
|
except AssertionError as e:
|
||
|
|
raise AnsibleParserError("A malformed block was encountered while loading handlers", obj=self._ds, orig_exc=e)
|
||
|
|
|
||
|
|
def _load_roles(self, attr, ds):
|
||
|
|
'''
|
||
|
|
Loads and returns a list of RoleInclude objects from the datastructure
|
||
|
|
list of role definitions and creates the Role from those objects
|
||
|
|
'''
|
||
|
|
|
||
|
|
if ds is None:
|
||
|
|
ds = []
|
||
|
|
|
||
|
|
try:
|
||
|
|
role_includes = load_list_of_roles(ds, play=self, variable_manager=self._variable_manager, loader=self._loader)
|
||
|
|
except AssertionError as e:
|
||
|
|
raise AnsibleParserError("A malformed role declaration was encountered.", obj=self._ds, orig_exc=e)
|
||
|
|
|
||
|
|
roles = []
|
||
|
|
for ri in role_includes:
|
||
|
|
roles.append(Role.load(ri, play=self))
|
||
|
|
|
||
|
|
return self._extend_value(
|
||
|
|
self.roles,
|
||
|
|
roles,
|
||
|
|
prepend=True
|
||
|
|
)
|
||
|
|
|
||
|
|
def _load_vars_prompt(self, attr, ds):
|
||
|
|
new_ds = preprocess_vars(ds)
|
||
|
|
vars_prompts = []
|
||
|
|
if new_ds is not None:
|
||
|
|
for prompt_data in new_ds:
|
||
|
|
if 'name' not in prompt_data:
|
||
|
|
raise AnsibleParserError("Invalid vars_prompt data structure", obj=ds)
|
||
|
|
else:
|
||
|
|
vars_prompts.append(prompt_data)
|
||
|
|
return vars_prompts
|
||
|
|
|
||
|
|
def _compile_roles(self):
|
||
|
|
'''
|
||
|
|
Handles the role compilation step, returning a flat list of tasks
|
||
|
|
with the lowest level dependencies first. For example, if a role R
|
||
|
|
has a dependency D1, which also has a dependency D2, the tasks from
|
||
|
|
D2 are merged first, followed by D1, and lastly by the tasks from
|
||
|
|
the parent role R last. This is done for all roles in the Play.
|
||
|
|
'''
|
||
|
|
|
||
|
|
block_list = []
|
||
|
|
|
||
|
|
if len(self.roles) > 0:
|
||
|
|
for r in self.roles:
|
||
|
|
# Don't insert tasks from ``import/include_role``, preventing
|
||
|
|
# duplicate execution at the wrong time
|
||
|
|
if r.from_include:
|
||
|
|
continue
|
||
|
|
block_list.extend(r.compile(play=self))
|
||
|
|
|
||
|
|
return block_list
|
||
|
|
|
||
|
|
def compile_roles_handlers(self):
|
||
|
|
'''
|
||
|
|
Handles the role handler compilation step, returning a flat list of Handlers
|
||
|
|
This is done for all roles in the Play.
|
||
|
|
'''
|
||
|
|
|
||
|
|
block_list = []
|
||
|
|
|
||
|
|
if len(self.roles) > 0:
|
||
|
|
for r in self.roles:
|
||
|
|
if r.from_include:
|
||
|
|
continue
|
||
|
|
block_list.extend(r.get_handler_blocks(play=self))
|
||
|
|
|
||
|
|
return block_list
|
||
|
|
|
||
|
|
def compile(self):
|
||
|
|
'''
|
||
|
|
Compiles and returns the task list for this play, compiled from the
|
||
|
|
roles (which are themselves compiled recursively) and/or the list of
|
||
|
|
tasks specified in the play.
|
||
|
|
'''
|
||
|
|
|
||
|
|
# create a block containing a single flush handlers meta
|
||
|
|
# task, so we can be sure to run handlers at certain points
|
||
|
|
# of the playbook execution
|
||
|
|
flush_block = Block.load(
|
||
|
|
data={'meta': 'flush_handlers'},
|
||
|
|
play=self,
|
||
|
|
variable_manager=self._variable_manager,
|
||
|
|
loader=self._loader
|
||
|
|
)
|
||
|
|
|
||
|
|
block_list = []
|
||
|
|
|
||
|
|
block_list.extend(self.pre_tasks)
|
||
|
|
block_list.append(flush_block)
|
||
|
|
block_list.extend(self._compile_roles())
|
||
|
|
block_list.extend(self.tasks)
|
||
|
|
block_list.append(flush_block)
|
||
|
|
block_list.extend(self.post_tasks)
|
||
|
|
block_list.append(flush_block)
|
||
|
|
|
||
|
|
return block_list
|
||
|
|
|
||
|
|
def get_vars(self):
|
||
|
|
return self.vars.copy()
|
||
|
|
|
||
|
|
def get_vars_files(self):
|
||
|
|
if self.vars_files is None:
|
||
|
|
return []
|
||
|
|
elif not isinstance(self.vars_files, list):
|
||
|
|
return [self.vars_files]
|
||
|
|
return self.vars_files
|
||
|
|
|
||
|
|
def get_handlers(self):
|
||
|
|
return self.handlers[:]
|
||
|
|
|
||
|
|
def get_roles(self):
|
||
|
|
return self.roles[:]
|
||
|
|
|
||
|
|
def get_tasks(self):
|
||
|
|
tasklist = []
|
||
|
|
for task in self.pre_tasks + self.tasks + self.post_tasks:
|
||
|
|
if isinstance(task, Block):
|
||
|
|
tasklist.append(task.block + task.rescue + task.always)
|
||
|
|
else:
|
||
|
|
tasklist.append(task)
|
||
|
|
return tasklist
|
||
|
|
|
||
|
|
def serialize(self):
|
||
|
|
data = super(Play, self).serialize()
|
||
|
|
|
||
|
|
roles = []
|
||
|
|
for role in self.get_roles():
|
||
|
|
roles.append(role.serialize())
|
||
|
|
data['roles'] = roles
|
||
|
|
data['included_path'] = self._included_path
|
||
|
|
|
||
|
|
return data
|
||
|
|
|
||
|
|
def deserialize(self, data):
|
||
|
|
super(Play, self).deserialize(data)
|
||
|
|
|
||
|
|
self._included_path = data.get('included_path', None)
|
||
|
|
if 'roles' in data:
|
||
|
|
role_data = data.get('roles', [])
|
||
|
|
roles = []
|
||
|
|
for role in role_data:
|
||
|
|
r = Role()
|
||
|
|
r.deserialize(role)
|
||
|
|
roles.append(r)
|
||
|
|
|
||
|
|
setattr(self, 'roles', roles)
|
||
|
|
del data['roles']
|
||
|
|
|
||
|
|
def copy(self):
|
||
|
|
new_me = super(Play, self).copy()
|
||
|
|
new_me.ROLE_CACHE = self.ROLE_CACHE.copy()
|
||
|
|
new_me._included_conditional = self._included_conditional
|
||
|
|
new_me._included_path = self._included_path
|
||
|
|
return new_me
|