1:dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Zddl m Z dd lmZdd lmZdd lmZdd lmZdd lmZddlmZdZdZdZd!dZd"dZd#dZ d$dZ d$dZ!dZ" d$dZ#dZ$dZ%dZ&dZ' d%dZ(dZ) d$d Z*y)&z Utilities for all CRUD commands.)absolute_import)division)unicode_literals)encoding) exceptions)util) operations)waiter) constants) pretty_print)log) resources) console_iocXtjj|jdS)Nz+gkemulticloud.projects.locations.operations collection)rREGISTRYParseRelativeNamename)ops Flib/googlecloudsdk/command_lib/container/gkemulticloud/command_util.py_GetOperationResourcer"s*    - -ggG . cd}|j(tj|j}d|vr|d}tjj |dS)Ntargetz1gkemulticloud.projects.locations.attachedClustersr)metadatarMessageToPyValuerrr)r op_clusterrs r_GetOperationTargetr (s[*[[((5H8H%j    - -P . rct|}tj|tj|j ry|s't j}|j||yy)N)kind) rr CreatedResourcer LRO_KINDdone op_api_utilOperationsClientWait)rasync_messageop_ref op_clients r_LogAndWaitForOperationr-3sR  $&f9#5#56WW ,,.I NN67# rNcdj|}|rdj||z}|r|r|dj||z }|S)Nzcluster [{name}]r {action} action in {kind} region [{region}]r"regionformat)rr2r"r5msgs rClusterMessager9?sU!!t!,#   F  +c 1C f ) 0 0d6 0 JJC *rcdj|}|rdj||z}|r|dj|z }|r|r|dj||z }|S) Nznode pool [{name}]r/r0r1z in cluster [{cluster}])clusterr3r4r6)rr2r;r"r5r8s rNodePoolMessager<Hsq###.#   F  +c 1C  $ + +G + <SsQ  d +#   F  +c 1C   " "& " 11C *rc|j||}t|dd}|rd|_yt|dd}t|||t j ||||j |S)aRuns a create command for gkemulticloud. Args: resource_ref: obj, resource reference. resource_client: obj, client for the resource. args: obj, arguments parsed from the command. kind: str, the kind of resource e.g. AWS Cluster, Azure Node Pool. message: str, message to display while waiting for LRO to complete. Returns: The details of the created resource. validate_onlyFdisableNr)r"is_async)Creategetattrr7r-r r#Get resource_refresource_clientargsr"r*rr@r)s rrDrD\nlD1"$7-DK 45 )&"fg.l?   \ **rc|j||}t|dd}|rd|_yt|dd}t|||t j ||||j |S)aRuns an update command for gkemulticloud. Args: resource_ref: obj, resource reference. resource_client: obj, client for the resource. args: obj, arguments parsed from the command. kind: str, the kind of resource e.g. AWS Cluster, Azure Node Pool. message: str, message to display while waiting for LRO to complete. Returns: The details of the updated resource. r@FrANr)rB)UpdaterEr7r-r UpdatedResourcerFrGs rrMrMvrKrcd}|tjk(s&|tjk(s|tjk(r|j d}n\|tj k(s|tj k(r|j d}n$|tjk(r|j d}tjtj||ddy)z)Generates a delete prompt for a resource.z!The following {} will be deleted.clusters node poolclientTr*throw_if_unattended cancel_on_noN) r AWS_CLUSTER_KINDAZURE_CLUSTER_KINDATTACHED_CLUSTER_KINDr7AWS_NODEPOOL_KINDAZURE_NODEPOOL_KINDAZURE_CLIENT_KINDrPromptContinuegke_util ConstructList)r"itemstitles r _DeletePromptras -% i((( -- - 00 0 LL $E i))) .. . LL %E y*** LL "E $$UE2rct|dd}|s t||gt|dd}t|dd}t|dd}|j||||} |rd|_yt | |d |zt j ||| y) aRuns a delete command for gkemulticloud. Args: resource_ref: obj, resource reference. resource_client: obj, client for the resource. args: obj, arguments parsed from the command. kind: str, the kind of resource e.g. AWS Cluster, Azure Node Pool. message: str, message to display while waiting for LRO to complete. Returns: The details of the updated resource. r@Fr) allow_missing ignore_errors)r@rcrdrANz Deleting rB)rEraDeleter7r-r DeletedResource) rHrIrJr"r*r@r)rcrdrs rreres$7- $ " 45 )&$7-$7-!!! " DK "fkG&;<l?rcd}|tjk(s|tjk(rd}|js t ||| t |||||}|S#t j$r7}|js$tjdj||d}~wtj$r7}|js$tjdj||d}~wwxYw)aCalls the delete command and suggests using --ignore-errors on failure. Args: args: obj, arguments parsed from the command. resource_client: obj, client for the resource. resource_ref: obj, resource reference. message: str, message to display while waiting for LRO to complete. kind: str, the kind of resource e.g. AWS Cluster, Azure Node Pool. Returns: The details of the updated resource. r;rQ)rHrIrJr*r"z9Delete {} failed. Try re-running with `--ignore-errors`. N) r rYrZrd_PromptIgnoreErrorsrer OperationErrorr Infor7apitools_exceptions HttpError)rJrIrHr*r"resretes rDeleteWithIgnoreErrorsrps # i))) .. . C  o|< !'    C. *!       F M M  G  & &    F M M  G s$AC)*2BC)22C$$C)c | |j|}t j }|j jj|j jj|jjj|jjj|jjj|jjj|jjj|jjj|jjj|jjjg }|j|vrytj dddd|_y#tj$r}|jdk(rYd}~y|d}~wwxYw)zNPrompt for --ignore-errors flag if the resource is in ERROR or DEGRADED state.iNzQCluster or node pool is in ERROR or DEGRADED state. Setting --ignore-errors flag.TF)r*rTrUdefault)rFrkrl status_coderGetMessagesModule)GoogleCloudGkemulticloudV1AttachedClusterStateValueValuesEnumERRORDEGRADED$GoogleCloudGkemulticloudV1AwsCluster%GoogleCloudGkemulticloudV1AwsNodePool&GoogleCloudGkemulticloudV1AzureCluster'GoogleCloudGkemulticloudV1AzureNodePoolstaterr\rd)rJrIrHrespromessages error_statess rrhrhs`   | ,D  # # %(88MMSS88MMVV33HHNN33HHQQ44IIOO44IIRR55JJPP55JJSS66KKQQ66KKTT , ZZ|# !00 ,$'  & & }} G sF F; F64F66F;c,d}|j|||S)zMessage to display after cancelling an LRO operation. Args: name: str, name of the operation. kind: str, the kind of LRO operation e.g. AWS or Azure. Returns: The operation cancellation message. zCancelation of operation {0} has been requested. Please use gcloud container {1} operations describe {2} to check if the operation has been cancelled successfully.r6)rr"r8s rCancelOperationMessagers"@ D$ %%rcVd}tj|j|ddy)zhPrompt the user before cancelling an LRO operation. Args: op_name: str, name of the operation. z$The operation {0} will be cancelled.TrSN)rr\r7)op_namer*s rCancelOperationPromptr-s) 3' nnW%rc|j|||}t|dd}|rd|_yt|dd}t|||t |} t j | ||y)aRuns an import command for gkemulticloud. Args: location_ref: obj, location reference. resource_client: obj, client for the resource. fleet_membership_ref: obj, fleet membership reference. args: obj, arguments parsed from the command. kind: str, the kind of resource e.g. AWS Cluster, Azure Node Pool. message: str, message to display while waiting for LRO to complete. Returns: The details of the imported resource. ) location_reffleet_membership_refrJr@FrANr)rB)ImportrEr7r-r r ImportResource) rrIrrJr"r*rr@r) op_targets rrr;su*/ " $7-DK 45 )&"fg.!"%)YTF;rc`d}tjtj||ddy)z7Generates a rollback prompt for the node pool resource.z,The following node pool will be rolled back.TrSN)rr\r]r^)r_r`s r_RollbackPromptr_s+ 8% $$UE2rct|gt|dd}|j||}t||d|zt j ||||j |S)aRuns a rollback command for gkemulticloud. Args: resource_ref: obj, resource reference. resource_client: obj, client for the resource. args: obj, arguments parsed from the command. kind: str, the kind of resource e.g. AWS Cluster, Azure Node Pool. message: str, message to display while waiting for LRO to complete. Returns: The details of the updated resource. r)Fz Rolling back rB)rrERollbackr-r rNrF)rHrIrJr"r*r)rs rrrisb7) 45 )& d3""fo&?@l?   \ **r)NNN)NNNN)NN)NNNNN)NNNNNN)+__doc__ __future__rrrapitools.base.pyrrrk googlecloudsdk.api_lib.containerrr].googlecloudsdk.api_lib.container.gkemulticloudr r&googlecloudsdk.api_lib.utilr 2googlecloudsdk.command_lib.container.gkemulticloudr googlecloudsdk.command_lib.runr googlecloudsdk.corer rgooglecloudsdk.core.consolerrr r-r9r<r>rDrMrarerprhrrrrrrrrs'&'%>=T?.H7#)2  $   LP+6LP+40LP@D- `D&$   !