Created
September 11, 2018 14:33
-
-
Save versionsix/2e07fa43e50ffe0f31c1eece9a51763b to your computer and use it in GitHub Desktop.
Vagrant ansible dynamic inventory
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
# Adapted from Mark Mandel's implementation | |
# https://github.com/ansible/ansible/blob/devel/plugins/inventory/vagrant.py | |
import argparse | |
import json | |
import paramiko | |
import subprocess | |
import sys | |
def parse_args(): | |
parser = argparse.ArgumentParser(description="Vagrant inventory script") | |
group = parser.add_mutually_exclusive_group(required=True) | |
group.add_argument('--list', action='store_true') | |
group.add_argument('--host') | |
return parser.parse_args() | |
def list_running_hosts(): | |
cmd = "vagrant status --machine-readable" | |
status = subprocess.check_output(cmd.split()).rstrip() | |
hosts = [] | |
for line in status.split('\n'): | |
(_, host, key, value) = line.split(',',3) | |
if key == 'state' and value == 'running': | |
hosts.append(host) | |
return hosts | |
def get_host_details(host): | |
cmd = "vagrant ssh-config {}".format(host) | |
p = subprocess.Popen(cmd.split(), stdout=subprocess.PIPE) | |
config = paramiko.SSHConfig() | |
config.parse(p.stdout) | |
c = config.lookup(host) | |
return {'ansible_ssh_host': c['hostname'], | |
'ansible_ssh_port': c['port'], | |
'ansible_ssh_user': c['user'], | |
'ansible_ssh_private_key_file': c['identityfile'][0]} | |
def main(): | |
args = parse_args() | |
if args.list: | |
hosts = list_running_hosts() | |
json.dump({'vagrant': hosts}, sys.stdout) | |
else: | |
details = get_host_details(args.host) | |
json.dump(details, sys.stdout) | |
if __name__ == '__main__': | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment