summaryrefslogtreecommitdiffstats
path: root/fuel/deploy/install_fuel_master.py
blob: bb8e7e175c7dd620976ba6186e6356fb88352873 (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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
import common
import time
import os
from ssh_client import SSHClient
from dha_adapters.libvirt_adapter import LibvirtAdapter

log = common.log
err = common.err
clean = common.clean

TRANSPLANT_FUEL_SETTINGS = 'transplant_fuel_settings.py'
BOOTSTRAP_ADMIN = '/usr/local/sbin/bootstrap_admin_node'

class InstallFuelMaster(object):

    def __init__(self, dea_file, dha_file, fuel_ip, fuel_username, fuel_password,
                 fuel_node_id, iso_file, work_dir):
        self.dea_file = dea_file
        self.dha = LibvirtAdapter(dha_file)
        self.fuel_ip = fuel_ip
        self.fuel_username = fuel_username
        self.fuel_password = fuel_password
        self.fuel_node_id = fuel_node_id
        self.iso_file = iso_file
        self.work_dir = work_dir
        self.file_dir = os.path.dirname(os.path.realpath(__file__))
        self.ssh = SSHClient(self.fuel_ip, self.fuel_username,
                             self.fuel_password)

    def install(self):
        log('Start Fuel Installation')

        self.dha.node_power_off(self.fuel_node_id)

        self.zero_mbr_set_boot_order()

        self.proceed_with_installation()

    def custom_install(self):
        log('Start Custom Fuel Installation')

        self.dha.node_power_off(self.fuel_node_id)

        log('Zero the MBR')
        self.dha.node_zero_mbr(self.fuel_node_id)

        self.dha.node_set_boot_order(self.fuel_node_id, ['disk', 'iso'])

        self.proceed_with_installation()

    def proceed_with_installation(self):
        log('Eject ISO')
        self.dha.node_eject_iso(self.fuel_node_id)

        log('Insert ISO %s' % self.iso_file)
        self.dha.node_insert_iso(self.fuel_node_id, self.iso_file)

        self.dha.node_power_on(self.fuel_node_id)

        log('Waiting for Fuel master to accept SSH')
        self.wait_for_node_up()

        log('Wait until Fuel menu is up')
        fuel_menu_pid = self.wait_until_fuel_menu_up()

        log('Inject our own astute.yaml settings')
        self.inject_own_astute_yaml()

        log('Let the Fuel deployment continue')
        log('Found FUEL menu as PID %s, now killing it' % fuel_menu_pid)
        self.ssh_exec_cmd('kill %s' % fuel_menu_pid)

        log('Wait until installation complete')
        self.wait_until_installation_completed()

        log('Waiting for one minute for Fuel to stabilize')
        time.sleep(60)

        log('Eject ISO')
        self.dha.node_eject_iso(self.fuel_node_id)

        log('Fuel Master installed successfully !')

    def zero_mbr_set_boot_order(self):
        if self.dha.node_can_zero_mbr(self.fuel_node_id):
            log('Fuel Node %s capable of zeroing MBR so doing that...'
                % self.fuel_node_id)
            self.dha.node_zero_mbr(self.fuel_node_id)
            self.dha.node_set_boot_order(self.fuel_node_id, ['disk', 'iso'])
        elif self.dha.node_can_set_boot_order_live(self.fuel_node_id):
            log('Node %s can change ISO boot order live' % self.fuel_node_id)
            self.dha.node_set_boot_order(self.fuel_node_id, ['iso', 'disk'])
        else:
            err('No way to install Fuel node')

    def wait_for_node_up(self):
        WAIT_LOOP = 60
        SLEEP_TIME = 10
        success = False
        for i in range(WAIT_LOOP):
            try:
                self.ssh.open()
                success = True
                break
            except Exception as e:
                log('EXCEPTION [%s] received when SSH-ing into Fuel VM %s ... '
                    'sleeping %s seconds' % (e, self.fuel_ip, SLEEP_TIME))
                time.sleep(SLEEP_TIME)
            finally:
                self.ssh.close()

        if not success:
           err('Could not SSH into Fuel VM %s' % self.fuel_ip)

    def wait_until_fuel_menu_up(self):
        WAIT_LOOP = 60
        SLEEP_TIME = 10
        CMD = 'ps -ef'
        SEARCH = 'fuelmenu'
        fuel_menu_pid = None
        with self.ssh:
            for i in range(WAIT_LOOP):
                ret = self.ssh.exec_cmd(CMD)
                fuel_menu_pid = self.get_fuel_menu_pid(ret, SEARCH)
                if not fuel_menu_pid:
                    time.sleep(SLEEP_TIME)
                else:
                    break
        if not fuel_menu_pid:
            err('Could not find the Fuel Menu Process ID')
        return fuel_menu_pid

    def get_fuel_menu_pid(self, printout, search):
        fuel_menu_pid = None
        for line in printout.splitlines():
            if search in line:
                fuel_menu_pid = clean(line)[1]
                break
        return fuel_menu_pid

    def ssh_exec_cmd(self, cmd):
        with self.ssh:
            ret = self.ssh.exec_cmd(cmd)
        return ret

    def inject_own_astute_yaml(self):
        dest ='~/%s/' % self.work_dir

        with self.ssh as s:
            s.exec_cmd('rm -rf %s' % self.work_dir, check=False)
            s.exec_cmd('mkdir ~/%s' % self.work_dir)
            s.scp_put(self.dea_file, dest)
            s.scp_put('%s/common.py' % self.file_dir, dest)
            s.scp_put('%s/dea.py' % self.file_dir, dest)
            s.scp_put('%s/transplant_fuel_settings.py' % self.file_dir, dest)
            log('Modifying Fuel astute')
            s.run('python ~/%s/%s ~/%s/%s'
                  % (self.work_dir, TRANSPLANT_FUEL_SETTINGS,
                     self.work_dir, os.path.basename(self.dea_file)))

    def wait_until_installation_completed(self):
        WAIT_LOOP = 180
        SLEEP_TIME = 10
        CMD = 'ps -ef | grep %s | grep -v grep' % BOOTSTRAP_ADMIN

        install_completed = False
        with self.ssh:
            for i in range(WAIT_LOOP):
                ret = self.ssh.exec_cmd(CMD)
                if not ret:
                    install_completed = True
                    break
                else:
                    time.sleep(SLEEP_TIME)

        if not install_completed:
            err('Fuel installation did not complete')