summaryrefslogtreecommitdiffstats
path: root/utils/infra_setup/heat/manager.py
blob: 1a360b789fe055657a45674483df3056fa5f1846 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
##############################################################################
# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################

import time
import common as op_utils
from glanceclient.client import Client as GlanceClient
from novaclient.client import Client as NovaClient
from neutronclient.v2_0.client import Client as NeutronClient


def _get_glance_client():
    sess = op_utils.get_session()
    return GlanceClient(
        op_utils.get_glance_api_version(),
        session=sess)


def _get_nova_client():
    sess = op_utils.get_session()

    return NovaClient(
        op_utils.get_nova_api_version(),
        session=sess)


def _get_neutron_client():
    sess = op_utils.get_session()
    neutron_client = NeutronClient(session=sess)
    return neutron_client


def create_images(
        imagefile=None,
        image_name="bottlenecks_image"):
    print "========== Create image in OS =========="

    if imagefile is None:
        print "imagefile not set/found"
        return False

    glance = _get_glance_client()
    image = glance.images.create(
        name=image_name,
        disk_format="qcow2",
        container_format="bare")
    with open(imagefile) as fimage:
        glance.images.upload(image.id, fimage)

    timeInQueue = 0
    img_status = image.status
    while img_status == "queued" and timeInQueue < 30:
        print "  image's status: " + img_status
        time.sleep(1)
        timeInQueue = timeInQueue + 1
        img_status = glance.images.get(image.id).status

    print "After %d seconds,image status is [%s]" % (timeInQueue, img_status)
    return True if img_status == "active" else False


def stack_create_keypairs(key_path, name="bottlenecks_keypair"):
    print "========== Add keypairs in OS =========="
    nova = _get_nova_client()
    with open(key_path) as pkey:
        nova.keypairs.create(name=name, public_key=pkey.read())


def stack_create_flavors(
        name="bottlenecks_flavor",
        ram=4096,
        vcpus=2,
        disk=10):
    print "========== Create flavors in OS =========="
    nova = _get_nova_client()
    nova.flavors.create(name=name, ram=ram, vcpus=vcpus, disk=disk)