156 lines
4.2 KiB
Python
156 lines
4.2 KiB
Python
#
|
|
# Copyright (c) 2015-2024 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
import re
|
|
import requests
|
|
import signal
|
|
import time
|
|
|
|
from software_client.common import base
|
|
from software_client.common import utils
|
|
from software_client import constants
|
|
|
|
|
|
class Deploy(base.Resource):
|
|
def __repr__(self):
|
|
return "<address pool %s>" % self._info
|
|
|
|
|
|
class DeployManager(base.Manager):
|
|
resource_class = Deploy
|
|
|
|
def precheck(self, args):
|
|
# args.deployment is a string
|
|
deployment = args.deployment
|
|
|
|
path = "/v1/deploy/%s/precheck" % (deployment)
|
|
body = {}
|
|
if args.force:
|
|
body["force"] = "true"
|
|
|
|
if args.region_name:
|
|
body["region_name"] = args.region_name
|
|
|
|
res = self._post(path, body=body)
|
|
return res
|
|
|
|
def start(self, args):
|
|
# args.deployment is a string
|
|
deployment = args.deployment
|
|
|
|
# Ignore interrupts during this function
|
|
signal.signal(signal.SIGINT, signal.SIG_IGN)
|
|
|
|
# Issue deploy_start request
|
|
path = "/v1/deploy/%s/start" % (deployment)
|
|
body = {}
|
|
if args.force:
|
|
body["force"] = "true"
|
|
|
|
return self._post(path, body=body)
|
|
|
|
def host(self, args):
|
|
# args.deployment is a string
|
|
hostname = args.host
|
|
|
|
# Issue deploy_host request and poll for results
|
|
path = "/v1/deploy_host/%s" % (hostname)
|
|
|
|
if args.force:
|
|
path += "/force"
|
|
|
|
return self._create(path)
|
|
|
|
def activate(self, args):
|
|
# Ignore interrupts during this function
|
|
signal.signal(signal.SIGINT, signal.SIG_IGN)
|
|
|
|
# Issue deploy_start request
|
|
path = "/v1/deploy/activate"
|
|
|
|
return self._create(path, body={})
|
|
|
|
def complete(self, args):
|
|
# Ignore interrupts during this function
|
|
signal.signal(signal.SIGINT, signal.SIG_IGN)
|
|
|
|
# Issue deploy_start request
|
|
path = "/v1/deploy/complete"
|
|
|
|
return self._create(path, body={})
|
|
|
|
def host_list(self):
|
|
path = '/v1/deploy_host'
|
|
return self._list(path, "")
|
|
|
|
def show(self):
|
|
path = '/v1/deploy'
|
|
return self._list(path)
|
|
|
|
def wait_for_install_complete(self, hostname):
|
|
url = "/v1/deploy_host"
|
|
rc = 0
|
|
|
|
max_retries = 4
|
|
retriable_count = 0
|
|
|
|
while True:
|
|
# Sleep on the first pass as well, to allow time for the
|
|
# agent to respond
|
|
time.sleep(5)
|
|
|
|
try:
|
|
req, data = self._list(url)
|
|
except requests.exceptions.ConnectionError:
|
|
# The local software-controller may have restarted.
|
|
retriable_count += 1
|
|
if retriable_count > max_retries:
|
|
print("Lost communications with the software controller")
|
|
rc = 1
|
|
return rc
|
|
else:
|
|
break
|
|
|
|
if req.status_code == 200:
|
|
if not data:
|
|
print("Invalid host-list data returned:")
|
|
utils.print_result_debug(req, data)
|
|
rc = 1
|
|
|
|
host_state = None
|
|
|
|
for d in data:
|
|
if d['hostname'] == hostname:
|
|
host_state = d.get('host_state')
|
|
|
|
if host_state == constants.DEPLOYING:
|
|
print("\nDeployment started.")
|
|
rc = 0
|
|
elif host_state == constants.FAILED:
|
|
print("\nDeployment failed. Please check logs for details.")
|
|
rc = 1
|
|
elif host_state == constants.DEPLOYED:
|
|
print("\nDeployment was successful.")
|
|
rc = 0
|
|
elif host_state == constants.PENDING:
|
|
print("\nDeployment pending.")
|
|
else:
|
|
print("\nReported unknown state: %s" % host_state)
|
|
rc = 1
|
|
|
|
elif req.status_code == 500:
|
|
print("An internal error has occurred. Please check /var/log/software.log for details")
|
|
rc = 1
|
|
else:
|
|
m = re.search("(Error message:.*)", req.text, re.MULTILINE)
|
|
if m:
|
|
print(m.group(0))
|
|
else:
|
|
print(vars(req))
|
|
rc = 1
|
|
|
|
return rc
|