dZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z e jjfd Ze jj"fd Ze jjfd Ze jjfd Ze jjfd ZGdde j,ZGdde j,Zy)z3Utilities for calling the Metastore Operations API.)absolute_import)division)unicode_literals)util)waiter)base)logcBtj|jS)N release_track)api_utilGetClientInstanceprojects_locations_operationsr s 7lib/googlecloudsdk/api_lib/metastore/operations_util.py GetOperationrs  # #! !!"ct|jtj|j |S)aDCalls the Metastore Operations.Cancel method. Args: relative_resource_name: str, the relative resource name of the Metastore operation to cancel. release_track: base.ReleaseTrack, the release track of command. Will dictate which Dataproc Metastore client library will be used. Returns: Empty r name)rCancelr GetMessagesModule1MetastoreProjectsLocationsOperationsCancelRequestrelative_resource_namer s rrr!@ M 2 9 9  %99% :  rct|jtj|j |S)aDCalls the Metastore Operations.Delete method. Args: relative_resource_name: str, the relative resource name of the Metastore operation to delete. release_track: base.ReleaseTrack, the release track of command. Will dictate which Dataproc Metastore client library will be used. Returns: Empty r r)rDeleter r1MetastoreProjectsLocationsOperationsDeleteRequestrs rrr6rrc^tjt||j|ddS)aYWaits for an operation to complete and return it. Polls the operation at least every 15 seconds, showing a progress indicator. Returns when the operation has completed. The timeout periods of this operation is one hour. 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 Metastore client library will be used. Returns: poller.GetResult(operation). r 6: max_wait_mswait_ceiling_ms)rWaitFor_OperationPollerWithErrorr operationmessager s rPollAndReturnOperationr*Ks-& m<nn   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 Metastore client library will be used. r r r!r"N)rr%_OperationPollerrr's rWaitForOperationr-gs( ..]3nn  rcTeZdZdZej j ffd ZdZxZ S)r,z3Class for polling Metastore longrunning Operations.cDtt| t|dy)Nr c|SNxs rz+_OperationPoller.__init__..Qr)superr,__init__rselfr __class__s rr8z_OperationPoller.__init__s D*=1;rc|jsy|jr4tj|j|jj y)NFT)doneerrorr OperationErrorrr))r:r(s rIsDonez_OperationPoller.IsDones7 >>   # #INNIOO4K4K LL r __name__ __module__ __qualname____doc__r ReleaseTrackGAr8r@ __classcell__r;s@rr,r,}s;#'#4#4#7#7 rr,cTeZdZdZej j ffd ZdZxZ S)r&zDClass for polling Metastore longrunning Operations and print errors.cDtt| t|dy)Nr c|Sr1r2r3s rr5z4_OperationPollerWithError.__init__..r6r)r7r&r8rr9s rr8z"_OperationPollerWithError.__init__s #T3=1;rc|jsy|jrW|jjr4tjj d|jj|jj r4tjj d|jj |jjr|jjdjD]g}|jdk(stjj d|jjjdjjitj|j dy) NFz Status Code:zError message:rdetailszError details:T)r=r>coder statusPrintr)rNadditionalPropertieskeyvalue object_value properties string_valuer r?r)r:r(r)s rr@z _OperationPollerWithError.IsDones >>     )=)=>   )9??+B+BC   ..q1FFG [[I % JJ    **55a8>>KK G  # #INNB 77 rrArIs@rr&r&sL#'#4#4#7#7 rr&N)rE __future__rrr googlecloudsdk.api_lib.metastorerr googlecloudsdk.api_lib.utilrgooglecloudsdk.calliopergooglecloudsdk.corer rFrGrALPHArrr*r-CloudOperationPollerNoResourcesr,r&r2rrr`s:&'=.(# $0033" 261B1B1H1H*261B1B1E1E,'+&7&7&:&:88<7H7H7K7K, v==   F Fr