(TdZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z ddlm Zdd lmZdd lmZdd lmZdd lmZdd lmZddlZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#d(dZ$d)dZ%dZ&dZ'dZ(dZ)d(dZ*dZ+d Z,d!Z-d)d"Z.d#Z/d$Z0d%Z1d*d&Z2d'Z3y)+z2Common helper methods for Runtime Config commands.)absolute_import)division)unicode_literalsN)encoding) exceptions)apis)log) properties) resources)progress_tracker)retryzread operation timed outic(djd|gS)N/projects)join)projects 1lib/googlecloudsdk/api_lib/runtime_config/util.py ProjectPathr5s :w' ((c<djt|d|gS)Nrconfigs)rr)rconfigs r ConfigPathr9s ;w'F; <djt||d|gS)Nrwaiters)rr)rrwaiters r WaiterPathr#Bs :gv. 6B CCrcVdti}tjj|d|S)zParse a config name or URL, and return a resource. Args: config_name: The config name. Returns: The parsed resource. projectsIdzruntimeconfig.projects.configs collectionparams)Projectr REGISTRYParse) config_namer(s rParseConfigNamer-Gs8G &    ! !+-M)/ " 11rc^fdfdd}tjj|d|S)zParse a variable name or URL, and return a resource. Args: variable_name: The variable name. args: CLI arguments, possibly containing a config name. Returns: The parsed resource. c@ttjSNr- ConfigNamer%argssrz#ParseVariableName..hOJt,<=HHrc@ttjSr0r-r2 configsIdr3srr5z#ParseVariableName..i?:d+;<FFrr%r9z(runtimeconfig.projects.configs.variablesr&r r*r+) variable_namer4r(s ` rParseVariableNamer>Xs< IF &    ! !;  " rc^fdfdd}tjj|d|S)zParse a waiter name or URL, and return a resource. Args: waiter_name: The waiter name. args: CLI arguments, possibly containing a config name. Returns: The parsed resource. c@ttjSr0r1r3srr5z!ParseWaiterName..}r6rc@ttjSr0r8r3srr5z!ParseWaiterName..~r:rr;z&runtimeconfig.projects.configs.waitersr&r<) waiter_namer4r(s ` rParseWaiterNamerCrs<IF &    ! !9  " rcf|r#t|ddstjddt|ddS)Nr,rz$--config-name parameter is required.)getattrsdk_exceptionsRequiredArgumentException)r4requireds rr2r2s: gdM48  2 28 :: }d ++rcjtjdd}|||j_|||_|SN runtimeconfigv1beta1)rGetClientInstancehttptimeout num_retries)rOrPclients rClientrRs9  ! !/9 =& !FKK$F -rc ,tdi|jSN)rRprojects_configskwargss r ConfigClientrYs  &  * **rc ,tdi|jSrT)rRprojects_configs_variablesrWs rVariableClientr\s  &  4 44rc ,tdi|jSrT)rRprojects_configs_waitersrWs r WaiterClientr_s  &  2 22rc.tjddSrJ)rGetMessagesModulerUrrMessagesrbs   ;;rcjtjjjj |S)NrH)r VALUEScorerGetrds rr)r)s(      ' ' + +X + >>rc"t|dddk(S)N status_codeirEerrors rIsBadGatewayErrorrm  t , 33rc"t|dddk(S)Nriirjrks rIsDeadlineExceededErrorrprnrcpt|tjxstt j |vSr0) isinstancesocketrOTIMEOUT_ERR_TEXTsix text_typerks rIsSocketTimeoutrws- UFNN + 6 u!5 57rc`||nt}||nt}t}tj|dz}|j j j|j}tjdj|j5 |j|j|g|dzd} dddj$|j$j&5d j|j|j$j&}nd j|j}t)j$||S#tj$r4t!j"dj|j|wxYw#1swYxYw) aWait for a waiter to finish. Args: waiter_resource: The waiter resource to wait for. sleep: The number of seconds to sleep between status checks. max_wait: The maximum number of seconds to wait before an error is raised. Returns: The last retrieved value of the Waiter. Raises: WaitTimeoutError: If the wait operation takes longer than the maximum wait time. Ni) max_wait_msnamez"Waiting for waiter [{0}] to finishc|j Sr0)done)wss rr5zWaitForWaiter..s aff*r)r4sleep_msshould_retry_ifz/Waiter [{0}] did not finish within {1} seconds.z(Waiter [{0}] finished with an error: {1}z$Waiter [{0}] finished with an error.)DEFAULT_WAITER_SLEEPMAX_WAITER_TIMEOUTr_r RetryerrQMESSAGES_MODULE-RuntimeconfigProjectsConfigsWaitersGetRequest RelativeNamer ProgressTrackerformatName RetryOnResultrg WaitExceptionrtc_exceptionsWaitTimeoutErrorrlmessager )waiter_resourcesleepmax_wait waiter_clientretryerrequestresultrs r WaitForWaiterrs$%*>%!-X3E(.- MMho 6'  ! ! 1 1;;%224<6 ''*11/2F2F2HIK1$$]%6%6+2).3dl5L%NfK \\ ||':AA    &,,"6"68g7==    "gIIg -   1  + + ; B B""$h 0 111KKsF$ $EAF!!F$$F-c|jduS)z0Returns True if the specified waiter has failed.Nrk)r"s rIsFailedWaiterrs T !!rctj|}d|vr%d|vr td|d|d<||d|d<|S)aIReturns a dict representation of the message with a shortened name value. This method does three things: 1. converts message to a dict. 2. shortens the value of the name field using name_converter 3. sets atomicName to the original value of name. Args: message: A protorpclite message. name_converter: A function that takes an atomic name as a parameter and returns a shortened name. Returns: A dict representation of the message with a shortened name field. Raises: ValueError: If the original message already contains an atomicName field. r{ atomicNamez4Original message cannot contain an atomicName field.)r MessageToDict ValueError)rname_converter message_dicts r_DictWithShortNamers[&''0, ||# M NN!-f!5L),v*>?L rct|dS)z;Returns the config message as a dict with a shortened name.c*|jddSNrsplitrzs rr5zFormatConfig..$**S/"2Errrs r FormatConfigr G%E FFrcvt|d}|r)d|vr |d|d<|Stj|d|d<|S)aReturns the variable message as a dict with a shortened name. This method first converts the variable message to a dict with a shortened name and an atomicName. Then, decodes the variable value in the dict if the output_value flag is True. Args: message: A protorpclite message. output_value: A bool flag indicates whether we want to decode and output the values of the variables. The default value of this flag is False. Returns: A dict representation of the message with a shortened name field. cPdj|jdtdS)Nr)rrVARIABLE_NAME_PREFIX_LENGTHrzs rr5z FormatVariable..1s388DJJsO,G,HIJrtextvalue)rbase64 b64decode)r output_valuers rFormatVariablers]$$ JL,*62l7 %..|G/DEl7 rct|dS)z;Returns the waiter message as a dict with a shortened name.c*|jddSrrrzs rr5zFormatWaiter..Crrrrs r FormatWaiterr>rr)T)NN)F)4__doc__ __future__rrrrrsapitools.base.pyr%googlecloudsdk.api_lib.runtime_configrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioperFgooglecloudsdk.corer r r googlecloudsdk.core.consoler googlecloudsdk.core.utilr rurtrrrrrrr#r-r>rCr2rRrYr\r_rbr)rmrprwrrrrrrrUrrrs 9&' %N,@#*)8* ." )=* D 1"4*,+53<?447-`" BGBGr