main.py

import os

from utils import pricing_util
from utils.aws_spot_instance import AWSSpotInstance
from utils.aws_spot_exception import SpotConstraintException
import aws_spot_bot.user_config as uconf


def launch_instances(qty):
    """Launches QTY instances and returns the instance objects."""
    best_az = pricing_util.get_best_az()
    launched_instances = []
    print "Best availability zone:", best_az.name

    for idx in range(qty):
        print '>> Launching instance #%s' % idx
        si = AWSSpotInstance(best_az.region, best_az.name, uconf.INSTANCE_TYPES[0], uconf.AMI_ID, uconf.BID)
        si.request_instance()
        try:
            si.get_ip()
        except SpotConstraintException, e:
            print(">> ", e.message)
            si.cancel_spot_request()
            continue
        launched_instances.append(si)

    return launched_instances


if __name__ == '__main__':
    instances = launch_instances(uconf.QTY_INSTANCES)

    for si in instances:
        if uconf.WAIT_FOR_HTTP:
            si.wait_for_http()
        if uconf.WAIT_FOR_SSH:
            si.wait_for_ssh()
        if uconf.OPEN_IN_BROWSER:
            si.open_in_browser()
        if uconf.OPEN_SSH:
             si.open_ssh_term()
        if uconf.ADD_TO_ANSIBLE_HOSTS:
            si.add_to_ansible_hosts()

    if uconf.RUN_ANSIBLE:
        os.system('cd ansible && ansible-playbook -s play.yml')