123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113 |
- #!/usr/bin/env python3
- #
- # Copyright (c) 2017-2020 Joe Clarke <jclarke@cisco.com>
- # All rights reserved.
- #
- # Redistribution and use in source and binary forms, with or without
- # modification, are permitted provided that the following conditions
- # are met:
- # 1. Redistributions of source code must retain the above copyright
- # notice, this list of conditions and the following disclaimer.
- # 2. Redistributions in binary form must reproduce the above copyright
- # notice, this list of conditions and the following disclaimer in the
- # documentation and/or other materials provided with the distribution.
- #
- # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
- # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- # ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
- # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- # SUCH DAMAGE.
- import sys
- import re
- import subprocess
- import os
- import json
- import argparse
- from yaml import load, dump
- try:
- from yaml import CLoader as Loader, CDumper as Dumper
- except ImportError:
- from yaml import Loader, Dumper
- def main():
- parser = argparse.ArgumentParser(prog=sys.argv[0], description="Convert Ansible hosts file to JSON")
- parser.add_argument("--output-file", "-o", metavar="<OUTPUT_FILE_PATH>", help="Path to the file to store the JSON", required=True)
- parser.add_argument(
- "--playbook",
- "-p",
- metavar="<PLAYBOOK_NAME>",
- help="Name of playbook to use to generate file (default: add-to-librenms-playbook.yml)",
- default="add-to-librenms-playbook.yml",
- )
- parser.add_argument(
- "--format", "-f", metavar="<JSON|YAML>", help="Output format (default: JSON)", choices=["YAML", "JSON"], default="JSON"
- )
- parser.add_argument("--limit", metavar="<LIMIT_STRING>", help="Optional set of hosts or groups to limit")
- args = parser.parse_args()
- os.environ["ANSIBLE_FORCE_COLOR"] = "True"
- os.environ["ANSIBLE_HOST_KEY_CHECKING"] = "False"
- os.environ["ANSIBLE_PERSISTENT_COMMAND_TIMEOUT"] = "300"
- command = [
- "ansible-playbook",
- "-i",
- "inventory/hosts",
- "--list-hosts",
- "-e",
- "ansible_python_interpreter={}".format(sys.executable),
- args.playbook,
- ]
- if args.limit:
- command += ["--limit", args.limit]
- p = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
- hosts = ""
- for c in iter(lambda: p.stdout.read(1), b""):
- hosts += c.decode("utf-8")
- p.poll()
- hlist = []
- reading_hosts = False
- for h in hosts.split("\n"):
- if not reading_hosts and re.search(r"hosts \(\d+\):", h):
- reading_hosts = True
- continue
- if not reading_hosts:
- continue
- h = h.strip()
- if h == "":
- continue
- hlist.append(h.strip())
- fd = None
- if args.output_file == "-":
- fd = sys.stdout
- else:
- fd = open(args.output_file, "w")
- if args.format == "JSON":
- json.dump(hlist, fd, indent=4)
- else:
- dump(hlist, fd, Dumper=Dumper)
- fd.write("\n")
- if args.output_file == "-":
- fd.flush()
- else:
- fd.close()
- if __name__ == "__main__":
- main()
|