mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
0d46805979
- Remove shebangs from: - ini files - unit tests - module_utils - plugins - module_docs_fragments - non-executable Makefiles - Change non-modules from '/usr/bin/python' to '/usr/bin/env python'. - Change '/bin/env' to '/usr/bin/env'. Also removed main functions from unit tests (since they no longer have a shebang) and fixed a python 3 compatibility issue with update_bundled.py so it does not need to specify a python 2 shebang. A script was added to check for unexpected shebangs in files. This script is run during CI on Shippable.
461 lines
18 KiB
Python
Executable file
461 lines
18 KiB
Python
Executable file
#!/usr/bin/env python
|
|
# vim: set fileencoding=utf-8 :
|
|
#
|
|
# Copyright (C) 2016 Guido Günther <agx@sigxcpu.org>
|
|
#
|
|
# This script 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 it. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# This is loosely based on the foreman inventory script
|
|
# -- Josh Preston <jpreston@redhat.com>
|
|
#
|
|
|
|
from __future__ import print_function
|
|
import argparse
|
|
import ConfigParser
|
|
import os
|
|
import re
|
|
from time import time
|
|
import requests
|
|
from requests.auth import HTTPBasicAuth
|
|
import warnings
|
|
|
|
try:
|
|
import json
|
|
except ImportError:
|
|
import simplejson as json
|
|
|
|
|
|
class CloudFormsInventory(object):
|
|
def __init__(self):
|
|
"""
|
|
Main execution path
|
|
"""
|
|
self.inventory = dict() # A list of groups and the hosts in that group
|
|
self.hosts = dict() # Details about hosts in the inventory
|
|
|
|
# Parse CLI arguments
|
|
self.parse_cli_args()
|
|
|
|
# Read settings
|
|
self.read_settings()
|
|
|
|
# Cache
|
|
if self.args.refresh_cache or not self.is_cache_valid():
|
|
self.update_cache()
|
|
else:
|
|
self.load_inventory_from_cache()
|
|
self.load_hosts_from_cache()
|
|
|
|
data_to_print = ""
|
|
|
|
# Data to print
|
|
if self.args.host:
|
|
if self.args.debug:
|
|
print("Fetching host [%s]" % self.args.host)
|
|
data_to_print += self.get_host_info(self.args.host)
|
|
else:
|
|
self.inventory['_meta'] = {'hostvars': {}}
|
|
for hostname in self.hosts:
|
|
self.inventory['_meta']['hostvars'][hostname] = {
|
|
'cloudforms': self.hosts[hostname],
|
|
}
|
|
# include the ansible_ssh_host in the top level
|
|
if 'ansible_ssh_host' in self.hosts[hostname]:
|
|
self.inventory['_meta']['hostvars'][hostname]['ansible_ssh_host'] = self.hosts[hostname]['ansible_ssh_host']
|
|
|
|
data_to_print += self.json_format_dict(self.inventory, self.args.pretty)
|
|
|
|
print(data_to_print)
|
|
|
|
def is_cache_valid(self):
|
|
"""
|
|
Determines if the cache files have expired, or if it is still valid
|
|
"""
|
|
if self.args.debug:
|
|
print("Determining if cache [%s] is still valid (< %s seconds old)" % (self.cache_path_hosts, self.cache_max_age))
|
|
|
|
if os.path.isfile(self.cache_path_hosts):
|
|
mod_time = os.path.getmtime(self.cache_path_hosts)
|
|
current_time = time()
|
|
if (mod_time + self.cache_max_age) > current_time:
|
|
if os.path.isfile(self.cache_path_inventory):
|
|
if self.args.debug:
|
|
print("Cache is still valid!")
|
|
return True
|
|
|
|
if self.args.debug:
|
|
print("Cache is stale or does not exist.")
|
|
|
|
return False
|
|
|
|
def read_settings(self):
|
|
"""
|
|
Reads the settings from the cloudforms.ini file
|
|
"""
|
|
config = ConfigParser.SafeConfigParser()
|
|
config_paths = [
|
|
os.path.dirname(os.path.realpath(__file__)) + '/cloudforms.ini',
|
|
"/etc/ansible/cloudforms.ini",
|
|
]
|
|
|
|
env_value = os.environ.get('CLOUDFORMS_INI_PATH')
|
|
if env_value is not None:
|
|
config_paths.append(os.path.expanduser(os.path.expandvars(env_value)))
|
|
|
|
if self.args.debug:
|
|
for config_path in config_paths:
|
|
print("Reading from configuration file [%s]" % config_path)
|
|
|
|
config.read(config_paths)
|
|
|
|
# CloudForms API related
|
|
if config.has_option('cloudforms', 'url'):
|
|
self.cloudforms_url = config.get('cloudforms', 'url')
|
|
else:
|
|
self.cloudforms_url = None
|
|
|
|
if not self.cloudforms_url:
|
|
warnings.warn("No url specified, expected something like 'https://cfme.example.com'")
|
|
|
|
if config.has_option('cloudforms', 'username'):
|
|
self.cloudforms_username = config.get('cloudforms', 'username')
|
|
else:
|
|
self.cloudforms_username = None
|
|
|
|
if not self.cloudforms_username:
|
|
warnings.warn("No username specified, you need to specify a CloudForms username.")
|
|
|
|
if config.has_option('cloudforms', 'password'):
|
|
self.cloudforms_pw = config.get('cloudforms', 'password')
|
|
else:
|
|
self.cloudforms_pw = None
|
|
|
|
if not self.cloudforms_pw:
|
|
warnings.warn("No password specified, you need to specify a password for the CloudForms user.")
|
|
|
|
if config.has_option('cloudforms', 'ssl_verify'):
|
|
self.cloudforms_ssl_verify = config.getboolean('cloudforms', 'ssl_verify')
|
|
else:
|
|
self.cloudforms_ssl_verify = True
|
|
|
|
if config.has_option('cloudforms', 'version'):
|
|
self.cloudforms_version = config.get('cloudforms', 'version')
|
|
else:
|
|
self.cloudforms_version = None
|
|
|
|
if config.has_option('cloudforms', 'limit'):
|
|
self.cloudforms_limit = config.getint('cloudforms', 'limit')
|
|
else:
|
|
self.cloudforms_limit = 100
|
|
|
|
if config.has_option('cloudforms', 'purge_actions'):
|
|
self.cloudforms_purge_actions = config.getboolean('cloudforms', 'purge_actions')
|
|
else:
|
|
self.cloudforms_purge_actions = True
|
|
|
|
if config.has_option('cloudforms', 'clean_group_keys'):
|
|
self.cloudforms_clean_group_keys = config.getboolean('cloudforms', 'clean_group_keys')
|
|
else:
|
|
self.cloudforms_clean_group_keys = True
|
|
|
|
if config.has_option('cloudforms', 'nest_tags'):
|
|
self.cloudforms_nest_tags = config.getboolean('cloudforms', 'nest_tags')
|
|
else:
|
|
self.cloudforms_nest_tags = False
|
|
|
|
# Ansible related
|
|
try:
|
|
group_patterns = config.get('ansible', 'group_patterns')
|
|
except (ConfigParser.NoOptionError, ConfigParser.NoSectionError):
|
|
group_patterns = "[]"
|
|
|
|
self.group_patterns = eval(group_patterns)
|
|
|
|
# Cache related
|
|
try:
|
|
cache_path = os.path.expanduser(config.get('cache', 'path'))
|
|
except (ConfigParser.NoOptionError, ConfigParser.NoSectionError):
|
|
cache_path = '.'
|
|
(script, ext) = os.path.splitext(os.path.basename(__file__))
|
|
self.cache_path_hosts = cache_path + "/%s.hosts" % script
|
|
self.cache_path_inventory = cache_path + "/%s.inventory" % script
|
|
self.cache_max_age = config.getint('cache', 'max_age')
|
|
|
|
if self.args.debug:
|
|
print("CloudForms settings:")
|
|
print("cloudforms_url = %s" % self.cloudforms_url)
|
|
print("cloudforms_username = %s" % self.cloudforms_username)
|
|
print("cloudforms_pw = %s" % self.cloudforms_pw)
|
|
print("cloudforms_ssl_verify = %s" % self.cloudforms_ssl_verify)
|
|
print("cloudforms_version = %s" % self.cloudforms_version)
|
|
print("cloudforms_limit = %s" % self.cloudforms_limit)
|
|
print("cloudforms_purge_actions = %s" % self.cloudforms_purge_actions)
|
|
print("Cache settings:")
|
|
print("cache_max_age = %s" % self.cache_max_age)
|
|
print("cache_path_hosts = %s" % self.cache_path_hosts)
|
|
print("cache_path_inventory = %s" % self.cache_path_inventory)
|
|
|
|
def parse_cli_args(self):
|
|
"""
|
|
Command line argument processing
|
|
"""
|
|
parser = argparse.ArgumentParser(description='Produce an Ansible Inventory file based on CloudForms managed VMs')
|
|
parser.add_argument('--list', action='store_true', default=True, help='List instances (default: True)')
|
|
parser.add_argument('--host', action='store', help='Get all the variables about a specific instance')
|
|
parser.add_argument('--pretty', action='store_true', default=False, help='Pretty print JSON output (default: False)')
|
|
parser.add_argument('--refresh-cache', action='store_true', default=False,
|
|
help='Force refresh of cache by making API requests to CloudForms (default: False - use cache files)')
|
|
parser.add_argument('--debug', action='store_true', default=False, help='Show debug output while running (default: False)')
|
|
self.args = parser.parse_args()
|
|
|
|
def _get_json(self, url):
|
|
"""
|
|
Make a request and return the JSON
|
|
"""
|
|
results = []
|
|
|
|
ret = requests.get(url,
|
|
auth=HTTPBasicAuth(self.cloudforms_username, self.cloudforms_pw),
|
|
verify=self.cloudforms_ssl_verify)
|
|
|
|
ret.raise_for_status()
|
|
|
|
try:
|
|
results = json.loads(ret.text)
|
|
except ValueError:
|
|
warnings.warn("Unexpected response from {0} ({1}): {2}".format(self.cloudforms_url, ret.status_code, ret.reason))
|
|
results = {}
|
|
|
|
if self.args.debug:
|
|
print("=======================================================================")
|
|
print("=======================================================================")
|
|
print("=======================================================================")
|
|
print(ret.text)
|
|
print("=======================================================================")
|
|
print("=======================================================================")
|
|
print("=======================================================================")
|
|
|
|
return results
|
|
|
|
def _get_hosts(self):
|
|
"""
|
|
Get all hosts by paging through the results
|
|
"""
|
|
limit = self.cloudforms_limit
|
|
|
|
page = 0
|
|
last_page = False
|
|
|
|
results = []
|
|
|
|
while not last_page:
|
|
offset = page * limit
|
|
ret = self._get_json("%s/api/vms?offset=%s&limit=%s&expand=resources,tags,hosts,&attributes=ipaddresses" % (self.cloudforms_url, offset, limit))
|
|
results += ret['resources']
|
|
if ret['subcount'] < limit:
|
|
last_page = True
|
|
page += 1
|
|
|
|
return results
|
|
|
|
def update_cache(self):
|
|
"""
|
|
Make calls to cloudforms and save the output in a cache
|
|
"""
|
|
self.groups = dict()
|
|
self.hosts = dict()
|
|
|
|
if self.args.debug:
|
|
print("Updating cache...")
|
|
|
|
for host in self._get_hosts():
|
|
# Ignore VMs that are not powered on
|
|
if host['power_state'] != 'on':
|
|
if self.args.debug:
|
|
print("Skipping %s because power_state = %s" % (host['name'], host['power_state']))
|
|
continue
|
|
|
|
# purge actions
|
|
if self.cloudforms_purge_actions and 'actions' in host:
|
|
del host['actions']
|
|
|
|
# Create ansible groups for tags
|
|
if 'tags' in host:
|
|
|
|
# Create top-level group
|
|
if 'tags' not in self.inventory:
|
|
self.inventory['tags'] = dict(children=[], vars={}, hosts=[])
|
|
|
|
if not self.cloudforms_nest_tags:
|
|
# don't expand tags, just use them in a safe way
|
|
for group in host['tags']:
|
|
# Add sub-group, as a child of top-level
|
|
safe_key = self.to_safe(group['name'])
|
|
if safe_key:
|
|
if self.args.debug:
|
|
print("Adding sub-group '%s' to parent 'tags'" % safe_key)
|
|
|
|
if safe_key not in self.inventory['tags']['children']:
|
|
self.push(self.inventory['tags'], 'children', safe_key)
|
|
|
|
self.push(self.inventory, safe_key, host['name'])
|
|
|
|
if self.args.debug:
|
|
print("Found tag [%s] for host which will be mapped to [%s]" % (group['name'], safe_key))
|
|
else:
|
|
# expand the tags into nested groups / sub-groups
|
|
# Create nested groups for tags
|
|
safe_parent_tag_name = 'tags'
|
|
for tag in host['tags']:
|
|
tag_hierarchy = tag['name'][1:].split('/')
|
|
|
|
if self.args.debug:
|
|
print("Working on list %s" % tag_hierarchy)
|
|
|
|
for tag_name in tag_hierarchy:
|
|
if self.args.debug:
|
|
print("Working on tag_name = %s" % tag_name)
|
|
|
|
safe_tag_name = self.to_safe(tag_name)
|
|
if self.args.debug:
|
|
print("Using sanitized name %s" % safe_tag_name)
|
|
|
|
# Create sub-group
|
|
if safe_tag_name not in self.inventory:
|
|
self.inventory[safe_tag_name] = dict(children=[], vars={}, hosts=[])
|
|
|
|
# Add sub-group, as a child of top-level
|
|
if safe_parent_tag_name:
|
|
if self.args.debug:
|
|
print("Adding sub-group '%s' to parent '%s'" % (safe_tag_name, safe_parent_tag_name))
|
|
|
|
if safe_tag_name not in self.inventory[safe_parent_tag_name]['children']:
|
|
self.push(self.inventory[safe_parent_tag_name], 'children', safe_tag_name)
|
|
|
|
# Make sure the next one uses this one as it's parent
|
|
safe_parent_tag_name = safe_tag_name
|
|
|
|
# Add the host to the last tag
|
|
self.push(self.inventory[safe_parent_tag_name], 'hosts', host['name'])
|
|
|
|
# Set ansible_ssh_host to the first available ip address
|
|
if 'ipaddresses' in host and host['ipaddresses'] and isinstance(host['ipaddresses'], list):
|
|
host['ansible_ssh_host'] = host['ipaddresses'][0]
|
|
|
|
# Create additional groups
|
|
for key in ('location', 'type', 'vendor'):
|
|
safe_key = self.to_safe(host[key])
|
|
|
|
# Create top-level group
|
|
if key not in self.inventory:
|
|
self.inventory[key] = dict(children=[], vars={}, hosts=[])
|
|
|
|
# Create sub-group
|
|
if safe_key not in self.inventory:
|
|
self.inventory[safe_key] = dict(children=[], vars={}, hosts=[])
|
|
|
|
# Add sub-group, as a child of top-level
|
|
if safe_key not in self.inventory[key]['children']:
|
|
self.push(self.inventory[key], 'children', safe_key)
|
|
|
|
if key in host:
|
|
# Add host to sub-group
|
|
self.push(self.inventory[safe_key], 'hosts', host['name'])
|
|
|
|
self.hosts[host['name']] = host
|
|
self.push(self.inventory, 'all', host['name'])
|
|
|
|
if self.args.debug:
|
|
print("Saving cached data")
|
|
|
|
self.write_to_cache(self.hosts, self.cache_path_hosts)
|
|
self.write_to_cache(self.inventory, self.cache_path_inventory)
|
|
|
|
def get_host_info(self, host):
|
|
"""
|
|
Get variables about a specific host
|
|
"""
|
|
if not self.hosts or len(self.hosts) == 0:
|
|
# Need to load cache from cache
|
|
self.load_hosts_from_cache()
|
|
|
|
if host not in self.hosts:
|
|
if self.args.debug:
|
|
print("[%s] not found in cache." % host)
|
|
|
|
# try updating the cache
|
|
self.update_cache()
|
|
|
|
if host not in self.hosts:
|
|
if self.args.debug:
|
|
print("[%s] does not exist after cache update." % host)
|
|
# host might not exist anymore
|
|
return self.json_format_dict({}, self.args.pretty)
|
|
|
|
return self.json_format_dict(self.hosts[host], self.args.pretty)
|
|
|
|
def push(self, d, k, v):
|
|
"""
|
|
Safely puts a new entry onto an array.
|
|
"""
|
|
if k in d:
|
|
d[k].append(v)
|
|
else:
|
|
d[k] = [v]
|
|
|
|
def load_inventory_from_cache(self):
|
|
"""
|
|
Reads the inventory from the cache file sets self.inventory
|
|
"""
|
|
cache = open(self.cache_path_inventory, 'r')
|
|
json_inventory = cache.read()
|
|
self.inventory = json.loads(json_inventory)
|
|
|
|
def load_hosts_from_cache(self):
|
|
"""
|
|
Reads the cache from the cache file sets self.hosts
|
|
"""
|
|
cache = open(self.cache_path_hosts, 'r')
|
|
json_cache = cache.read()
|
|
self.hosts = json.loads(json_cache)
|
|
|
|
def write_to_cache(self, data, filename):
|
|
"""
|
|
Writes data in JSON format to a file
|
|
"""
|
|
json_data = self.json_format_dict(data, True)
|
|
cache = open(filename, 'w')
|
|
cache.write(json_data)
|
|
cache.close()
|
|
|
|
def to_safe(self, word):
|
|
"""
|
|
Converts 'bad' characters in a string to underscores so they can be used as Ansible groups
|
|
"""
|
|
if self.cloudforms_clean_group_keys:
|
|
regex = "[^A-Za-z0-9\_]"
|
|
return re.sub(regex, "_", word.replace(" ", ""))
|
|
else:
|
|
return word
|
|
|
|
def json_format_dict(self, data, pretty=False):
|
|
"""
|
|
Converts a dict to a JSON object and dumps it as a formatted string
|
|
"""
|
|
if pretty:
|
|
return json.dumps(data, sort_keys=True, indent=2)
|
|
else:
|
|
return json.dumps(data)
|
|
|
|
CloudFormsInventory()
|