mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-05-02 23:31:25 -07:00
108 lines
4.2 KiB
Python
108 lines
4.2 KiB
Python
# Copyright 2017 RedHat, inc
|
|
#
|
|
# 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/>.
|
|
#############################################
|
|
'''
|
|
DOCUMENTATION:
|
|
vars: host_group_vars
|
|
version_added: "2.4"
|
|
short_description: In charge of loading group_vars and host_vars
|
|
description:
|
|
- Loads YAML vars into corresponding groups/hosts in group_vars/ and host_vars/ directories.
|
|
- Files are restricted by extension to one of .yaml, .json, .yml or no extension.
|
|
- Only applies to inventory sources that are existing paths.
|
|
notes:
|
|
- It takes the place of the previously hardcoded group_vars/host_vars loading.
|
|
'''
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import os
|
|
from ansible import constants as C
|
|
from ansible.errors import AnsibleParserError
|
|
from ansible.module_utils._text import to_bytes, to_text
|
|
from ansible.plugins.vars import BaseVarsPlugin
|
|
from ansible.inventory.host import Host
|
|
from ansible.inventory.group import Group
|
|
from ansible.utils.vars import combine_vars
|
|
|
|
|
|
class VarsModule(BaseVarsPlugin):
|
|
|
|
def get_vars(self, loader, path, entities):
|
|
''' parses the inventory file '''
|
|
|
|
if not isinstance(entities, list):
|
|
entities = [entities]
|
|
|
|
super(VarsModule, self).get_vars(loader, path, entities)
|
|
|
|
data = {}
|
|
for entity in entities:
|
|
if isinstance(entity, Host):
|
|
subdir = 'host_vars'
|
|
elif isinstance(entity, Group):
|
|
subdir = 'group_vars'
|
|
else:
|
|
raise AnsibleParserError("Supplied entity must be Host or Group, got %s instead" % (type(entity)))
|
|
|
|
try:
|
|
# load vars
|
|
opath = os.path.realpath(os.path.join(self._basedir, subdir))
|
|
b_opath = to_bytes(opath)
|
|
# no need to do much if path does not exist for basedir
|
|
if os.path.exists(b_opath):
|
|
if os.path.isdir(b_opath):
|
|
self._display.debug("\tprocessing dir %s" % opath)
|
|
for found in self._find_vars_files(opath, entity.name):
|
|
self._display.debug("READING %s" % found)
|
|
new_data = loader.load_from_file(found, cache=True, unsafe=True)
|
|
if new_data: # ignore empty files
|
|
data = combine_vars(data, new_data)
|
|
else:
|
|
self._display.warning("Found %s that is not a directory, skipping: %s" % (subdir, opath))
|
|
|
|
except Exception as e:
|
|
raise AnsibleParserError(to_text(e))
|
|
return data
|
|
|
|
def _find_vars_files(self, path, name):
|
|
""" Find {group,host}_vars files """
|
|
|
|
b_path = to_bytes(os.path.join(path, name))
|
|
found = []
|
|
for ext in C.YAML_FILENAME_EXTENSIONS + ['']:
|
|
|
|
if '.' in ext:
|
|
full_path = b_path + to_bytes(ext)
|
|
elif ext:
|
|
full_path = b'.'.join([b_path, to_bytes(ext)])
|
|
else:
|
|
full_path = b_path
|
|
|
|
if os.path.exists(full_path):
|
|
self._display.debug("\tfound %s" % to_text(full_path))
|
|
if os.path.isdir(full_path):
|
|
# matched dir name, so use all files included recursively
|
|
for spath in os.listdir(full_path):
|
|
if os.path.isdir(spath):
|
|
found.extend(self._find_vars_files(spath, name))
|
|
else:
|
|
found.append(spath)
|
|
else:
|
|
found.append(full_path)
|
|
return found
|