NewsBlur/utils/hostname_ssh.py

import sys
import os
import digitalocean
from django.conf import settings

sys.path.append('/srv/newsblur')

os.environ['DJANGO_SETTINGS_MODULE'] = 'newsblur_web.settings'

if __name__ == '__main__':
    # Check and clean second argument (ex: sshdo task 2)
    second_arg = sys.argv[2] if len(sys.argv) > 2 else "1"
    droplet_index = int(second_arg) if second_arg.isnumeric() else 1

    # Use correct Digital Ocean team based on "old"
    doapi = digitalocean.Manager(token=settings.DO_TOKEN_SSH)
    if second_arg == "old":
        doapi = digitalocean.Manager(token=settings.DO_TOKEN_FABRIC)

    # Cycle through droplets finding match
    droplets = doapi.get_all_droplets()
    for droplet in droplets:
        if droplet.name.startswith(sys.argv[1]):
            if droplet_index > 1:
                droplet_index -= 1
                continue
            print(droplet.ip_address)
            break