3dZddlmZddlmZddlmZddlmZddlm Z ddlmZddlm Z ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZGddeZdZGddeZGddeZdZy)z"Utilities for Cloud Workflows API.)absolute_import)division)unicode_literals) exceptions)apis)waiter)cache) poller_utils) arg_parsers)base) arg_utils) labels_util)flags) resourcesceZdZdZy)UnsupportedReleaseTrackErrorzDRaised when requesting API version for an unsupported release track.N)__name__ __module__ __qualname____doc__1lib/googlecloudsdk/api_lib/workflows/workflows.pyrr#sLrrc|tjjk(ry|tjjk(ry|tjjk(ryt |)Nv1alpha1v1betav1)r ReleaseTrackALPHABETAGAr) release_tracks rReleaseTrackToApiVersionr#'sRd''--- ))... )),,,  &} 55rc4eZdZdZdZdZdZdZdZdZ y) WorkflowsClientz8Client for Workflows service in the Cloud Workflows API.ctjd||_|jj|_|jj |_y)N workflows)rGetClientInstanceclientMESSAGES_MODULEmessagesprojects_locations_workflows_serviceself api_versions r__init__zWorkflowsClient.__init__5s8((kBDKKK//DMKK< create_reqs rrCzWorkflowsClient.CreateKsYOO""$113$$&PJ ==   ++rcdjt|}|jj|j ||}|j j |S)aYUpdates a Workflow. If updated fields are specified it uses patch semantics. Args: workflow_ref: Resource reference to the Workflow to update. workflow: Workflow resource message to update. updated_fields: List of the updated fields used in a patch request. Returns: Long-running operation for update. ,)r4 updateMaskr>)joinsortedr+/WorkflowsProjectsLocationsWorkflowsPatchRequestr6r-Patch)r/r:r>updated_fields update_mask patch_reqs rrKzWorkflowsClient.Patch\sZ((6.12K MM  & & (NI ==  y ))rc|jj}g}tj|||tj|||tj |||t j||jjj}tj||||tjjk(rtj|||d}|jdr5t j||jjjd}|jdrt!|j"tj$kDr3t'j(dj+tj$t j||jjjd}|jdrtj,}tj.|||d}|jdrS|j0j2Dcic]}|j4|j6}}|j9|j:|jdr|j0j2Dcic]}|j4|j6}}|j<D]:} | |vr|| = t&j>jAdd j+|  tjB||||jd r[|jjjD} tGjH|jJ| gd } tjL| |||jdr[|jjjN} tGjH|jP| gd} tjR| |||jdr#|!t&j>jAdd tjT|||jjjV||fScc}wcc}w)aCreates a workflow from command-line arguments. Args: args: The arguments of the gcloud command. old_workflow: The workflow from previous revision. release_track: The gcloud release track used in the command. Returns: workflow: The consturcted Workflow message from the passed in arguments. updated_fields: The workflow fields that are updated. N set_env_vars env_vars_filez3too many environment variables, limit is {max_len}.)max_lenclear_env_varsupdate_env_varsremove_env_varszkey {k} is not found.)k)argumentmessagecall_log_levelnonez log-all-callszlog-errors-onlyzlog-none valid_choicesexecution_history_levelr[zexecution-history-basiczexecution-history-detailedtagsz/tags cannot be updated for an existing workflow),r+Workflowr SetSourceSetDescriptionSetServiceAccountrParseCreateArgs LabelsValue SetLabelsr rr! SetKmsKey IsSpecifiedUserEnvVarsValuelenrQUSER_ENV_VARS_LIMITr ArgumentTypeErrorformatCLEAR_ENVIRONMENTSetUserEnvVars userEnvVarsadditionalPropertieskeyvalueupdaterTrUargparse ArgumentErrorUpdateUserEnvVarsCallLogLevelValueValuesEnumr ChoiceToEnumrYSetWorkflowLoggingArg$ExecutionHistoryLevelValueValuesEnumr^#SetWorkflowExecutionHistoryLevelArgSetWorkflowsTagsArg TagsValue)r/args old_workflowr"r>rLlabelsenv_varspvcall_log_level_enum log_levelexecution_history_level_enum history_levels rBuildWorkflowFromArgsz%WorkflowsClient.BuildWorkflowFromArgsqs}}%%'HN OOD(N3 x8 D(N;  ( ( dmm$$00F OOFHn5)),,, oodHn5h  . ).. $--((99>    / * t!! "U%>%> >--CJJ33K  .. $--((99?    * +** 8X~>h  + ,"--BB B EE177NB   ,,-  + ,"--BB B EE177NB  %%A (] &&44/666;5  & h.A  * +"mm44PP**       ##IxH  3 4 MM " " G G %"..  ( ( (  11 8^ 5  & !l&>""00E1    $--00:: ^ ##q  s ,QQcHtjj|jd}t j |j |j}t j|||}dj|j}tj|||S)z9Waits until the given long-running operation is complete.z'workflows.projects.locations.operations) collection)r' operationsr:z&Waiting for operation [{}] to complete) rREGISTRYParser4r OperationsClientr)r+WorkflowsOperationPollerrnrBrWaitFor)r/ operationr: operation_refrpollerprogress_strings rWaitForOperationz WorkflowsClient.WaitForOperations&&,,<-M..t{{DMMJJ  2 2!F ?EEO >>&- AArN) rrrrr1r7rCrKrrrrrr%r%2s&@= ","**g$RBrr%c2eZdZdZdZ ddZdZdZy)WorkflowExecutionClientzLClient for Workflows Execution service in the Cloud Workflows Execution API.ctjd||_|jj|_|jj |_y)Nworkflowexecutions)rr(r)r*r+'projects_locations_workflows_executionsr-r.s rr1z WorkflowExecutionClient.__init__s9(()={KDKKK//DMKKGGDMrNcr|jj}||_|rd|_|||_|D|dk7r?|jjj }t j||gd|_|D|dk7r?|jjj} t j|| gd|_ |jj|j|} |jj| S#tj $r} t#j$| d d} ~ wwxYw) aCreates a Workflow execution. Args: workflow_ref: Resource reference to the Workflow to execute. data: Argments to use for executing the workflow. call_log_level: Level of call logging to apply during execution. execution_history_level: Level of execution history to apply for the execution. labels: Labels associated to the execution. overflow_buffering_disabled: If set to true, the execution will not be backlogged when the concurrency quota is exhausted. Backlogged executions start when the concurrency quota becomes available. Returns: Execution: The workflow execution. TNr[rZr\r_)r= execution {message} error_format)r+ ExecutionrW(disableConcurrencyQuotaOverflowBufferingrryr rz callLogLevelr|executionHistoryLevelCWorkflowexecutionsProjectsLocationsWorkflowsExecutionsCreateRequestr6r-rCr8 HttpErrorr HttpException) r/r:datarYr^roverflow_buffering_disabledrrrrDes rrCzWorkflowExecutionClient.Creates22 '')II";?i8 i!n&> MM33OO(55    i + #v - -- ! ! F F#)2(>(> ! & )i%bb((*icJB ]] ! !* --  # #B  $ $Q[ AABs,DD6D11D6c|tj}|jj|j } |j j |S#tj$r}tj|dd}~wwxYw)zGets a workflow execution. Args: execution_ref: Resource reference to the Workflow execution to get. Returns: Workflow: The workflow execution if it exists, an error exception otherwise. Nr3rr) r get_cached_execution_idr+@WorkflowexecutionsProjectsLocationsWorkflowsExecutionsGetRequestr6r-r7r8rrr)r/ execution_refr;rs rr7zWorkflowExecutionClient.Get7s335mmm\\  ' ' )]GB ]]  w ''  # #B  $ $Q[ AABsAB 0BB c |tj}tj|}dj |j } t j|||ddddS#t j$r*t jdj |tt j$r%t jd j |wxYw) zPWaits until the given execution is complete or the maximum wait time is reached.)workflow_executionz&Waiting for execution [{}] to completedi\&g?i`)pre_start_sleep_ms max_wait_msexponential_sleep_multiplierwait_ceiling_msz/Execution {0} has not finished in 24 hours. {1}z Aborting wait for execution {0}.) r rr ExecutionsPollerrnrBrr TimeoutError_TIMEOUT_MESSAGEAbortWaitError)r/rrrs rWaitForExecutionz(WorkflowExecutionClient.WaitForExecutionLs335m  * *d CF>EEO ^^    '+!!       ; B B-      ! ! , 3 3M B s A))A3C)NNNF)rrrrr1rCr7rrrrrrs,TH" "'ABFB*rrzThe execution may still be underway remotely and may still succeed; use gcloud list and describe commands or https://console.developers.google.com/ to check resource state.N) r __future__rrrapitools.base.pyrr8googlecloudsdk.api_lib.utilrr googlecloudsdk.api_lib.workflowsr r googlecloudsdk.callioper r $googlecloudsdk.command_lib.util.apisr $googlecloudsdk.command_lib.util.argsr$googlecloudsdk.command_lib.workflowsrgooglecloudsdk.corer Exceptionrr#objectr%rrrrrrsu)&'9,2.29/(:<6)M9M6wBfwBt|f|BFr