#!/usr/bin/python

import subprocess,json,time


def create_servers(count, auth, imageid, sshid):

        for num in range(0,count):
                o1 = subprocess.check_output('''curl -s -X POST -H 'Content-Type: application/json' -H 'Authorization: Bearer %s' -d '{"name":"%s","region":"ams3","size":"2gb","image":"%s","ssh_keys":[%s],"backups":false,"ipv6":false,"user_data":null,"private_networking":null}' "https://api.digitalocean.com/v2/droplets"''' % (auth, num, imageid, sshid), shell=True)

        return

def get_ips(auth):
        o = subprocess.check_output('''curl -s -X GET -H 'Content-Type: application/json' -H 'Authorization: Bearer %s' "https://api.digitalocean.com/v2/droplets"''' % auth, shell=True)
        p = json.loads(o)

        ips = []
        for num in range(0,len(p['droplets'])):
                ips.append(p['droplets'][num]['networks']['v4'][0]['ip_address'])

        return ips

def ssh_and_run(ips,chn):

        for num in range(0,len(ips)):
                o = subprocess.check_output('''ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no root@%s ./startSOP %s''' % (ips[num],chn), shell=True)
                time.sleep(5)