I dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd l mZGd d ej ZGd deZdZddZdZy)z9General utilities using operations in Privateca commands.)absolute_import)division)unicode_literals)encoding)base)messages)waiter) exceptions) resourcesceZdZdZy)OperationErrorz3Exception for errors encountered from an operation.N__name__ __module__ __qualname____doc__6lib/googlecloudsdk/command_lib/privateca/operations.pyr r s;rr ceZdZdZy)OperationTimeoutErrorz*Exception for when an operation times out.Nrrrrrr!s2rrcVtjj|jdS)z5Get a resource reference to a long running operation.z'privateca.projects.locations.operations)r REGISTRYParseRelativeNamename) operations rGetOperationRefr%s%    - -nn? rcv|jr7|jrt|jj|jSt |}t jtj|j} t j|||S#t j$r tdwxYw)ayWaits for operation to complete while displaying in-progress indicator. Args: operation: The Operation resource. progress_message: The message to display with the in-progress indicator. api_version: The API version. Returns: The resource that is the result of the operation. Raises: OperationError: if the operation did not complete successfully zRequested action timed out. Please run the describe command on your resource to see if changes were successful, or try again in a few minutes.)doneerrorr messageresponserr CloudOperationPollerNoResourcesrGetClientInstanceprojects_locations_operationsWaitFor TimeoutErrorr)rprogress_message api_version operation_refpollers rAwaitr,,s^^ 9??22 33   !),-  1 1 [)GG & >>&-1A BB      s BB8cjtj|}d|vr|d=tj||dS)aReturns a message from the ResponseValue. Operations normally return a ResponseValue object in their response field that is somewhat difficult to use. This functions returns the corresponding message type to make it easier to parse the response. Args: response: The ResponseValue object that resulted from an Operation. message_type: The type of the message that should be returned Returns: An instance of message_type with the values from the response filled in. z@typeF)throw_on_unexpected_fields)r MessageToDict messages_utilDictToMessageWithErrorCheck)r" message_type message_dicts rGetMessageFromResponser4Ms>''1, W  2 2LU rN)v1)r __future__rrrapitools.base.pyr googlecloudsdk.api_lib.privatecargooglecloudsdk.api_lib.utilrr0r googlecloudsdk.corer r Errorr rrr,r4rrrr<sT@&'%1A.*)