def CreateAutoscalerResource(self, igm_ref, args):
if _IsZonalGroup(igm_ref):
scope_type = 'zone'
location = managed_instance_groups_utils.CreateZoneRef(
self.resources, igm_ref)
zones, regions = [location], None
else:
scope_type = 'region'
location = managed_instance_groups_utils.CreateRegionRef(
self.resources, igm_ref)
zones, regions = None, [location]
autoscaler = managed_instance_groups_utils.AutoscalerForMig(
mig_name=igm_ref.Name(),
autoscalers=managed_instance_groups_utils.AutoscalersForLocations(
regions=regions,
zones=zones,
compute=self.compute,
http=self.http,
batch_url=self.batch_url),
location=location,
scope_type=scope_type)
autoscaler_name = getattr(autoscaler, 'name', None)
new_one = autoscaler_name is None
autoscaler_name = autoscaler_name or args.name
if _IsZonalGroup(igm_ref):
autoscaler_resource = managed_instance_groups_utils.BuildAutoscaler(
args, self.messages, igm_ref, autoscaler_name)
else:
autoscaler_resource = managed_instance_groups_utils.BuildAutoscaler(
args, self.messages, igm_ref, autoscaler_name)
return autoscaler_resource, new_one
def _GenerateAutoscalerDeleteRequests(self, holder, project, mig_requests):
"""Generates Delete requestes for autoscalers attached to instance groups.
Args:
holder: ComputeApiHolder, object encapsulating compute api.
project: str, project this request should apply to.
mig_requests: Messages which will be sent to delete instance group
managers.
Returns:
Messages, which will be sent to delete autoscalers.
"""
mig_requests = zip(*mig_requests)[2] if mig_requests else []
zone_migs = [(request.instanceGroupManager, 'zone',
managed_instance_groups_utils.CreateZoneRef(
holder.resources, request))
for request in mig_requests
if hasattr(request, 'zone') and request.zone is not None]
region_migs = [
(request.instanceGroupManager, 'region',
managed_instance_groups_utils.CreateRegionRef(
holder.resources, request)) for request in mig_requests
if hasattr(request, 'region') and request.region is not None
]
zones = zip(*zone_migs)[2] if zone_migs else []
regions = zip(*region_migs)[2] if region_migs else []
client = holder.client.apitools_client
messages = client.MESSAGES_MODULE
autoscalers_to_delete = managed_instance_groups_utils.AutoscalersForMigs(
migs=zone_migs + region_migs,
autoscalers=managed_instance_groups_utils.AutoscalersForLocations(
zones=zones,
regions=regions,
compute=client,
http=client.http,
batch_url=holder.client.batch_url))
requests = []
for autoscaler in autoscalers_to_delete:
if autoscaler.zone:
service = client.autoscalers
request = messages.ComputeAutoscalersDeleteRequest(
zone=path_simplifier.Name(autoscaler.zone))
else:
service = client.regionAutoscalers
request = messages.ComputeRegionAutoscalersDeleteRequest(
region=path_simplifier.Name(autoscaler.region))
request.autoscaler = autoscaler.name
request.project = project
requests.append((service, 'Delete', request))
return requests
def GetAutoscalerResource(self, client, resources, igm_ref, args):
if _IsZonalGroup(igm_ref):
scope_type = 'zone'
location = managed_instance_groups_utils.CreateZoneRef(
resources, igm_ref)
zones, regions = [location], None
else:
scope_type = 'region'
location = managed_instance_groups_utils.CreateRegionRef(
resources, igm_ref)
zones, regions = None, [location]
autoscaler = managed_instance_groups_utils.AutoscalerForMig(
mig_name=args.name,
autoscalers=managed_instance_groups_utils.AutoscalersForLocations(
regions=regions, zones=zones, client=client),
location=location,
scope_type=scope_type)
if autoscaler is None:
raise managed_instance_groups_utils.ResourceNotFoundException(
'The managed instance group is not autoscaled.')
return autoscaler
