} dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z Gd d e jZ Gd d ejZd ZdZdZy)zCRM operations utilities.)absolute_import)division)unicode_literals)tags)waiter) exceptions) resourcesc eZdZy)OperationErrorN)__name__ __module__ __qualname__=lib/googlecloudsdk/command_lib/resource_manager/operations.pyr r srr ceZdZdZdZdZy)ReturnOperationPollerzPolls for operations that retrieve the operation rather than the resource. This is needed for Delete operations, where the response is Empty. It is also needed for services that do not have a Get* method, such as TagBindings. c||_y)zSets up poller for polling operations. Args: operation_service: apitools.base.py.base_api.BaseApiService, api service for retrieving information about ongoing operation. N)operation_service)selfrs r__init__zReturnOperationPoller.__init__&s /Drc|S)aNOverrides. Response for Deletion Operation is of type google.protobuf.Empty and hence we can return the operation itself as the result. For operations without a Get[Resource] method, we have no choice but to return the operation. Args: operation: api_name_messages.Operation. Returns: operation r)r operations r GetResultzReturnOperationPoller.GetResult/s  rN)r r r__doc__rrrrrrrs / rrcVttj}t|||S)aWaits for the given google.longrunning.Operation to complete. Args: operation: The operation to poll. message: String to display for default progress_tracker. Raises: apitools.base.py.HttpError: if the request returns an HTTP error Returns: operation )rrOperationsService_WaitForOperation)rmessagepollers rWaitForReturnOperationr!?s& !!7!7!9 :& 9gv 66rcltj|tj}t |||S)aWaits for the given google.longrunning.Operation to complete. Args: operation: The operation to poll. message: String to display for default progress_tracker. service: The service to get the resource after the long running operation completes. Raises: apitools.base.py.HttpError: if the request returns an HTTP error Returns: The TagKey or TagValue resource. )rCloudOperationPollerrrr)rrservicer s rWaitForOperationr%Ps.  & &w0F0F0H I& 9gv 66rc|j|r|j|Stjj |j d}t j|||S)Nzcloudresourcemanager.operations) collection)IsDonerr REGISTRYParsenamerWaitFor)rrr operation_refs rrrcsZ ]]9   I &&$$**nn!B+D-  w 77rN)r __future__rrr'googlecloudsdk.api_lib.resource_managerrgooglecloudsdk.api_lib.utilrgooglecloudsdk.corerr Errorr r#rr!r%rrrrr3sS &'8.*)Z%%F77@7"7&8r