mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-04-29 13:51:28 -07:00
e0b0e61b32
This exposed some additional errors in logic in IncludeFile, which had to be fixed to deal with the fact that the role path (unlike paths from includes) are always absolute paths. Fixes #27345
143 lines
6.2 KiB
Python
143 lines
6.2 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
|
|
|
|
import os
|
|
|
|
from ansible.playbook.task_include import TaskInclude
|
|
from ansible.playbook.role_include import IncludeRole
|
|
from ansible.template import Templar
|
|
|
|
try:
|
|
from __main__ import display
|
|
except ImportError:
|
|
from ansible.utils.display import Display
|
|
display = Display()
|
|
|
|
|
|
class IncludedFile:
|
|
|
|
def __init__(self, filename, args, task):
|
|
self._filename = filename
|
|
self._args = args
|
|
self._task = task
|
|
self._hosts = []
|
|
|
|
def add_host(self, host):
|
|
if host not in self._hosts:
|
|
self._hosts.append(host)
|
|
|
|
def __eq__(self, other):
|
|
return other._filename == self._filename and other._args == self._args
|
|
|
|
def __repr__(self):
|
|
return "%s (%s): %s" % (self._filename, self._args, self._hosts)
|
|
|
|
@staticmethod
|
|
def process_include_results(results, tqm, iterator, inventory, loader, variable_manager):
|
|
included_files = []
|
|
|
|
def get_original_host(host):
|
|
if host.name in inventory._hosts_cache:
|
|
return inventory._hosts_cache[host.name]
|
|
else:
|
|
return inventory.get_host(host.name)
|
|
|
|
for res in results:
|
|
|
|
original_host = res._host
|
|
original_task = res._task
|
|
|
|
if original_task.action in ('include', 'include_tasks'):
|
|
if original_task.loop:
|
|
if 'results' not in res._result:
|
|
continue
|
|
include_results = res._result['results']
|
|
else:
|
|
include_results = [res._result]
|
|
|
|
for include_result in include_results:
|
|
# if the task result was skipped or failed, continue
|
|
if 'skipped' in include_result and include_result['skipped'] or 'failed' in include_result:
|
|
continue
|
|
|
|
task_vars = variable_manager.get_vars(play=iterator._play, host=original_host, task=original_task)
|
|
templar = Templar(loader=loader, variables=task_vars)
|
|
|
|
include_variables = include_result.get('include_variables', dict())
|
|
loop_var = 'item'
|
|
if original_task.loop_control:
|
|
loop_var = original_task.loop_control.loop_var or 'item'
|
|
if loop_var in include_result:
|
|
task_vars[loop_var] = include_variables[loop_var] = include_result[loop_var]
|
|
|
|
include_file = None
|
|
if original_task:
|
|
if original_task.static:
|
|
continue
|
|
|
|
if original_task._parent:
|
|
# handle relative includes by walking up the list of parent include
|
|
# tasks and checking the relative result to see if it exists
|
|
parent_include = original_task._parent
|
|
cumulative_path = None
|
|
while parent_include is not None:
|
|
if not isinstance(parent_include, TaskInclude):
|
|
parent_include = parent_include._parent
|
|
continue
|
|
if isinstance(parent_include, IncludeRole):
|
|
parent_include_dir = parent_include._role_path
|
|
else:
|
|
parent_include_dir = os.path.dirname(templar.template(parent_include.args.get('_raw_params')))
|
|
if cumulative_path is not None and not os.path.isabs(cumulative_path):
|
|
cumulative_path = os.path.join(parent_include_dir, cumulative_path)
|
|
else:
|
|
cumulative_path = parent_include_dir
|
|
include_target = templar.template(include_result['include'])
|
|
if original_task._role:
|
|
new_basedir = os.path.join(original_task._role._role_path, 'tasks', cumulative_path)
|
|
include_file = loader.path_dwim_relative(new_basedir, 'tasks', include_target)
|
|
else:
|
|
include_file = loader.path_dwim_relative(loader.get_basedir(), cumulative_path, include_target)
|
|
|
|
if os.path.exists(include_file):
|
|
break
|
|
else:
|
|
parent_include = parent_include._parent
|
|
|
|
if include_file is None:
|
|
if original_task._role:
|
|
include_target = templar.template(include_result['include'])
|
|
include_file = loader.path_dwim_relative(original_task._role._role_path, 'tasks', include_target)
|
|
else:
|
|
include_file = loader.path_dwim(include_result['include'])
|
|
|
|
include_file = templar.template(include_file)
|
|
inc_file = IncludedFile(include_file, include_variables, original_task)
|
|
|
|
try:
|
|
pos = included_files.index(inc_file)
|
|
inc_file = included_files[pos]
|
|
except ValueError:
|
|
included_files.append(inc_file)
|
|
|
|
inc_file.add_host(original_host)
|
|
|
|
return included_files
|