def power_off_vm(self, command_context):
"""Power off Azure VM
:param ResourceCommandContext command_context:
:return:
"""
with LoggingSessionContext(command_context) as logger:
with ErrorHandlingContext(logger):
logger.info('Starting power off operation on Azure VM...')
group_name = self.model_parser.convert_to_reservation_model(command_context.remote_reservation) \
.reservation_id
resource = command_context.remote_endpoints[0]
data_holder = self.model_parser.convert_app_resource_to_deployed_app(
resource)
vm_name = data_holder.name
with CloudShellSessionContext(
command_context) as cloudshell_session:
cloud_provider_model = self.model_parser.convert_to_cloud_provider_resource_model(
resource=command_context.resource,
cloudshell_session=cloudshell_session)
azure_clients = AzureClientsManager(cloud_provider_model)
self.power_vm_operation.power_off(
compute_client=azure_clients.compute_client,
resource_group_name=group_name,
vm_name=vm_name)
logger.info(
'Azure VM {} was successfully powered off'.format(vm_name))
def delete_azure_vm(self, command_context):
with LoggingSessionContext(command_context) as logger:
with ErrorHandlingContext(logger):
logger.info('Deleting Azure VM...')
resource_group_name = command_context.remote_reservation.reservation_id
with CloudShellSessionContext(
command_context) as cloudshell_session:
cloud_provider_model = self.model_parser.convert_to_cloud_provider_resource_model(
resource=command_context.resource,
cloudshell_session=cloudshell_session)
azure_clients = AzureClientsManager(cloud_provider_model)
vm_name = command_context.remote_endpoints[0].fullname
self.delete_azure_vm_operation.delete(
compute_client=azure_clients.compute_client,
network_client=azure_clients.network_client,
storage_client=azure_clients.storage_client,
group_name=resource_group_name,
vm_name=vm_name,
logger=logger)
logger.info('End Deleting Azure VM')
def cleanup_connectivity(self, command_context, request):
with LoggingSessionContext(command_context) as logger:
with ErrorHandlingContext(logger):
logger.info('Teardown...')
with CloudShellSessionContext(
command_context) as cloudshell_session:
cloud_provider_model = self.model_parser.convert_to_cloud_provider_resource_model(
resource=command_context.resource,
cloudshell_session=cloudshell_session)
azure_clients = AzureClientsManager(cloud_provider_model)
resource_group_name = command_context.reservation.reservation_id
cleanup_connectivity_request = getattr(
DeployDataHolder(jsonpickle.decode(request)),
'driverRequest', None)
result = self.delete_azure_vm_operation.cleanup_connectivity(
network_client=azure_clients.network_client,
resource_client=azure_clients.resource_client,
cloud_provider_model=cloud_provider_model,
resource_group_name=resource_group_name,
request=cleanup_connectivity_request,
logger=logger)
logger.info('End Teardown')
return self.command_result_parser.set_command_result(
{'driverResponse': {
'actionResults': [result]
}})
def _validate_api_credentials(self, cloud_provider_model, logger):
"""Verify Azure API Credentials and return AzureClientsManager instance
:param cloud_provider_model: cloudshell.cp.azure.models.AzureCloudProviderResourceModel instance
:param logger: logging.Logger instance
:return: cloudshell.cp.azure.common.azure_clients.AzureClientsManager instance
"""
try:
return AzureClientsManager(cloud_provider_model)
except AuthenticationError:
error_msg = "Failed to connect to Azure API, please check the log for more details"
logger.exception(error_msg)
raise AutoloadException(error_msg)
def get_vm_details(self, command_context, cancellation_context,
requests_json):
"""Get vm details for specific deployed app
:param requests_json:
:param cancellation_context:
:param command_context: ResourceRemoteCommandContext
"""
with LoggingSessionContext(command_context) as logger:
with ErrorHandlingContext(logger):
logger.info("Starting get_vm_details operation...")
requests = DeployDataHolder(
jsonpickle.decode(requests_json)).items
group_name = self.model_parser.convert_to_reservation_model(command_context.reservation) \
.reservation_id
# resource = command_context.remote_endpoints[0]
# data_holder = self.model_parser.convert_app_resource_to_deployed_app(resource)
# vm_name = data_holder.name
# data_holder_request = self.model_parser.convert_app_resource_to_request(resource)
# deployment_service = data_holder_request.deploymentService
# is_market_place = filter(lambda x: x.name == "Image SKU", deployment_service.attributes)
with CloudShellSessionContext(
command_context) as cloudshell_session:
cloud_provider_model = self.model_parser.convert_to_cloud_provider_resource_model(
resource=command_context.resource,
cloudshell_session=cloudshell_session)
azure_clients = AzureClientsManager(cloud_provider_model)
vm_details = self.vm_details_operation.get_vm_details(
compute_client=azure_clients.compute_client,
group_name=group_name,
requests=requests,
logger=logger,
network_client=azure_clients.network_client,
model_parser=self.model_parser,
cancellation_context=cancellation_context)
return self.command_result_parser.set_command_result(
vm_details)
def refresh_ip(self, command_context):
"""Refresh private and public IPs on the Cloudshell resource
:param ResourceRemoteCommandContext command_context:
"""
with LoggingSessionContext(command_context) as logger:
with ErrorHandlingContext(logger):
logger.info("Starting Refresh IP operation...")
resource = command_context.remote_endpoints[0]
data_holder = self.model_parser.convert_app_resource_to_deployed_app(
resource)
vm_name = data_holder.name
group_name = self.model_parser.convert_to_reservation_model(command_context.remote_reservation) \
.reservation_id
private_ip = self.model_parser.get_private_ip_from_connected_resource_details(
command_context)
public_ip = self.model_parser.get_public_ip_from_connected_resource_details(
command_context)
resource_fullname = self.model_parser.get_connected_resource_fullname(
command_context)
with CloudShellSessionContext(
command_context) as cloudshell_session:
cloud_provider_model = self.model_parser.convert_to_cloud_provider_resource_model(
resource=command_context.resource,
cloudshell_session=cloudshell_session)
azure_clients = AzureClientsManager(cloud_provider_model)
self.refresh_ip_operation.refresh_ip(
cloudshell_session=cloudshell_session,
compute_client=azure_clients.compute_client,
network_client=azure_clients.network_client,
resource_group_name=group_name,
vm_name=vm_name,
private_ip_on_resource=private_ip,
public_ip_on_resource=public_ip,
resource_fullname=resource_fullname,
logger=logger)
logger.info(
'Azure VM IPs were successfully refreshed'.format(vm_name))
def deploy_vm_from_custom_image(self, command_context, deploy_action,
cancellation_context):
"""Deploy Azure Image from given Image URN
:param ResourceCommandContext command_context: ResourceCommandContext instance
:param cloudshell.cp.core.models.DeployApp deploy_action: describes the desired deployment
:param CancellationContext cancellation_context:
:return:
"""
with LoggingSessionContext(command_context) as logger:
with ErrorHandlingContext(logger):
logger.info('Deploying Azure VM From Custom Image...')
with CloudShellSessionContext(
command_context) as cloudshell_session:
azure_vm_deployment_model = self.model_parser. \
convert_to_deploy_azure_vm_from_custom_image_resource_model(
deploy_action=deploy_action,
cloudshell_session=cloudshell_session,
logger=logger)
cloud_provider_model = self.model_parser.convert_to_cloud_provider_resource_model(
resource=command_context.resource,
cloudshell_session=cloudshell_session)
azure_clients = AzureClientsManager(cloud_provider_model)
result = self.deploy_azure_vm_operation.deploy_from_custom_image(
deployment_model=azure_vm_deployment_model,
cloud_provider_model=cloud_provider_model,
reservation=self.model_parser.convert_to_reservation_model(
command_context.reservation),
network_client=azure_clients.network_client,
compute_client=azure_clients.compute_client,
storage_client=azure_clients.storage_client,
cancellation_context=cancellation_context,
logger=logger,
cloudshell_session=cloudshell_session)
logger.info('End deploying Azure VM From Custom Image')
return result
def get_access_key(self, command_context):
"""Returns public key
:param ResourceRemoteCommandContext command_context:
:rtype str:
"""
with LoggingSessionContext(command_context) as logger:
with ErrorHandlingContext(logger):
logger.info("Starting GetAccessKey...")
with CloudShellSessionContext(
command_context) as cloudshell_session:
cloud_provider_model = self.model_parser.convert_to_cloud_provider_resource_model(
resource=command_context.resource,
cloudshell_session=cloudshell_session)
azure_clients = AzureClientsManager(cloud_provider_model)
resource_group_name = \
self.model_parser.convert_to_reservation_model(command_context.remote_reservation).reservation_id
return self.access_key_operation.get_access_key(
storage_client=azure_clients.storage_client,
group_name=resource_group_name)
def prepare_connectivity(self, context, actions, cancellation_context):
"""
Creates a connectivity for the Sandbox:
1.Resource group
2.Storage account
3.Key pair
4.Network Security Group
5.Creating a subnet under the
:param context:
:param actions: list[cloudshell.cp.core.models.RequestActionBase]
:param cancellation_context cloudshell.shell.core.driver_context.CancellationContext instance
:return:
"""
with LoggingSessionContext(context) as logger:
with ErrorHandlingContext(logger):
logger.info('Preparing Connectivity for Azure VM...')
with CloudShellSessionContext(context) as cloudshell_session:
cloud_provider_model = self.model_parser.convert_to_cloud_provider_resource_model(
resource=context.resource,
cloudshell_session=cloudshell_session)
azure_clients = AzureClientsManager(cloud_provider_model)
result = self.prepare_connectivity_operation.prepare_connectivity(
reservation=self.model_parser.convert_to_reservation_model(
context.reservation),
cloud_provider_model=cloud_provider_model,
storage_client=azure_clients.storage_client,
resource_client=azure_clients.resource_client,
network_client=azure_clients.network_client,
logger=logger,
actions=actions,
cancellation_context=cancellation_context)
logger.info('End Preparing Connectivity for Azure VM')
return result