tdZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z dd l m Z dd l m Z ddlm Zdd lmZdd lmZddlZdd lmZddddZdZdZdZdZdZdZGdde j:ZdZdZ dZ!ddddZ" d+d Z#efd!Z$d"Z%d#Z&d$Z'd%Z(d&Z) d,d'Z*Gd(d)ejVZ,e,jZfd*Z.y)-zDConvenience functions and classes for dealing with instances groups.)absolute_import)division)unicode_literalsN)encoding) exceptions)lister)path_simplifier)utils)log) properties)rangez4Lists the named ports for an instance group resourcea Named ports are key:value pairs metadata representing the service name and the port that it's running on. Named ports can be assigned to an instance group, which indicates that the service is available on all instances in the group. This information is used by Application Load Balancers and proxy Network Load Balancers. *{command}* lists the named ports (name and port tuples) for an instance group. z For example, to list named ports for an instance group: $ {command} example-instance-group --zone=us-central1-a The above example lists named ports assigned to an instance group named 'example-instance-group' in the ``us-central1-a'' zone. )brief DESCRIPTIONEXAMPLESic(|jdk(S)z#Checks if group reference is zonal.compute.instanceGroups) Collection) group_refs ;lib/googlecloudsdk/api_lib/compute/instance_groups_utils.py IsZonalGroupr=s    #; ;;c|Dcgc]!}|j|k7r|j#}}t|r(tjdddj |zycc}w)aDValidate if provided list in zone given as parameter. Args: instances: list of instances resources to be validated zone: a zone all instances must be in order to pass validation Raises: InvalidArgumentException: If any instance is in different zone than given as parameter. instanceszaThe zone of instance must match the instance group zone. Following instances has invalid zone: %sz, N)zoneSelfLinkanycalliope_exceptionsInvalidArgumentExceptionjoin)rrinstinvalid_instancess rValidateInstanceInZoner"Bsx$-C#,4 T0A}}#,C   6 63 ))% &' ((Cs&A!c#DK|D]}t||D]}|yw)zDExtracts items stored in given attribute of instance group response.N)getattr) responses attr_nameresponseitems rUnwrapResponser)Vs&h), j-s c|jS)aNUriFunc for listing instance-group related subresources. Function returns field with URI for objects being subresources of instance-groups, with instance fields. Works for list-instances and instance-configs list commands. Args: resource: instance-group subresource with instance field Returns: URI of instance )instance)resources r$UriFuncForListInstanceRelatedObjectsr-]s   rc|j}|jdk(rI|j}|jd|j |j |j }nH|j}|jd|j |j|j }|j|d|fg}tt|dS)z)Gets the request to fetch instance group.rGet) instanceGrouprproject)r0regionr1requests namedPorts) apitools_clientrinstanceGroupsGetRequestTypeNamerr1regionInstanceGroupsr2 MakeRequestslistr))rcompute_clientcomputeservicerequestresultss rOutputNamedPortsForGrouprBms  * *'77$$G+g$$U+nn& ^^!!#G **G+g$$U+nn&!!#G  ' '7E72K1L ' M' nWl3 44rceZdZdZy)FingerprintFetchExceptionzBException thrown when there is a problem with getting fingerprint.N)__name__ __module__ __qualname____doc__rrrDrDsJrrDc|j}t|rI|j}|jj |j |j |j}nH|j}|jj|j |j|j}g}|j|d|fg|}|rtj|td|djS)z)Gets fingerprint of given instance group.)r1rr0)r1r2r0r/)r4errors_to_collectz'Could not set named ports for resource:) error_messager)r6rr7MESSAGES_MODULEComputeInstanceGroupsGetRequestr1rr0r:%ComputeRegionInstanceGroupsGetRequestr2r;r RaiseExceptionrD fingerprint)r=rr>r?r@errors resourcess r_GetGroupFingerprintrTs  * *')$$G%%EE!! ^^--F/G **G%%KK!!--L/G &))%)** )  !?A 1 ! !!rc|j}|j}t||}t|rV|j ||}|j |j ||j|j|jfS|j||}|j|j ||j|j|jfS)aReturns a request to get named ports and service to send request. Args: compute_client: GCE API client, group_ref: reference to instance group (zonal or regional), ports: list of named ports to set Returns: request, message to send in order to set named ports on instance group, service, service where request should be sent - regionInstanceGroups for regional groups - instanceGroups for zonal groups )rQr5)r0"instanceGroupsSetNamedPortsRequestrr1)r0(regionInstanceGroupsSetNamedPortsRequestr2r1)r6messagesrTr"InstanceGroupsSetNamedPortsRequest)ComputeInstanceGroupsSetNamedPortsRequestr9rr1r7(RegionInstanceGroupsSetNamedPortsRequest/ComputeRegionInstanceGroupsSetNamedPortsRequestr2r:)r=rportsr>rXrQ request_bodys rGetSetNamedPortsRequestForGroupr_s  * *'  $ $( %^Y?+)>>?L  = =nn&+7 ^^!! > #%,$:$:  ;; DDEL  C Cnn&1=!! D #%,$@$@  AArc 6g}|D]}|jddk7rtj|d|jd\}}|j stj|d|j |j |t||S)zValidates named ports flags.:z+Named ports should follow NAME:PORT format.)nameport)countrrsplitisdigitappend NamedPortint)rX named_portsr] named_porthostrds rValidateAndParseNamedPortsArgsrns %j!  8 8 C EE!!#&JD$ <<>  8 8 C EE LL##CI#>?  ,rz2Sets the list of named ports for an instance groupa Named ports are key:value pairs metadata representing the service name and the port that it's running on. Named ports can be assigned to an instance group, which indicates that the service is available on all instances in the group. This information is used by Application Load Balancers and proxy Network Load Balancers. *{command}* sets the list of named ports for all instances in an instance group. Note: Running this command will clear all existing named ports. a2 For example, to apply the named ports to an entire instance group: $ {command} example-instance-group --named-ports=example-service:1111 --zone=us-central1-a The above example will assign a name 'example-service' for port 1111 to the instance group called 'example-instance-group' in the ``us-central1-a'' zone. The command removes any named ports that are already set for this instance group. To clear named ports from instance group provide empty named ports list as parameter: $ {command} example-instance-group --named-ports="" --zone=us-central1-a cZtjdddg}g}|Dcgc]}tj|} }|j dk(r[| D]T} |j | |j |jdd} |j|| | jV|S|j d k(rA|jj} | jd |j|j|j } i}|j| d | fg D]} | jsW|s%t!d j#| j$|g}|jdj#| j$f| j|tj| j<| D]<} | |vr|j|| || $|j|| d>|St!dj#|j cc}w)aCreates reference to instances in instance group (zonal or regional). Args: resources: Resources parser for the client. compute_client: Client for the current release track. igm_ref: URL to the target IGM. instance_names_or_urls: names or full URLs of target instances. skip_instances_on_validation_error: If true, skip instances that are not yet allocated to any zone. This can happen when the instance is being created in a regional IGM by a resize-request and is still in a queue. warnings: A list to collect warnings for skipped instances if skip_instances_on_validation_error is true. Returns: A dict where instance names are keys, and corresponding references are values. Unresolved names are present in dict with value None. InstanceNameWithReference instance_nameinstance_referencecompute.instanceGroupManagers)r1rzcompute.instancesparams collection)rqrr#compute.regionInstanceGroupManagersListManagedInstances)instanceGroupManagerr2r1r3zPCannot perform action on instance {name} as it is not yet allocated to any zone.)rcNzTSkipped performing action on instance {name} as it is not yet allocated to any zone.Unknown reference type {0}) collections namedtupler r9rParser1rrhrr6regionInstanceGroupManagersr8r2r;r+ ValueErrorformatrc)rSr=igm_refinstance_names_or_urls"skip_instances_on_validation_errorwarnings_InstanceNameWithReferenceinstance_references name_or_urlnames_to_resolverq instance_refr?r@resolved_referencess rCreateInstanceReferencesrsp2 +55!O5I#J L0/+;'/ <<) __  ll ) %*l  $)!-!6!6!8 :;* DD,,HHG [(a, b, c)].NrI)r?methodr4r@s rGenerateRequestTuplesr~sg FG $$scDg}|D]B\}}}g}|j|||fg||j||f|j|Dg} |D]O\}}|rd} nd} t||jD]*} | j| t j | | d,Q| S)aMake *-instances requests with feedback per instance. Args: client: Compute client. requests: [(service, method, request)]. instances_holder_field: name of field inside request holding list of instances. errors_to_collect: A list for capturing errors. If any response contains an error, it is added to this list. Returns: A list of request statuses per instance. Requests status is a dictionary object, see SendInstancesRequestsAndPostProcessOutputs for details. FAILSUCCESSselfLink instanceNamestatus)r;rhextendr$rr r9) clientr4instances_holder_fieldrKrequest_resultsr?rr@rRstatus_per_instanceinstance_statusr+s r#MakeRequestsAndGetStatusPerInstancers&/"*gvw F '6734f=GV,-V$ #+(ogv o!oG%;<FF  )..x8#" G ) rct}|jxsgD]}|j|jjk7r|j |j Bd}d}|jxsgD]>}|jdk(r |j}|jdk(s|jdk(s=d}@|r|r|j ||<|j |j |S)a-Extract from operation instances skipped due to graceful validation. Args: operation: Operation containing warnings. warnings_to_collect: A list to collect warnings unrelated to graceful validation. Returns: Dict from resource path of a skipped instance to validation error. NFr+validation_errorvalidation_outcomeT) dictrcodeCodeValueValuesEnumNOT_CRITICAL_ERRORrhmessagedatakeyvalue) operationwarnings_to_collectskipped_instanceswarningskipped_instance_pathis_graceful_validation_warningwarning_metadatas r.ExtractSkippedInstancesAndCollectOtherWarningsrsf##)r)g||w22EEE  1 %*"$LL.B.    + 0 6 6   #5 5   #7 7)-& / &*?18-.  1)** rc g}|D]P\}}}g} |j|||fg| ddd} | xsdg\} |j|| | f|j| Rg} |D]\}} } | rDt||jD]*} | j| t j | dd,N| jr8tjjdj| jt| |}t||jD]R} | | jdd zd}d}||vrd }||}nd }| j| t j | ||d T | S) aMake *-instances requests with feedback per instance. Specialized version of MakeRequestsAndGetStatusPerInstance. Checks operations for warnings presence to evaluate statuses per instance. Gracefully validated requests may produce warnings on operations, indicating instances skipped. It would be merged with MakeRequestsAndGetStatusPerInstance after we see there's no issues with this implementation. Args: client: Compute client. requests: [(service, method, request)]. instances_holder_field: name of field inside request holding list of instances. warnings_to_collect: A list for capturing warnings. If any completed operation will contain skipped instances, function will append warning suggesting how to find additional details on the operation, warnings unrelated to graceful validation will be collected as is. errors_to_collect: A list for capturing errors. If any response contains an error, it is added to this list. Returns: See MakeRequestsAndGetStatusPerInstance. FT) log_warnings no_followupalways_return_operationNrrzUpdated [{0}]. z /projects/rbSKIPPEDr)rrrvalidationError)r;rhrr$rr r9 targetLinkr rwriterrfind)rr4rrrKrr?rr@rR operationsrrr+r instance_pathrrs r0MakeRequestsAndGetStatusPerInstanceFromOperationrs:/"*gvw F$$w&@%A%+2715=A %CJ &KYGY78V$#+$3 gy& g'=>HH("" +00:$  I    +2293G3GHIH (*g'=>HH(!|!?@&I  " "6 ** )rc #K|j}|jdk(r|jj} nR|jdk(r|jj} n(t dj |jg} g} t|j||||| } | D cgc]} | jr | j}} |t||_ t||}t| ||}g}|r|jt|||| | n|jt!|||| | D cgc]} | js | j"}} |j|Dcgc]}t%|dc}|D]}|| r)t'j(t+j,d| | rt+j.| ycc} wcc} wcc}ww)apMake *-instances requests and format output. Method resolves instance references, splits them to make batch of requests, adds to results statuses for unresolved instances, and yields all statuses raising errors, if any, in the end. Args: api_holder: Compute API holder. method_name: Name of the (region) instance groups managers service method to call. request_template: Partially filled *-instances request (no instances). instances_holder_field: Name of the field inside request holding instances field. igm_ref: URL to the target IGM. instances: A list of names of the instances to apply method to. per_instance_status_enabled: Enable functionality parsing resulting operation for graceful validation related warnings to allow per-instance status output. The plan is to gradually enable this for all per-instance commands in GA (even where graceful validation is not available / not used). skip_instances_on_validation_error: If true, skip instances that are not yet allocated to any zone. This can happen when the instance is being created in a regional IGM by a resize-request and is still in a queue. Yields: A list of request statuses per instance. Requests status is a dictionary object with link to an instance keyed with 'selfLink', instance name keyed with 'instanceName', and status indicating if operation succeeded for instance keyed with 'status'. Status might be 'FAIL', 'SUCCESS', 'SKIPPED' in case of graceful validation, or 'MEMBER_NOT_FOUND' (in case of regional MIGs, when instance name cannot be resolved). rsrwrzMEMBER_NOT_FOUND)rrz!Some requests generated warnings:N)rrr6rr~rrrrSrrr$rrrrrrrqrr rr ConstructListr)rrrrrrper_instance_status_enabledrrr?rKrinstances_with_referencesr+rr4request_tuplesrequest_status_per_instanceunresolved_instance_namesrcrs r*SendInstancesRequestsAndPostProcessOutputsrLs%R   & <<$$::GDD$$@@G 1889K9K9MN OO6  ( 0/(  $ $!!/ /B  ""+ $%57M N((+xH. " &&8 N$:j@k(rI|s|t>jBk(r_|rdnd|d<|r||d<|jE||Scc}wcc}wcc}w)a;Add information if instance group is managed. Args: compute_holder: ComputeApiHolder, The compute API holder items: list of instance group messages, filter_mode: InstanceGroupFilteringMode, managed/unmanaged filtering options Returns: list of instance groups with computed dynamic properties rr1z compute.zonesrtN)r?r1requested_zones filter_exprhttp batch_urlrRr:r2zcompute.regions)rv)r?r1requested_regionsrrrrRrz/instanceGroups/z/instanceGroupManagers/YesNo isManagedinstanceGroupManagerUri)#rrSr<setr}r VALUEScorer1 GetOrFailaddrsix iteritemsrrGetZonalResourcesr6rrrhasattrr2GetRegionalResourcesr~r ScopedSuffixrr rreplacerrrrh)compute_holderitems filter_moderrSrRig zone_linksproject_to_zonesrzone_refzonal_instance_group_managersr1zones regional_instance_group_managers region_linksproject_to_regionsr2 region_refregionsinstance_group_managersigminstance_group_managers_refsrAr( self_link igm_self_link scoped_suffix is_manageds r%ComputeInstanceGroupManagerMembershipr sm  &&&) & u+%?2&B,BvJ?@*d z((--55?? # $H //+.5x''(X%%&**8==9#% &67ngu!(()A)A&&<<  # # ( (""*8&(" V # #%;<uGuB8 uGHL??66G?Hj   #5 514:--.++,001B1BC   MM*<=&--f.I.I((DD#%%**$$/> () -./"%(&*(#""3<<0(&*"+  V$ 'dZ I%%57M#00?M">>J1GGG 3GG G",$D(5d $% NN4#& .W@8H(&*s M$M$= M)M)&M.)FN)FF)/rH __future__rrrr{enumapitools.base.pyrgooglecloudsdk.api_lib.computerrr r googlecloudsdk.calliopergooglecloudsdk.corer r r six.movesr +INSTANCE_GROUP_GET_NAMED_PORT_DETAILED_HELPINSTANCES_MAX_LENGTHrr"r)r-rBErrorrDrTr_rnSET_NAMED_PORTS_HELPrrrrrrrrEnumrrr rIrrrsK&' %51:0E#* D /+.< ((  5&K 0 0K":&AR  B L(-  XPz(<&R%)X"JNb" X!&',d6N(B'L'L[r