mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
832dd55144
* sanity: Add future boilerplate Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com> * Module Utils Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com> * Scripts Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com> * sanity: Add future boilerplate * Tests Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com> * CI failure Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com>
101 lines
2.4 KiB
Python
101 lines
2.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
# (c) 2015, Marc Abramowitz <marca@surveymonkey.com>
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
# Dynamic inventory script which lets you use nodes discovered by Serf
|
|
# (https://serfdom.io/).
|
|
#
|
|
# Requires the `serfclient` Python module from
|
|
# https://pypi.org/project/serfclient/
|
|
#
|
|
# Environment variables
|
|
# ---------------------
|
|
# - `SERF_RPC_ADDR`
|
|
# - `SERF_RPC_AUTH`
|
|
#
|
|
# These variables are described at https://www.serfdom.io/docs/commands/members.html#_rpc_addr
|
|
|
|
import argparse
|
|
import collections
|
|
import os
|
|
import sys
|
|
|
|
# https://pypi.org/project/serfclient/
|
|
from serfclient import SerfClient, EnvironmentConfig
|
|
|
|
import json
|
|
|
|
_key = 'serf'
|
|
|
|
|
|
def _serf_client():
|
|
env = EnvironmentConfig()
|
|
return SerfClient(host=env.host, port=env.port, rpc_auth=env.auth_key)
|
|
|
|
|
|
def get_serf_members_data():
|
|
return _serf_client().members().body['Members']
|
|
|
|
|
|
def get_nodes(data):
|
|
return [node['Name'] for node in data]
|
|
|
|
|
|
def get_groups(data):
|
|
groups = collections.defaultdict(list)
|
|
|
|
for node in data:
|
|
for key, value in node['Tags'].items():
|
|
groups[value].append(node['Name'])
|
|
|
|
return groups
|
|
|
|
|
|
def get_meta(data):
|
|
meta = {'hostvars': {}}
|
|
for node in data:
|
|
meta['hostvars'][node['Name']] = node['Tags']
|
|
return meta
|
|
|
|
|
|
def print_list():
|
|
data = get_serf_members_data()
|
|
nodes = get_nodes(data)
|
|
groups = get_groups(data)
|
|
meta = get_meta(data)
|
|
inventory_data = {_key: nodes, '_meta': meta}
|
|
inventory_data.update(groups)
|
|
print(json.dumps(inventory_data))
|
|
|
|
|
|
def print_host(host):
|
|
data = get_serf_members_data()
|
|
meta = get_meta(data)
|
|
print(json.dumps(meta['hostvars'][host]))
|
|
|
|
|
|
def get_args(args_list):
|
|
parser = argparse.ArgumentParser(
|
|
description='ansible inventory script reading from serf cluster')
|
|
mutex_group = parser.add_mutually_exclusive_group(required=True)
|
|
help_list = 'list all hosts from serf cluster'
|
|
mutex_group.add_argument('--list', action='store_true', help=help_list)
|
|
help_host = 'display variables for a host'
|
|
mutex_group.add_argument('--host', help=help_host)
|
|
return parser.parse_args(args_list)
|
|
|
|
|
|
def main(args_list):
|
|
args = get_args(args_list)
|
|
if args.list:
|
|
print_list()
|
|
if args.host:
|
|
print_host(args.host)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv[1:])
|