create_deployments.py

import json
import os
import sys
import time
from subprocess import call

from cloudify_cli import utils
from cloudify_rest_client.executions import Execution
from scp import scp


management_ip = utils.get_management_server_ip()
client = utils.get_rest_client(management_ip)

deployments_json = sys.stdin.read()
deployments = json.loads(deployments_json)

host_magic_dir = '/tmp/cloudify_migration_data_2g25qt4/'
magic_file = '/tmp/cloudify_migration_data_5ovg6bht'

del_template = ("curl -s -XDELETE 'http://localhost:9200/"
                "cloudify_storage/node_instance/_query?q=deployment_id:{} '")
bulk_template = ("curl -s XPOST 'http://localhost:9200/"
                 "{index}/_bulk' --data-binary @{file}")
update_exec_workflow_id_template = (
    "curl -s XPOST 'http://localhost:9200/cloudify_storage/execution/"
    "{execution_id}/_update' -d '{{\"doc\":{{\"workflow_id\":\"{w_id}\"}}}}'"
)


with open(os.devnull, 'w') as FNULL:
    create_deployment_executions = []
    for dep in deployments:
        new_dep = client.deployments.create(
            dep['blueprint_id'],
            dep['id'],
            dep['inputs']
        )

        del_command = del_template.format(dep['id'])
        call(['cfy', 'ssh', '-c', del_command],
             stdout=FNULL, stderr=FNULL)

        create_dep_execution = client.executions.list(
            deployment_id=new_dep.id
        )[0]
        create_deployment_executions.append(create_dep_execution.id)
        call(['cfy', 'ssh', '-c', update_exec_workflow_id_template.format(
            execution_id=create_dep_execution.id,
            w_id='create_deployment_environment_' + os.environ['NEW_MANAGER_VER']
        )], stdout=FNULL, stderr=FNULL)

        print 'Recreated deployment %s' % (new_dep['id'],)

    scp(host_magic_dir + 'migration_deps_storage', magic_file, True)
    call(['cfy', 'ssh', '-c',  bulk_template.format(
        file=magic_file,
        index='cloudify_storage'
        )], stdout=FNULL, stderr=FNULL)

    scp(host_magic_dir + 'migration_deps_events', magic_file, True)
    call(['cfy', 'ssh', '-c',  bulk_template.format(
        file=magic_file,
        index='cloudify_events'
        )], stdout=FNULL, stderr=FNULL)

    for execution_id in create_deployment_executions:
        print 'Waiting for execution {0}'.format(execution_id)
        execution = client.executions.get(execution_id)
        while execution.status not in Execution.END_STATES:
            time.sleep(2)
            execution = client.executions.get(execution_id)
        if execution.status != Execution.TERMINATED:
            print 'Execution {0} failed.'.format(execution_id)
            sys.exit(1)
        else:
            print 'Execution {0} finished.'.format(execution_id)
print 'Deployments migrated.'