)dZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z dd l mZddlZd ZGd d e j"ZGd de j"ZGdde j"ZGddej*ej,eZGddeZGddeZ ddZ ddZdZy)z-Utilities to support long running operations.)absolute_import)division)unicode_literalsN)encoding) exceptions)progress_tracker)retryzThe operations 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.c eZdZy) TimeoutErrorN__name__ __module__ __qualname__)lib/googlecloudsdk/api_lib/util/waiter.pyr r &rr c eZdZy)AbortWaitErrorNr rrrrr*rrrc eZdZy)OperationErrorNr rrrrr.rrrc|eZdZdZej dZej dZej dZy)OperationPolleraInterface for defining operation which can be polled and waited on. This construct manages operation_ref, operation and result abstract objects. Operation_ref is an identifier for operation which is a proxy for result object. OperationPoller has three responsibilities: 1. Given operation object determine if it is done. 2. Given operation_ref fetch operation object 3. Given operation object fetch result object cy)zGiven result of Poll determines if result is done. Args: operation: object representing operation returned by Poll method. Returns: Trself operations rIsDonezOperationPoller.IsDone= rcy)zRetrieves operation given its reference. Args: operation_ref: str, some id for operation. Returns: object which represents operation. Nr)r operation_refs rPollzOperationPoller.PollIrrcy)zGiven operation message retrieves result it represents. Args: operation: object, representing operation returned by Poll method. Returns: some object created by given operation. Nrrs r GetResultzOperationPoller.GetResultUs rN) r rr__doc__abcabstractmethodrr"r$rrrrr2sU    rrc(eZdZdZdZdZdZdZy)CloudOperationPollerzlManages a longrunning Operations. See https://cloud.google.com/speech/reference/rpc/google.longrunning c ||_||_y)aSets up poller for cloud operations. Args: result_service: apitools.base.py.base_api.BaseApiService, api service for retrieving created result of initiated operation. operation_service: apitools.base.py.base_api.BaseApiService, api service for retrieving information about ongoing operation. Note that result_service and operation_service Get request must have single attribute called 'name'. N)result_serviceoperation_service)rr+r,s r__init__zCloudOperationPoller.__init__gs)D.Drct|jr,|jrt|jjyyz Overrides.TFdoneerrorrmessagers rrzCloudOperationPoller.IsDonev+~~ Y__4455  rc|jjd}|jj||jSzOverrides. Args: operation_ref: googlecloudsdk.core.resources.Resource. Returns: fetched operation message. Getname)r,GetRequestTyper7 RelativeNamerr! request_types rr"zCloudOperationPoller.Poll~sD))88?L  ! ! % %-4467 99rc|jjd}tj|j}|jj ||dS)zOverrides. Args: operation: api_name_messages.Operation. Returns: result of result_service.Get request. r7r9r8)r+r:rMessageToPyValueresponser7)rrr= response_dicts rr$zCloudOperationPoller.GetResultsQ&&55ez:CloudOperationPollerNoResources.__init__..s 0@rN)r,get_name)rr, get_name_funcs rr-z(CloudOperationPollerNoResources.__init__s/D!A&@DMrct|jr,|jrt|jjyyr/r0rs rrz&CloudOperationPollerNoResources.IsDoner4rc|jjd}|jj||j|Sr6)r,r:r7rJr<s rr"z$CloudOperationPollerNoResources.PollsD))88?L  ! ! % %$-- 67 99rc|jS)zOverrides to get the response from the completed operation. Args: operation: api_name_messages.Operation. Returns: the 'response' field of the Operation. )r@rs rr$z)CloudOperationPollerNoResources.GetResults   rrGrBrrrrDrDsB" 9 rrDc 0dj|} |stj|| n|5|r t|fd} t |||||| | | | }ddd|jS#1swYxYw#t j $r$tdj||dztt j$rQ}tdj||jjdz|jjtd}~wwxYw)aAWaits for poller.Poll and displays pending operation spinner. Args: poller: OperationPoller, poller to use during retrials. operation_ref: object, passed to operation poller poll method. message: str, string to display for default progress_tracker. custom_tracker: ProgressTracker, progress_tracker to use for display. tracker_update_func: func(tracker, result, status), tracker update function. pre_start_sleep_ms: int, Time to wait before making first poll request. max_retrials: int, max number of retrials before raising RetryException. max_wait_ms: int, number of ms to wait before raising WaitException. exponential_sleep_multiplier: float, factor to use on subsequent retries. jitter_ms: int, random (up to the value) additional sleep between retries. wait_ceiling_ms: int, Maximum wait between retries. sleep_ms: int or iterable: for how long to wait between trials. Returns: poller.GetResult(operation). Raises: AbortWaitError: if ctrl-c was pressed. TimeoutError: if retryer has finished without being done. z!Aborting wait for operation {0}. )aborted_messagec@r ||yjyrG)Tick)resultstatustrackertracker_update_funcs r _StatusUpdatezWaitFor.._StatusUpdates  gvv 6 ,,.rNz2Operation {0} has not finished in {1} seconds. {2}zIOperation {0} has not finished in {1} seconds after max {2} retrials. {3})formatrProgressTracker_SleepMs PollUntilDoner WaitExceptionr _TIMEOUT_MESSAGEMaxRetrialsExceptionstatetime_passed_msretrialr$)pollerr!r3custom_trackerrVpre_start_sleep_ms max_retrials max_wait_msexponential_sleep_multiplier jitter_mswait_ceiling_mssleep_msrPrWrerUs ` @rWaitForrms!F9?? N/#  ) ) !/ 03: #$  -{ & ? M#i 0>   ) $$? 0 0$   G <  {d24D E GG  # ##  &  &&$.  " ###s0A<'A0A<0A95A<._IsNotDone?s}}Y' ''r)funcargsshould_retry_ifrk)r Retryer RetryOnResultr") rcr!rfrgrhrirjrk status_updateretryerrrrs ` rr\r\s\: MM#?%&  ('(## ;;    $) rc4tj|dz y)NrX)timesleep) milisecondss rr[r[Ks**[4  r) NNNrXN@wffffff?rX )Nr~rrXrrN)r% __future__rrrr&r{apitools.base.pyrgooglecloudsdk.corergooglecloudsdk.core.consolergooglecloudsdk.core.utilr sixr^Errorr rrwith_metaclassABCMetaobjectrr)rDrmr\r[rrrrs 4&' %*8* F :##Z%%Z%%,(c((f=,^5M?5Mp7o7x $#),"D%P $%/2 "( $.b!r