zuul-jobs/roles/write-inventory/library/write_inventory.py

92 lines
2.4 KiB
Python
Executable File

#!/usr/bin/env python3
#
# Copyright 2018 Red Hat, Inc
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import json
from ansible.module_utils.basic import AnsibleModule
# The list of variables we might include
VARS = [
'ansible_python_interpreter',
'ansible_connection',
'ansible_host',
'ansible_port',
'ansible_user'
]
def run(dest, hostvars, groups, include, exclude):
children = {}
for group, hostnames in groups.items():
if group == 'all' or group == 'ungrouped':
continue
children[group] = {}
children[group]['hosts'] = {}
for host in hostnames:
children[group]['hosts'][host] = None
out_all = {}
out = {
'all': {
'children': children,
'hosts': out_all
}
}
for host, hvars in hostvars.items():
d = {}
for v in VARS:
if v not in hvars:
continue
if include is not None:
if v not in include:
continue
if exclude is not None:
if v in exclude:
continue
d[v] = hvars[v]
out_all[host] = d
with open(dest, 'w') as f:
f.write(json.dumps(out))
def ansible_main():
module = AnsibleModule(
argument_spec=dict(
dest=dict(required=True, type='path'),
hostvars=dict(required=True, type='raw'),
groups=dict(required=True, type='raw'),
include_hostvars=dict(type='list'),
exclude_hostvars=dict(type='list'),
)
)
p = module.params
dest = p.get('dest')
hostvars = p.get('hostvars')
groups = p.get('groups')
include = p.get('include_hostvars')
exclude = p.get('exclude_hostvars')
run(dest, hostvars, groups, include, exclude)
module.exit_json(changed=True)
if __name__ == '__main__':
ansible_main()