system-config/playbooks/roles/nodepool-base/library/make_nodepool_zk_hosts.py

49 lines
1.4 KiB
Python

#!/usr/bin/env python3
#
# Copyright 2020 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.
from ansible.module_utils.basic import AnsibleModule
def main():
module = AnsibleModule(
argument_spec=dict(
hostvars=dict(required=True, type='dict'),
zk_group=dict(required=True, type='list'),
)
)
p = module.params
zk_hosts = []
try:
for host in p['zk_group']:
# Prefer public_v6 if set, otherwise ansible_host
addr = None
if 'public_v6' in p['hostvars'][host]:
addr = p['hostvars'][host]['public_v6']
if not addr:
addr = p['hostvars'][host]['ansible_host']
zk_hosts.append(dict(
host=addr,
port=2281
))
module.exit_json(hosts=zk_hosts, changed=True)
except Exception as e:
module.fail_json(msg=str(e), changed=True)
if __name__ == '__main__':
main()