2021-11-14 18:43:10 -05:00
|
|
|
#!/usr/bin/env python3
|
2013-05-20 12:03:04 -07:00
|
|
|
import sys
|
|
|
|
|
import os
|
2021-03-03 11:15:09 -05:00
|
|
|
import subprocess
|
|
|
|
|
import json
|
2015-09-21 16:13:09 -07:00
|
|
|
import digitalocean
|
2021-11-14 18:43:10 -05:00
|
|
|
# from django.conf import settings
|
2013-05-20 12:03:04 -07:00
|
|
|
|
2013-05-20 12:57:45 -07:00
|
|
|
sys.path.append('/srv/newsblur')
|
2013-05-20 12:03:04 -07:00
|
|
|
|
2020-10-13 22:10:26 +07:00
|
|
|
os.environ['DJANGO_SETTINGS_MODULE'] = 'newsblur_web.settings'
|
2013-05-20 12:03:04 -07:00
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
2021-02-18 13:51:06 -05:00
|
|
|
# Check and clean second argument (ex: sshdo task 2)
|
|
|
|
|
second_arg = sys.argv[2] if len(sys.argv) > 2 else "1"
|
2021-11-14 18:43:10 -05:00
|
|
|
droplet_index = int(second_arg) if str(second_arg).isnumeric() else 1
|
2021-03-03 11:15:09 -05:00
|
|
|
droplet_name = sys.argv[1]
|
2021-02-18 13:51:06 -05:00
|
|
|
|
|
|
|
|
# Use correct Digital Ocean team based on "old"
|
2021-11-14 18:43:10 -05:00
|
|
|
commands = ['ansible-inventory', '--list']
|
|
|
|
|
env = None
|
2021-02-18 13:51:06 -05:00
|
|
|
if second_arg == "old":
|
2021-11-14 18:43:10 -05:00
|
|
|
env = dict(os.environ, ANSIBLE_CONFIG="ansible.old.cfg")
|
|
|
|
|
hosts = subprocess.check_output(commands, env=env)
|
|
|
|
|
if not hosts:
|
|
|
|
|
print(" ***> Could not load ansible-inventory!")
|
2021-02-18 13:51:06 -05:00
|
|
|
|
2021-11-14 18:43:10 -05:00
|
|
|
hosts = json.loads(hosts)
|
|
|
|
|
for host, ip_host in hosts['_meta']['hostvars'].items():
|
|
|
|
|
if host.startswith(droplet_name):
|
|
|
|
|
print(ip_host['ansible_host'])
|
|
|
|
|
break
|
