JdZddlmZddlmZddlmZddlmZ ddZy) z.Utilities for AI Platform operations commands.)absolute_import)division)unicode_literals)logNctjtjtjd}|r*|j ||||j ||S|j ||||jS)alWaits for an operation if asynchronous flag is off. Args: operations_client: api_lib.ai.operations.OperationsClient, the client via which to poll. op: Cloud AI Platform operation, the operation to poll. op_ref: The operation reference to the operation resource. It's the result by calling resources.REGISTRY.Parse asynchronous: bool, whether to wait for the operation or return immediately log_method: Logging method used for synchronous operation. If None, no log message: str, the message to display while waiting for the operation. kind: str, the resource kind (instance, cluster, project, etc.), which will be passed to logging function. max_wait_ms: int, number of ms to wait before raising WaitException. Returns: The result of the operation if asynchronous is true, or the Operation message otherwise )createdeleteupdate)kind)message max_wait_ms)rCreatedResourceDeletedResourceUpdatedResourcegetnameWaitForOperationresponse) operations_clientopop_ref asynchronous log_methodr r r logging_functions 4lib/googlecloudsdk/command_lib/ai/operations_util.pyWaitForOpMaybers~<###### J'3"z"2776 I  + +&'{ ,  H )FNNNi@w)__doc__ __future__rrrgooglecloudsdk.corerrrrr"s,5&'#  ( r