#!/usr/bin/python
##############################################################################
# Copyright (c) 2016 ZTE Coreporation and others.
# hu.zhijiang@zte.com.cn
# lu.yao135@zte.com.cn
# 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
##############################################################################
from oslo_config import cfg
import sys
from daisyclient.v1 import client as daisy_client
import get_conf
import traceback
import time
import os
daisy_version = 1.0
daisyrc_path = "/root/daisyrc_admin"
iso_path = "/var/lib/daisy/kolla/"
deployment_interface = "ens3"
cluster_name = "clustertest"
_CLI_OPTS = [
cfg.StrOpt('dha',
help='The dha file path'),
cfg.StrOpt('network',
help='The network file path'),
cfg.StrOpt('cluster',
help='Config cluster'),
cfg.StrOpt('host',
help='Config host'),
cfg.StrOpt('install',
help='install daisy'),
cfg.IntOpt('isbare',
help='deploy environment'),
]
# ------------------------------------------------------------------------------------------
def parse(conf, args):
conf.register_cli_opts(_CLI_OPTS)
conf(args=args)
def print_bar(msg):
print ("--------------------------------------------")
print (msg)
print ("--------------------------------------------")
def get_endpoint(file_path):
for line in open(file_path):
if 'OS_ENDPOINT' in line:
daisyrc_admin_line = line.strip()
daisy_endpoint = daisyrc_admin_line.split("=")[1]
return daisy_endpoint
daisy_endpoint = get_endpoint(daisyrc_path)
client = daisy_client.Client(version=daisy_version, endpoint=daisy_endpoint)
def prepare_install():
try:
print("get config...")
conf = cfg.ConfigOpts()
parse(conf, sys.argv[1:])
host_interface_map, hosts_name, network_map, vip = \
get_conf.config(conf['dha'], conf['network'])
if conf['cluster'] and conf['cluster'] == 'yes':
print("add cluster...")
cluster_meta = {'name': cluster_name, 'description': '',
'target_systems': 'os+kolla'}
clusters_info = client.clusters.add(**cluster_meta)
cluster_id = clusters_info.id
print("cluster_id=%s." % cluster_id)
print("update network...")
update_network(cluster_id, network_map)
print("build pxe...")
build_pxe_for_discover(cluster_id)
elif conf['host'] and conf['host'] == 'yes':
print("discover host...")
discover_host(hosts_name)
print("update hosts interface...")
hosts_info = get_hosts()
cluster_info = get_cluster()
cluster_id = cluster_info.id
add_hosts_interface(cluster_id, hosts_info, hosts_name,
host_interface_map, vip)
if 'isbare' in conf and conf['isbare'] == 0:
install_os_for_vm_step1(cluster_id)
else:
print("daisy baremetal deploy start")
install_os_for_bm_oneshot(cluster_id)
elif conf['install'] and conf['install'] == 'yes':
install_os_for_vm_step2(cluster_id)
except Exception:
print("Deploy failed!!!.%s." % traceback.format_exc())
else:
print_bar("Everything is done!")
def build_pxe_for_discover(cluster_id):
cluster_meta = {'cluster_id': cluster_id,
'deployment_interface': deployment_interface}
client.install.install(**cluster_meta)
def install_os_for_vm_step1(cluster_id):
cluster_meta = {'cluster_id': cluster_id,
'pxe_only': "true"}
client.install.install(**cluster_meta)
def install_os_for_bm_oneshot(cluster_id):
cluster_meta = {'cluster_id': cluster_id,
'pxe_only': "false",
'skip_pxe_ipmi': "false"}
client.install.install(**cluster_meta)
def install_os_for_vm_step2(cluster_id):
cluster_meta = {'cluster_id': cluster_id,
'skip_pxe_ipmi': "true"}
client.install.install(**cluster_meta)
def discover_host(hosts_name):
while True:
hosts_info = get_hosts()
if len(hosts_info) == len(hosts_name):
print('discover hosts success!')
break
else:
time.sleep(10)
def update_network(cluster_id, network_map):
network_meta = {'filters': {'cluster_id': cluster_id}}
network_info_gernerator = client.networks.list(**network_meta)
network_info_list = [net for net in network_info_gernerator]
for net in network_info_list:
network_id = net.id
network_name = net.name
if network_map.get(network_name):
network_meta = network_map[network_name]
client.networks.update(network_id, **network_meta)
def get_hosts():
hosts_list_generator = client.hosts.list()
hosts_list = [host for host in hosts_list_generator]
hosts_info = []
for host in hosts_list:
host_info = client.hosts.get(host.id)
hosts_info.append(host_info)
return hosts_info
def get_cluster():
cluster_list_generator = client.clusters.list()
cluster_list = [cluster for cluster in cluster_list_generator]
for cluster in cluster_list:
cluster_info = client.clusters.get(cluster.id)
return cluster_info
def add_hosts_interface(cluster_id, hosts_info, hosts_name, host_interface_map,
vip):
for host_name, host in zip(hosts_name, hosts_info):
host = host.to_dict()
host['cluster'] = cluster_id
host['ipmi_user'] = 'zteroot'
host['ipmi_passwd'] = 'superuser'
for interface in host['interfaces']:
interface_name = interface['name']
if interface_name in host_interface_map:
interface['assigned_networks'] = \
host_interface_map[interface_name]
pathlist = os.listdir(iso_path)
for filename in pathlist:
if filename.endswith('iso'):
host['os_version'] = iso_path + filename
if host['os_version'] == iso_path:
print("do not have os iso file in /var/lib/daisy/kolla/.")
client.hosts.update(host['id'], **host)
print("update role...")
add_host_role(cluster_id, host['id'], host_name, host['name'], vip)
def add_host_role(cluster_id, host_id, host_exp_name, host_real_name, vip):
role_meta = {'filters': {'cluster_id': cluster_id}}
role_list_generator = client.roles.list(**role_meta)
role_list = [role for role in role_list_generator]
lb_role_id = [role.id for role in role_list if
role.name == "CONTROLLER_LB"][0]
computer_role_id = [role.id for role in role_list if
role.name == "COMPUTER"][0]
if host_exp_name in ['all_in_one']:
role_lb_update_meta = {'nodes': [host_id],
'cluster_id': cluster_id, 'vip': vip}
client.roles.update(lb_role_id, **role_lb_update_meta)
role_computer_update_meta = {'nodes': [host_id],
'cluster_id': cluster_id}
client.roles.update(computer_role_id, **role_computer_update_meta)
if host_exp_name in ['controller01', 'controller02', 'controller03']:
role_lb_update_meta = {'nodes': [host_id],
'cluster_id': cluster_id, 'vip': vip}
client.roles.update(lb_role_id, **role_lb_update_meta)
if host_exp_name in ['computer01', 'computer02']:
role_computer_update_meta = {'nodes': [host_id],
'cluster_id': cluster_id}
client.roles.update(computer_role_id, **role_computer_update_meta)
if __name__ == "__main__":
prepare_install()