dZddlmZddlmZddlmZddlmZddlmZddlm Z dd Z Gd d ejZ Gd d e Zy)z@Utilities for dealing with long-running operations (simple uri).)absolute_import)division)unicode_literals)apis)waiter) constantsNcNtjtj||S)N)no_http)rGetClientInstancerAI_PLATFORM_API_NAME) api_versionr s +lib/googlecloudsdk/api_lib/ai/operations.pyr r s$  $$k7 DDc.eZdZdZfdZdZdZxZS)AiPlatformOperationPollerzPoller for AI Platform operations API. This is necessary because the core operations library doesn't directly support simple_uri. c||_tt||jjj|jjjyN)clientsuperr__init__projects_locations_operations)selfr __class__s rrz"AiPlatformOperationPoller.__init__&s@DK #T3 88 88:rc8|jj|Sr)rGetr operation_refs rPollzAiPlatformOperationPoller.Poll,s ;;??= ))rc|Sr)r operations r GetResultz#AiPlatformOperationPoller.GetResult/s r)__name__ __module__ __qualname____doc__rrr" __classcell__)rs@rrrs : *rrcBeZdZdZddej fdZdZ ddZy)OperationsClientz9High-level client for the AI Platform operations surface.Nc|xsttj||_|xs|jj|_yr)r rAI_PLATFORM_API_VERSIONrMESSAGES_MODULEmessages)rrr-versions rrzOperationsClient.__init__6s<4-))'24DK; ; ;DMrc|jjj|jj |j S)N)name)rrrr-/AiplatformProjectsLocationsOperationsGetRequest RelativeNamers rrzOperationsClient.Get>sC ;; 4 4 8 8 EE++- F / 00rct|}|j|r|S|dj|j}t j ||||S)aMWait until the operation is complete or times out. Args: operation: The operation resource to wait on operation_ref: The operation reference to the operation resource. It's the result by calling resources.REGISTRY.Parse message: str, the message to print while waiting. max_wait_ms: int, number of ms to wait before raising WaitException. Returns: The operation resource when it has completed Raises: OperationTimeoutError: when the operation polling times out OperationError: when the operation completed with an error zWaiting for operation [{}]) max_wait_ms)rIsDoneformatNamerWaitFor)rr!rmessager4pollers rWaitForOperationz!OperationsClient.WaitForOperationCsX&'t ,F }}Y ,33M4F4F4HIg >> wK r)Ni@w) r#r$r%r&r BETA_VERSIONrrr;r rrr)r)3s,A --<0 AHrr))NF)r& __future__rrrgooglecloudsdk.api_lib.utilrrgooglecloudsdk.command_lib.airr CloudOperationPollerrobjectr)r rrrBsAG&',.3D  ; ;(+v+r