def remove(silent=None):
"""
Interactive removal part for the ASD manager
:param silent: If silent == '--force-yes' no question will be asked to confirm the removal
:type silent: str
:return: None
"""
os.environ['OVS_LOGTYPE_OVERRIDE'] = 'file'
print '\n' + Interactive.boxed_message(['ASD Manager removal'])
##############
# VALIDATION #
##############
local_client = LocalClient()
if not local_client.file_exists(filename=Toolbox.BOOTSTRAP_FILE):
print '\n' + Interactive.boxed_message(['The ASD Manager has already been removed'])
sys.exit(1)
print ' - Validating configuration file'
config = _validate_and_retrieve_pre_config()
if config is None or 'store' not in config:
print '\n' + Interactive.boxed_message(['Cannot remove the ASD manager because not all information could be retrieved from the pre-configuration file'])
sys.exit(1)
print ' - Validating node ID'
with open(Toolbox.BOOTSTRAP_FILE, 'r') as bs_file:
try:
alba_node_id = json.loads(bs_file.read())['node_id']
except:
print '\n' + Interactive.boxed_message(['JSON contents could not be retrieved from file {0}'.format(Toolbox.BOOTSTRAP_FILE)])
sys.exit(1)
print ' - Validating configuration management'
store = config['store']
try:
Configuration.list(key='ovs')
except:
if store == 'arakoon':
print '\n' + Interactive.boxed_message(['Could not connect to Arakoon'])
else:
print '\n' + Interactive.boxed_message(['Could not connect to Etcd.',
'Please make sure an Etcd proxy is available, pointing towards an OpenvStorage cluster.'])
sys.exit(1)
################
# CONFIRMATION #
################
os.environ['ASD_NODE_ID'] = alba_node_id
from source.app.api import API
print ' - Retrieving ASD information'
all_asds = {}
try:
all_asds = API.list_asds.original()
except:
print ' - ERROR: Failed to retrieve the ASD information'
interactive = silent != '--force-yes'
if interactive is True:
message = 'Are you sure you want to continue?'
if len(all_asds) > 0:
print '\n\n+++ ALERT +++\n'
message = 'DATA LOSS possible if proceeding! Continue?'
proceed = Interactive.ask_yesno(message=message, default_value=False)
if proceed is False:
print '\n' + Interactive.boxed_message(['Abort removal'])
sys.exit(1)
###########
# REMOVAL #
###########
print ' - Removing from configuration management'
Configuration.uninitialize(node_id=alba_node_id)
if len(all_asds) > 0:
print ' - Removing disks'
for device_id, disk_info in API.list_disks.original().iteritems():
if disk_info['available'] is True:
continue
try:
print ' - Retrieving ASD information for disk {0}'.format(disk_info['device'])
for asd_id, asd_info in API.list_asds_disk.original(disk_id=device_id).iteritems():
print ' - Removing ASD {0}'.format(asd_id)
API.asd_delete.original(disk_id=device_id, asd_id=asd_id)
API.delete_disk.original(disk_id=device_id)
except Exception as ex:
print ' - Deleting ASDs failed: {0}'.format(ex)
print ' - Removing services'
for service_name in API.list_maintenance_services.original()['services']:
print ' - Removing service {0}'.format(service_name)
API.remove_maintenance_service.original(name=service_name)
for service_name in [WATCHER_SERVICE, MANAGER_SERVICE]:
if ServiceManager.has_service(name=service_name, client=local_client):
print ' - Removing service {0}'.format(service_name)
ServiceManager.stop_service(name=service_name, client=local_client)
ServiceManager.remove_service(name=service_name, client=local_client)
if store == 'arakoon':
from source.tools.configuration.arakoon_config import ArakoonConfiguration
local_client.file_delete(filenames=ArakoonConfiguration.CACC_LOCATION)
local_client.file_delete(filenames=Toolbox.BOOTSTRAP_FILE)
print '\n' + Interactive.boxed_message(['ASD Manager removal completed'])
def setup():
"""
Interactive setup part for initial asd manager configuration
"""
print Interactive.boxed_message(['ASD Manager setup'])
print '- Verifying distribution'
local_client = LocalClient()
if ServiceManager.has_service(MANAGER_SERVICE, local_client):
print '' # Spacing
print Interactive.boxed_message(['The ASD Manager is already installed.'])
sys.exit(1)
ipaddresses = check_output("ip a | grep 'inet ' | sed 's/\s\s*/ /g' | cut -d ' ' -f 3 | cut -d '/' -f 1", shell=True).strip().splitlines()
ipaddresses = [found_ip.strip() for found_ip in ipaddresses if found_ip.strip() != '127.0.0.1']
if not ipaddresses:
print Interactive.boxed_message(['Could not retrieve IP information on current node'])
sys.exit(1)
config = _validate_and_retrieve_pre_config()
if config is None:
api_ip = Interactive.ask_choice(ipaddresses, 'Select the public IP address to be used for the API')
asd_ips = []
add_ips = True
api_port = Interactive.ask_integer("Select the port to be used for the API", 1025, 65535, 8500)
ipaddresses.append('All')
while add_ips:
current_ips = ' - Current selected IPs: {0}'.format(asd_ips)
new_asd_ip = Interactive.ask_choice(ipaddresses,
'Select an IP address or all IP addresses to be used for the ASDs{0}'.format(current_ips if len(asd_ips) > 0 else ''),
default_value='All')
if new_asd_ip == 'All':
ipaddresses.remove('All')
asd_ips = []
add_ips = False
else:
asd_ips.append(new_asd_ip)
ipaddresses.remove(new_asd_ip)
add_ips = Interactive.ask_yesno("Do you want to add another IP?")
asd_start_port = Interactive.ask_integer("Select the port to be used for the ASDs", 1025, 65435, 8600)
store = Interactive.ask_choice(['Arakoon', 'Etcd'],
question='Select the configuration management system',
default_value='Arakoon').lower()
else:
store = config.get('store', 'arakoon')
api_ip = config['api_ip']
api_port = config.get('api_port', 8500)
asd_ips = config.get('asd_ips', [])
asd_start_port = config.get('asd_start_port', 8600)
if api_ip not in ipaddresses:
print Interactive.boxed_message(['Unknown API IP provided, please choose from: {0}'.format(', '.join(ipaddresses))])
sys.exit(1)
if set(asd_ips).difference(set(ipaddresses)):
print Interactive.boxed_message(['Unknown ASD IP provided, please choose from: {0}'.format(', '.join(ipaddresses))])
sys.exit(1)
if api_port in range(asd_start_port, asd_start_port + 100):
print Interactive.boxed_message(['API port cannot be in the range of the ASD port + 100'])
sys.exit(1)
# Make sure to always have the information stored
with open(PRECONFIG_FILE, 'w') as preconfig:
preconfig.write(json.dumps({'asdmanager': {'store': store,
'api_ip': api_ip,
'asd_ips': asd_ips,
'api_port': api_port,
'asd_start_port': asd_start_port}}, indent=4))
if store == 'arakoon':
from source.tools.configuration.arakoon_config import ArakoonConfiguration
file_location = ArakoonConfiguration.CACC_LOCATION
source_location = ArakoonConfiguration.CACC_SOURCE
if not local_client.file_exists(file_location) and local_client.file_exists(source_location):
# Try to copy automatically
try:
local_client.file_upload(file_location, source_location)
except Exception:
pass
while not local_client.file_exists(file_location):
print 'Please place a copy of the Arakoon\'s client configuration file at: {0}'.format(file_location)
Interactive.ask_continue()
bootstrap_location = Configuration.BOOTSTRAP_CONFIG_LOCATION
if not local_client.file_exists(bootstrap_location):
local_client.file_create(bootstrap_location)
local_client.file_write(bootstrap_location, json.dumps({'configuration_store': store}, indent=4))
try:
alba_node_id = Configuration.initialize(api_ip, api_port, asd_ips, asd_start_port)
except:
print ''
if store == 'arakoon':
print Interactive.boxed_message(['Could not connect to Arakoon'])
else:
print Interactive.boxed_message(['Could not connect to Etcd.',
'Please make sure an Etcd proxy is available, pointing towards an OpenvStorage cluster.'])
sys.exit(1)
with open(Toolbox.BOOTSTRAP_FILE, 'w') as bs_file:
json.dump({'node_id': alba_node_id}, bs_file)
ServiceManager.add_service(MANAGER_SERVICE, local_client)
ServiceManager.add_service(WATCHER_SERVICE, local_client)
print '- Starting watcher service'
try:
ServiceManager.start_service(WATCHER_SERVICE, local_client)
except Exception as ex:
Configuration.uninitialize(alba_node_id)
print Interactive.boxed_message(['Starting watcher failed with error:', str(ex)])
sys.exit(1)
print Interactive.boxed_message(['ASD Manager setup completed'])
def _log(message):
print '{0} - {1}'.format(str(datetime.now()), message)
if __name__ == '__main__':
import os
import json
import glob
from source.tools.filemutex import file_mutex
from source.tools.localclient import LocalClient
from source.tools.services.service import ServiceManager
from source.tools.configuration import EtcdConfiguration
_log('Executing post-upgrade logic of package openvstorage-sdm')
with file_mutex('package_update'):
client = LocalClient('127.0.0.1', username='******')
migrate = False
service_name = 'alba-asdmanager'
if ServiceManager.has_service(service_name, client):
_log('Removing old alba-asdmanager service')
ServiceManager.stop_service(service_name, client)
ServiceManager.remove_service(service_name, client)
service_name = 'asd-manager'
if ServiceManager.has_service(service_name, client) and ServiceManager.get_service_status(service_name, client) is True:
_log('Stopping asd-manager service')
ServiceManager.stop_service(service_name, client)
# Migrate main configuration file
path = '/opt/alba-asdmanager/config/config.json'
