зеркало из https://github.com/Azure/Avere.git
Added support for passing params file; removed temp dir actions
This commit is contained in:
Omar Zevallos
Родитель
faa69674b3
Коммит
9fef4f0f14
|
|
@ -12,22 +12,45 @@ from string import ascii_lowercase, digits
|
|||
|
||||
# GLOBAL VARIABLES ############################################################
|
||||
|
||||
ARGS = None
|
||||
# AZ_PARAMS = {}
|
||||
AZ_PARAMS = {}
|
||||
DEFAULT_LOCATION = 'eastus2'
|
||||
|
||||
RANDOM_ID = 'av' + ''.join(random.choice(ascii_lowercase + digits) for _ in range(6))
|
||||
TMP_DIR = '/tmp/tmp.' + RANDOM_ID
|
||||
RG_NAME = 'aapipe-' + RANDOM_ID + '-rg'
|
||||
RG_CREATED = False
|
||||
|
||||
RESOURCE_GROUP_CREATED = False
|
||||
SCRIPT_ARGS = None
|
||||
SECRETS_FILE = None
|
||||
TEMPLATE_URL = 'https://raw.githubusercontent.com/Azure/Avere/master/src/vfxt/azuredeploy-auto.json'
|
||||
TEMPLATE_LOCAL = TMP_DIR + '/' + os.path.basename(TEMPLATE_URL)
|
||||
PARAMS_FILE = TMP_DIR + '/' + RANDOM_ID + '.params.json'
|
||||
TEMPLATE_LOCAL_FILE = os.path.basename(TEMPLATE_URL)
|
||||
|
||||
# FUNCTIONS ###################################################################
|
||||
|
||||
def create_params_json():
|
||||
def load_params():
|
||||
global AZ_PARAMS
|
||||
if SCRIPT_ARGS.param_file: # Open user-specified params file.
|
||||
with open(SCRIPT_ARGS.param_file) as config_file:
|
||||
AZ_PARAMS = json.load(config_file)
|
||||
else: # Generate and store params.
|
||||
random_id = 'av' + \
|
||||
''.join(random.choice(ascii_lowercase + digits) for _ in range(6))
|
||||
rg_name = 'aapipe-' + random_id + '-rg'
|
||||
AZ_PARAMS = {
|
||||
'resource-group': rg_name,
|
||||
'parameters' : {
|
||||
'virtualNetworkResourceGroup': rg_name,
|
||||
'virtualNetworkName': random_id + '-vnet',
|
||||
'virtualNetworkSubnetName': random_id + '-subnet',
|
||||
'avereBackedStorageAccountName': random_id + 'sa',
|
||||
'controllerName': random_id + '-con',
|
||||
'controllerAuthenticationType': 'password'
|
||||
}
|
||||
}
|
||||
with open(AZ_PARAMS['resource-group'] + '.params.json', 'w') as outfile:
|
||||
json.dump(AZ_PARAMS, outfile)
|
||||
print('AZ_PARAMS: {}'.format(AZ_PARAMS))
|
||||
|
||||
# command-line > params file > default
|
||||
if not SCRIPT_ARGS.location:
|
||||
SCRIPT_ARGS.location = AZ_PARAMS.pop('location', DEFAULT_LOCATION)
|
||||
|
||||
def load_secrets():
|
||||
exp_envars = [
|
||||
'controllerPassword',
|
||||
'adminPassword',
|
||||
|
|
@ -43,67 +66,58 @@ def create_params_json():
|
|||
raise Exception('The following envars must be defined: ' +
|
||||
', '.join(exp_envars))
|
||||
|
||||
with open(PARAMS_FILE, 'w') as params:
|
||||
global SECRETS_FILE
|
||||
SECRETS_FILE = AZ_PARAMS['resource-group'] + '.secrets.json'
|
||||
with open(SECRETS_FILE, 'w') as params:
|
||||
json.dump(data, params)
|
||||
|
||||
def download_template():
|
||||
print('> Downloading template: ' + TEMPLATE_URL)
|
||||
urlretrieve(TEMPLATE_URL, filename=TEMPLATE_LOCAL)
|
||||
urlretrieve(TEMPLATE_URL, filename=TEMPLATE_LOCAL_FILE)
|
||||
|
||||
def create_resource_group():
|
||||
global RG_CREATED
|
||||
print('> Creating resource group: ' + RG_NAME)
|
||||
global RESOURCE_GROUP_CREATED
|
||||
print('> Creating resource group: ' + AZ_PARAMS['resource-group'])
|
||||
_run_az_cmd('az group create --name {0} --location {1} {2}'.format(
|
||||
RG_NAME, ARGS.location, '--debug' if ARGS.az_debug else ''))
|
||||
RG_CREATED = True
|
||||
AZ_PARAMS['resource-group'], SCRIPT_ARGS.location,
|
||||
'--debug' if SCRIPT_ARGS.az_debug else ''))
|
||||
RESOURCE_GROUP_CREATED = True
|
||||
|
||||
def deploy_template():
|
||||
print('> Deploying template')
|
||||
cmd = """az group deployment create {0}
|
||||
--resource-group {1}
|
||||
--template-file {2}
|
||||
--parameters @{3}
|
||||
--parameters
|
||||
virtualNetworkResourceGroup={1}
|
||||
virtualNetworkName={4}-vnet
|
||||
virtualNetworkSubnetName={4}-subnet
|
||||
avereBackedStorageAccountName={4}sa
|
||||
controllerName={4}-con
|
||||
controllerAuthenticationType=password
|
||||
""".format('--debug' if ARGS.az_debug else '', RG_NAME, TEMPLATE_LOCAL,
|
||||
PARAMS_FILE, RANDOM_ID)
|
||||
--parameters @{3}""".format('--debug' if SCRIPT_ARGS.az_debug else '',
|
||||
AZ_PARAMS['resource-group'], TEMPLATE_LOCAL_FILE, SECRETS_FILE)
|
||||
|
||||
if AZ_PARAMS['parameters']: # There are more parameters.
|
||||
cmd += "\n\t--parameters"
|
||||
for key, val in AZ_PARAMS['parameters'].items():
|
||||
cmd += "\n\t\t{0}={1}".format(key, val)
|
||||
|
||||
_run_az_cmd(cmd)
|
||||
|
||||
def cleanup(starting_dir):
|
||||
os.chdir(starting_dir)
|
||||
if ARGS.skip_cleanup:
|
||||
def cleanup():
|
||||
os.remove(SECRETS_FILE) # Always remove this file.
|
||||
if SCRIPT_ARGS.skip_cleanup:
|
||||
print('> Skipping clean up')
|
||||
return
|
||||
print('> Cleaning up')
|
||||
if RG_CREATED:
|
||||
print('> Deleting resource group: ' + RG_NAME)
|
||||
_run_az_cmd('az group delete --yes --name {0} {1}'.format(
|
||||
RG_NAME, '--debug' if ARGS.az_debug else ''))
|
||||
|
||||
_debug('Removing temp directory: ' + TMP_DIR)
|
||||
shutil.rmtree(TMP_DIR)
|
||||
print('> Cleaning up')
|
||||
os.remove(TEMPLATE_LOCAL_FILE)
|
||||
if RESOURCE_GROUP_CREATED:
|
||||
print('> Deleting resource group: ' + AZ_PARAMS['resource-group'])
|
||||
_run_az_cmd('az group delete --yes --name {0} {1}'.format(
|
||||
AZ_PARAMS['resource-group'], '--debug' if SCRIPT_ARGS.az_debug else ''))
|
||||
|
||||
# HELPER FUNCTIONS ############################################################
|
||||
|
||||
# def _load_or_generate_params():
|
||||
# global AZ_PARAMS
|
||||
# if ARGS.config:
|
||||
# pass
|
||||
|
||||
# if ARGS.location:
|
||||
# AZ_PARAMS['location'] = ARGS.location
|
||||
|
||||
def _run_az_cmd(_cmd):
|
||||
cmd = _cmd.strip()
|
||||
_debug('az command: "' + cmd + '"')
|
||||
|
||||
if ARGS.print_az_cmds:
|
||||
if SCRIPT_ARGS.print_az_cmds:
|
||||
print('az command: "' + cmd + '"')
|
||||
else:
|
||||
cmd = cmd.split()
|
||||
|
|
@ -111,22 +125,14 @@ def _run_az_cmd(_cmd):
|
|||
subprocess.check_call(cmd, stderr=subprocess.STDOUT)
|
||||
|
||||
def _debug(s):
|
||||
if ARGS.debug:
|
||||
if SCRIPT_ARGS.debug:
|
||||
print('[DEBUG]: {}'.format(s))
|
||||
|
||||
# MAIN ########################################################################
|
||||
|
||||
def main():
|
||||
global ARGS
|
||||
if not ARGS.location:
|
||||
ARGS.location = DEFAULT_LOCATION
|
||||
starting_dir = os.getcwd()
|
||||
os.mkdir(TMP_DIR)
|
||||
os.chdir(TMP_DIR)
|
||||
_debug('Starting directory: ' + starting_dir)
|
||||
_debug('Temp directory created, now CWD: ' + TMP_DIR)
|
||||
|
||||
create_params_json()
|
||||
load_params()
|
||||
load_secrets()
|
||||
retcode = 0 # PASS
|
||||
try:
|
||||
download_template()
|
||||
|
|
@ -139,16 +145,17 @@ def main():
|
|||
retcode = 1 # FAIL
|
||||
raise
|
||||
finally:
|
||||
cleanup(starting_dir)
|
||||
cleanup()
|
||||
|
||||
print('> TEST COMPLETE. Resource Group: {} (region: {})'.format(RG_NAME, ARGS.location))
|
||||
print('> TEST COMPLETE. Resource Group: {} (region: {})'.format(
|
||||
AZ_PARAMS['resource-group'], SCRIPT_ARGS.location))
|
||||
print('> RESULT: ' + ('FAIL' if retcode else 'PASS'))
|
||||
sys.exit(retcode)
|
||||
|
||||
if __name__ == '__main__':
|
||||
arg_parser = argparse.ArgumentParser(description='Test Avere vFXT Azure template deployment.')
|
||||
# arg_parser.add_argument('-c', '--config', default=None,
|
||||
# help='Full path to JSON config file. Default: None (generate new config)')
|
||||
arg_parser.add_argument('-pf', '--param_file', default=None,
|
||||
help='Full path to JSON params file. Default: None (generate new params)')
|
||||
arg_parser.add_argument('-l', '--location', default=None,
|
||||
help='Azure location (region short name) to use for deployment. ' +
|
||||
'Default: ' + DEFAULT_LOCATION)
|
||||
|
|
@ -160,6 +167,6 @@ if __name__ == '__main__':
|
|||
help='Turn on "az" command debugging.')
|
||||
arg_parser.add_argument('-d', '--debug', action='store_true',
|
||||
help='Turn on script debugging.')
|
||||
ARGS = arg_parser.parse_args()
|
||||
SCRIPT_ARGS = arg_parser.parse_args()
|
||||
|
||||
main()
|
||||
Загрузка…
Ссылка в новой задаче