dZddlmZddlmZddlmZddlZddlmZddl m Z ddl m Z ddl mZe jj fd Ze jj fd Ze jj fd Zej(e jj fd Ze jj fd ZGdde j.Zy)z2Utilities for calling the Composer Operations API.)absolute_import)division)unicode_literalsN)util)waiter)basecBtj|jS)N release_track)api_utilGetClientInstanceprojects_locations_operationsr s 6lib/googlecloudsdk/api_lib/composer/operations_util.py GetServicers  # #! ##@#@Act|jtj|j |j S)a(Calls the Composer Operations.Delete method. Args: operation_resource: Resource, the Composer operation resource to delete. release_track: base.ReleaseTrack, the release track of command. Will dictate which Composer client library will be used. Returns: Empty r name)rDeleter GetMessagesModule0ComposerProjectsLocationsOperationsDeleteRequest RelativeNameoperation_resourcer s rrr#sF - 0 7 7  }=77!..082 33rct|jtj|j |j S)aDCalls the Composer Operations.Get method. Args: operation_resource: Resource, the Composer operation resource to retrieve. release_track: base.ReleaseTrack, the release track of command. Will dictate which Composer client library will be used. Returns: Operation: the requested operation r r)rGetr r-ComposerProjectsLocationsOperationsGetRequestrrs rrr5sF - 0 4 4  }=44!..052 33rc tjtj|jt ||d||dS)aHLists Composer Operations across all locations. Uses a hardcoded list of locations, as there is way to dynamically discover the list of supported locations. Support for new locations will be aligned with Cloud SDK releases. Args: location_refs: [core.resources.Resource], a list of resource reference to locations in which to list operations. page_size: An integer specifying the maximum number of resources to be returned in a single list call. limit: An integer specifying the maximum number of operations to list. None if all available operations should be returned. release_track: base.ReleaseTrack, the release track of command. Will dictate which Composer client library will be used. Returns: list: a generator over Operations within the locations in `location_refs`. r operationsr)limitlocation_attribute)r AggregateListResultsr.ComposerProjectsLocationsOperationsListRequestr) location_refs page_sizer r s rListr&GsB.  & &  }=55}-  !!rc`tjt||j|ddy)aWaits for an operation to complete. Polls the operation at least every 15 seconds, showing a progress indicator. Returns when the operation has completed. Args: operation: Operation Message, the operation to poll message: str, a message to display with the progress indicator. For example, 'Waiting for deletion of [some resource]'. release_track: base.ReleaseTrack, the release track of command. Will dictate which Composer client library will be used. r i6i:) max_wait_mswait_ceiling_msN)rWaitFor_OperationPollerr) operationmessager s rWaitForOperationr.is( ..]3nn  !rcTeZdZdZej j ffd ZdZxZ S)r+z4 Class for polling Composer longrunning Operations. cDtt| t|dy)Nr c|S)N)xs rz+_OperationPoller.__init__..s1r)superr+__init__r)selfr __class__s rr6z_OperationPoller.__init__s D*/>rc|jrA|jr4tj|j|jj yy)NTF)doneerror command_utilOperationErrorrr-)r7r,s rIsDonez_OperationPoller.IsDones>~~ )))..*3//*A*AC C  r) __name__ __module__ __qualname____doc__r ReleaseTrackGAr6r> __classcell__)r8s@rr+r+~s<#'#4#4#7#7>rr+)rB __future__rrrsysgooglecloudsdk.api_lib.composerrr googlecloudsdk.api_lib.utilrgooglecloudsdk.callioper#googlecloudsdk.command_lib.composerr<rCrDrrrmaxsizer&r.CloudOperationPollerNoResourcesr+r2rrrNs9&' <.(D"..11A .2->->-A-A3$+/*;*;*>*>3({{((++!D8<7H7H7K7K!* v== r