dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd Z dd ZGd d e jZGddeZy)z@Utilities for dealing with long-running operations (simple uri).)absolute_import)division)unicode_literals) list_pager)apis)waiter) properties) resourcesc.tjd|S)Nml)rGetMessagesModule)versions 2lib/googlecloudsdk/api_lib/ml_engine/operations.pyr r s  g ..c2tjd||S)Nr )no_http)rGetClientInstance)rrs rrr!s  gw ??rc.eZdZdZfdZdZdZxZS)CloudMlOperationPollerzPoller for Cloud ML Engine operations API. This is necessary because the core operations library doesn't directly support simple_uri. c||_tt||jjj|jjjyN)clientsuperr__init__projects_operations)selfr __class__s rrzCloudMlOperationPoller.__init__,s@DK $0 .. ..0rc8|jj|Sr)rGetr operation_refs rPollzCloudMlOperationPoller.Poll2s ;;??= ))rc|Sr)r operations r GetResultz CloudMlOperationPoller.GetResult5s r)__name__ __module__ __qualname____doc__rr"r& __classcell__)rs@rrr%s 0 *rrc2eZdZdZddZdZdZdZd dZy) OperationsClientz9Client for operations service in the Cloud ML Engine API.cZt||_|jj|_yr)rrMESSAGES_MODULEmessages)rrs rrzOperationsClient.__init__<s#G,DKKK//DMrctj|jj|jj |j ddS)Nname operationspageSize)fieldbatch_size_attribute)r YieldFromListrrr0MlProjectsOperationsListRequest RelativeName)r project_refs rListzOperationsClient.List@sK  # # '' 55))+ 6 -'  ))rc|jjj|jj |j SNr2)rrrr0MlProjectsOperationsGetRequestr:r s rrzOperationsClient.GetHsB ;; * * . . 44++- 5 / 00rc|jjj|jj |j Sr>)rrCancelr0!MlProjectsOperationsCancelRequestr:r s rrAzOperationsClient.CancelMsB ;; * * 1 1 77++- 8 / 00rNc pt|}|j|r|Stjj |j dt jjjjid}|dj|j}tj|||dddddd S) a~Wait until the operation is complete or times out. Args: operation: The operation resource to wait on message: str, the message to print while waiting. Returns: The operation resource when it has completed Raises: OperationTimeoutError: when the operation polling times out OperationError: when the operation completed with an error projectsIdzml.projects.operations)params collectionNzWaiting for operation [{}]ri6i)pre_start_sleep_ms max_wait_msexponential_sleep_multiplier jitter_mswait_ceiling_mssleep_ms)rIsDoner REGISTRYParser3r VALUEScoreproject GetOrFailformatNamerWaitFor)rr%messagepollerr!s rWaitForOperationz!OperationsClient.WaitForOperationRs$D )F }}Y &&,,j//44<<FFG+--M,33M4F4F4HIg >> w%) rv1r) r'r(r)r*rr<rrArYr$rrr-r-9sA0)0 0 rr-NrZ)r[F)r* __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.utilrrgooglecloudsdk.corer r r rCloudOperationPollerrobjectr-r$rrrbsLG&'',.*)/@V88(8v8r