| #!/usr/bin/python |
| |
| # ***************************************************************************** |
| # |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| # |
| # ****************************************************************************** |
| |
| import sys |
| import os |
| import json |
| from fabric.api import * |
| from Crypto.PublicKey import RSA |
| import dlab.ssn_lib |
| import dlab.fab |
| import dlab.actions_lib |
| import dlab.meta_lib |
| import logging |
| import traceback |
| |
| |
| if __name__ == "__main__": |
| local_log_filename = "{}_{}.log".format(os.environ['conf_resource'], os.environ['request_id']) |
| local_log_filepath = "/logs/" + os.environ['conf_resource'] + "/" + local_log_filename |
| logging.basicConfig(format='%(levelname)-8s [%(asctime)s] %(message)s', |
| level=logging.DEBUG, |
| filename=local_log_filepath) |
| try: |
| AzureMeta = dlab.meta_lib.AzureMeta() |
| AzureActions = dlab.actions_lib.AzureActions() |
| ssn_conf = dict() |
| ssn_conf['instance'] = 'ssn' |
| |
| logging.info('[DERIVING NAMES]') |
| print('[DERIVING NAMES]') |
| # Verify vpc deployment |
| if os.environ['conf_network_type'] == 'private' and not os.environ.get('azure_vpc_name') \ |
| and not os.environ.get('azure_source_vpc_name'): |
| raise Exception('Not possible to deploy private environment without predefined vpc or without source vpc') |
| if os.environ['conf_network_type'] == 'private' and not os.environ.get('azure_resource_group_name') \ |
| and not os.environ.get('azure_source_resource_group_name'): |
| raise Exception('Not possible to deploy private environment without predefined resource_group_name ' |
| 'or source_group_name') |
| # We need to cut service_base_name to 20 symbols do to the Azure Name length limitation |
| ssn_conf['service_base_name'] = os.environ['conf_service_base_name'] = dlab.fab.replace_multi_symbols( |
| os.environ['conf_service_base_name'][:20], '-', True) |
| # Check azure predefined resources |
| ssn_conf['resource_group_name'] = os.environ.get('azure_resource_group_name', |
| '{}-resource-group'.format(ssn_conf['service_base_name'])) |
| ssn_conf['source_resource_group_name'] = os.environ.get( |
| 'azure_source_resource_group_name', '{}-resource-group'.format(ssn_conf['service_base_name'])) |
| ssn_conf['vpc_name'] = os.environ.get('azure_vpc_name', '{}-vpc'.format(ssn_conf['service_base_name'])) |
| ssn_conf['subnet_name'] = os.environ.get('azure_subnet_name', '{}-subnet'.format(ssn_conf['service_base_name'])) |
| ssn_conf['security_group_name'] = os.environ.get('azure_security_group_name', |
| '{}-sg'.format(ssn_conf['service_base_name'])) |
| # Default variables |
| ssn_conf['region'] = os.environ['azure_region'] |
| ssn_conf['vpc_cidr'] = os.environ['conf_vpc_cidr'] |
| ssn_conf['subnet_prefix'] = '20' |
| ssn_conf['ssn_image_name'] = os.environ['azure_{}_image_name'.format(os.environ['conf_os_family'])] |
| ssn_conf['datalake_store_name'] = '{}-ssn-datalake'.format(ssn_conf['service_base_name']) |
| ssn_conf['datalake_shared_directory_name'] = '{}-shared-folder'.format(ssn_conf['service_base_name']) |
| ssn_conf['instance_name'] = '{}-ssn'.format(ssn_conf['service_base_name']) |
| ssn_conf['network_interface_name'] = '{}-ssn-nif'.format(ssn_conf['service_base_name']) |
| if os.environ['conf_network_type'] == 'private': |
| ssn_conf['static_public_ip_name'] = 'None' |
| else: |
| ssn_conf['static_public_ip_name'] = '{}-ssn-static-ip'.format(ssn_conf['service_base_name']) |
| ssn_conf['key'] = RSA.importKey(open('{}{}.pem'.format(os.environ['conf_key_dir'], |
| os.environ['conf_key_name']), 'rb').read()) |
| ssn_conf['instance_storage_account_type'] = 'Premium_LRS' |
| ssn_conf['public_ssh_key'] = ssn_conf['key'].publickey().exportKey("OpenSSH") |
| ssn_conf['instance_tags'] = {"Name": ssn_conf['instance_name'], |
| "SBN": ssn_conf['service_base_name'], |
| os.environ['conf_billing_tag_key']: os.environ['conf_billing_tag_value']} |
| |
| ssn_conf['datalake_store_tags'] = {"Name": ssn_conf['datalake_store_name'], |
| "SBN": ssn_conf['service_base_name'], |
| os.environ['conf_billing_tag_key']: os.environ['conf_billing_tag_value']} |
| ssn_conf['primary_disk_size'] = '32' |
| except Exception as err: |
| dlab.fab.append_result("Failed to generate variables dictionary.", str(err)) |
| sys.exit(1) |
| |
| if AzureMeta.get_instance(ssn_conf['resource_group_name'], ssn_conf['instance_name']): |
| dlab.fab.append_result("Service base name should be unique and less or equal 20 symbols. Please try again.") |
| sys.exit(1) |
| |
| try: |
| if 'azure_resource_group_name' in os.environ: |
| logging.info('Resource group predefined') |
| print('Resource group predefined') |
| else: |
| logging.info('[CREATING RESOURCE GROUP]') |
| print("[CREATING RESOURCE GROUP]") |
| params = "--resource_group_name {} --region {}".format(ssn_conf['resource_group_name'], ssn_conf['region']) |
| local("~/scripts/{}.py {}".format('ssn_create_resource_group', params)) |
| except Exception as err: |
| traceback.print_exc() |
| dlab.fab.append_result("Failed to create Resource Group.", str(err)) |
| sys.exit(1) |
| |
| try: |
| if 'azure_vpc_name' in os.environ: |
| logging.info('VPC predefined') |
| print('VPC predefined') |
| else: |
| logging.info('[CREATING VIRTUAL NETWORK]') |
| print("[CREATING VIRTUAL NETWORK]") |
| params = "--resource_group_name {} --vpc_name {} --region {} --vpc_cidr {}".format( |
| ssn_conf['resource_group_name'], ssn_conf['vpc_name'], ssn_conf['region'], ssn_conf['vpc_cidr']) |
| local("~/scripts/{}.py {}".format('ssn_create_vpc', params)) |
| except Exception as err: |
| traceback.print_exc() |
| dlab.fab.append_result("Failed to create VPC.", str(err)) |
| try: |
| if 'azure_resource_group_name' not in os.environ: |
| AzureActions.remove_resource_group(ssn_conf['resource_group_name'], ssn_conf['region']) |
| except Exception as err: |
| dlab.fab.append_result("Resources hasn't been removed.", str(err)) |
| sys.exit(1) |
| |
| try: |
| if 'azure_subnet_name' in os.environ: |
| logging.info('Subnet predefined') |
| print('Subnet predefined') |
| else: |
| logging.info('[CREATING SUBNET]') |
| print("[CREATING SUBNET]") |
| params = "--resource_group_name {} --vpc_name {} --region {} --vpc_cidr {} --subnet_name {} --prefix {}".\ |
| format(ssn_conf['resource_group_name'], ssn_conf['vpc_name'], ssn_conf['region'], |
| ssn_conf['vpc_cidr'], ssn_conf['subnet_name'], ssn_conf['subnet_prefix']) |
| local("~/scripts/{}.py {}".format('common_create_subnet', params)) |
| except Exception as err: |
| traceback.print_exc() |
| dlab.fab.append_result("Failed to create Subnet.", str(err)) |
| try: |
| if 'azure_vpc_name' not in os.environ: |
| AzureActions.remove_vpc(ssn_conf['resource_group_name'], ssn_conf['vpc_name']) |
| if 'azure_resource_group_name' not in os.environ: |
| AzureActions.remove_resource_group(ssn_conf['resource_group_name'], ssn_conf['region']) |
| except Exception as err: |
| print("Resources hasn't been removed: {}".format(str(err))) |
| dlab.fab.append_result("Resources hasn't been removed.", str(err)) |
| sys.exit(1) |
| |
| try: |
| if 'azure_vpc_name' not in os.environ and os.environ['conf_network_type'] == 'private': |
| logging.info('[CREATING VPC PEERING]') |
| print("[CREATING VPC PEERING]") |
| params = "--source_resource_group_name {} --destination_resource_group_name {} " \ |
| "--source_virtual_network_name {} --destination_virtual_network_name {}".format( |
| ssn_conf['source_resource_group_name'], ssn_conf['resource_group_name'], |
| os.environ['azure_source_vpc_name'], ssn_conf['vpc_name']) |
| local("~/scripts/{}.py {}".format('ssn_create_peering', params)) |
| except Exception as err: |
| traceback.print_exc() |
| try: |
| if 'azure_vpc_name' not in os.environ: |
| AzureActions.remove_vpc(ssn_conf['resource_group_name'], ssn_conf['vpc_name']) |
| if 'azure_resource_group_name' not in os.environ: |
| AzureActions.remove_resource_group(ssn_conf['resource_group_name'], ssn_conf['region']) |
| except Exception as err: |
| print("Resources hasn't been removed: " + str(err)) |
| dlab.fab.append_result("Resources hasn't been removed.", str(err)) |
| dlab.fab.append_result("Failed to create VPC peering.", str(err)) |
| sys.exit(1) |
| |
| try: |
| if 'azure_security_group_name' in os.environ: |
| logging.info('Security group predefined') |
| print('Security group predefined') |
| else: |
| logging.info('[CREATING SECURITY GROUP]') |
| print("[CREATING SECURITY GROUP]") |
| list_rules = [ |
| { |
| "name": "in-1", |
| "protocol": "Tcp", |
| "source_port_range": "*", |
| "destination_port_range": "80", |
| "source_address_prefix": "*", |
| "destination_address_prefix": "*", |
| "access": "Allow", |
| "priority": 100, |
| "direction": "Inbound" |
| }, |
| { |
| "name": "in-2", |
| "protocol": "Tcp", |
| "source_port_range": "*", |
| "destination_port_range": "443", |
| "source_address_prefix": "*", |
| "destination_address_prefix": "*", |
| "access": "Allow", |
| "priority": 110, |
| "direction": "Inbound" |
| }, |
| { |
| "name": "in-3", |
| "protocol": "Tcp", |
| "source_port_range": "*", |
| "destination_port_range": "22", |
| "source_address_prefix": "*", |
| "destination_address_prefix": "*", |
| "access": "Allow", |
| "priority": 120, |
| "direction": "Inbound" |
| }, |
| { |
| "name": "out-1", |
| "protocol": "*", |
| "source_port_range": "*", |
| "destination_port_range": "*", |
| "source_address_prefix": "*", |
| "destination_address_prefix": "*", |
| "access": "Allow", |
| "priority": 100, |
| "direction": "Outbound" |
| } |
| ] |
| params = "--resource_group_name {} --security_group_name {} --region {} --tags '{}' --list_rules '{}'".\ |
| format(ssn_conf['resource_group_name'], ssn_conf['security_group_name'], ssn_conf['region'], |
| json.dumps(ssn_conf['instance_tags']), json.dumps(list_rules)) |
| local("~/scripts/{}.py {}".format('common_create_security_group', params)) |
| except Exception as err: |
| traceback.print_exc() |
| dlab.fab.append_result("Error creating Security group", str(err)) |
| try: |
| if 'azure_subnet_name' not in os.environ: |
| AzureActions.remove_subnet(ssn_conf['resource_group_name'], ssn_conf['vpc_name'], |
| ssn_conf['subnet_name']) |
| if 'azure_vpc_name' not in os.environ: |
| AzureActions.remove_vpc(ssn_conf['resource_group_name'], ssn_conf['vpc_name']) |
| if 'azure_resource_group_name' not in os.environ: |
| AzureActions.remove_resource_group(ssn_conf['resource_group_name'], ssn_conf['region']) |
| except Exception as err: |
| print("Resources hasn't been removed: " + str(err)) |
| dlab.fab.append_result("Resources hasn't been removed.", str(err)) |
| sys.exit(1) |
| |
| if os.environ['azure_datalake_enable'] == 'true': |
| try: |
| logging.info('[CREATE DATA LAKE STORE]') |
| print('[CREATE DATA LAKE STORE]') |
| params = "--datalake_name {} --datalake_tags '{}' --resource_group_name {} --region {}". \ |
| format(ssn_conf['datalake_store_name'], json.dumps(ssn_conf['datalake_store_tags']), |
| ssn_conf['resource_group_name'], ssn_conf['region']) |
| try: |
| local("~/scripts/{}.py {}".format('ssn_create_datalake', params)) |
| except: |
| traceback.print_exc() |
| raise Exception |
| |
| logging.info('[CREATE DATA LAKE SHARED DIRECTORY]') |
| print('[CREATE DATA LAKE SHARED DIRECTORY]') |
| params = "--resource_group_name {} --datalake_name {} --directory_name {} --service_base_name {} --ad_group {}". \ |
| format(ssn_conf['resource_group_name'], ssn_conf['datalake_store_name'], |
| ssn_conf['datalake_shared_directory_name'], ssn_conf['service_base_name'], |
| os.environ['azure_ad_group_id']) |
| try: |
| local("~/scripts/{}.py {}".format('common_create_datalake_directory', params)) |
| except: |
| traceback.print_exc() |
| raise Exception |
| except Exception as err: |
| traceback.print_exc() |
| dlab.fab.append_result("Failed to create Data Lake Store.", str(err)) |
| for datalake in AzureMeta.list_datalakes(ssn_conf['resource_group_name']): |
| if ssn_conf['datalake_store_name'] == datalake.tags["Name"]: |
| AzureActions.delete_datalake_store(ssn_conf['resource_group_name'], datalake.name) |
| if 'azure_security_group_name' not in os.environ: |
| AzureActions.remove_security_group(ssn_conf['resource_group_name'], ssn_conf['security_group_name']) |
| if 'azure_subnet_name' not in os.environ: |
| AzureActions.remove_subnet(ssn_conf['resource_group_name'], ssn_conf['vpc_name'], |
| ssn_conf['subnet_name']) |
| if 'azure_vpc_name' not in os.environ: |
| AzureActions.remove_vpc(ssn_conf['resource_group_name'], ssn_conf['vpc_name']) |
| if 'azure_resource_group_name' not in os.environ: |
| AzureActions.remove_resource_group(ssn_conf['resource_group_name'], ssn_conf['region']) |
| sys.exit(1) |
| |
| if os.environ['conf_os_family'] == 'debian': |
| initial_user = 'ubuntu' |
| sudo_group = 'sudo' |
| if os.environ['conf_os_family'] == 'redhat': |
| initial_user = 'ec2-user' |
| sudo_group = 'wheel' |
| |
| try: |
| logging.info('[CREATE SSN INSTANCE]') |
| print('[CREATE SSN INSTANCE]') |
| params = "--instance_name {} --instance_size {} --region {} --vpc_name {} --network_interface_name {} \ |
| --security_group_name {} --subnet_name {} --service_base_name {} --resource_group_name {} \ |
| --dlab_ssh_user_name {} --public_ip_name {} --public_key '''{}''' --primary_disk_size {} \ |
| --instance_type {} --instance_storage_account_type {} --image_name {} --tags '{}'".\ |
| format(ssn_conf['instance_name'], os.environ['azure_ssn_instance_size'], ssn_conf['region'], |
| ssn_conf['vpc_name'], ssn_conf['network_interface_name'], ssn_conf['security_group_name'], |
| ssn_conf['subnet_name'], ssn_conf['service_base_name'], ssn_conf['resource_group_name'], |
| initial_user, ssn_conf['static_public_ip_name'], ssn_conf['public_ssh_key'], |
| ssn_conf['primary_disk_size'], 'ssn', ssn_conf['instance_storage_account_type'], |
| ssn_conf['ssn_image_name'], json.dumps(ssn_conf['instance_tags'])) |
| local("~/scripts/{}.py {}".format('common_create_instance', params)) |
| except Exception as err: |
| traceback.print_exc() |
| dlab.fab.append_result("Failed to create instance.", str(err)) |
| try: |
| AzureActions.remove_instance(ssn_conf['resource_group_name'], ssn_conf['instance_name']) |
| except: |
| print("The instance {} hasn't been created".format(ssn_conf['instance_name'])) |
| for datalake in AzureMeta.list_datalakes(ssn_conf['resource_group_name']): |
| if ssn_conf['datalake_store_name'] == datalake.tags["Name"]: |
| AzureActions.delete_datalake_store(ssn_conf['resource_group_name'], datalake.name) |
| if 'azure_security_group_name' not in os.environ: |
| AzureActions.remove_security_group(ssn_conf['resource_group_name'], ssn_conf['security_group_name']) |
| if 'azure_subnet_name' not in os.environ: |
| AzureActions.remove_subnet(ssn_conf['resource_group_name'], ssn_conf['vpc_name'], |
| ssn_conf['subnet_name']) |
| if 'azure_vpc_name' not in os.environ: |
| AzureActions.remove_vpc(ssn_conf['resource_group_name'], ssn_conf['vpc_name']) |
| if 'azure_resource_group_name' not in os.environ: |
| AzureActions.remove_resource_group(ssn_conf['resource_group_name'], ssn_conf['region']) |
| |
| sys.exit(1) |