SrSSKJr SSKJr SSKJr SSKrSSKJr SSK Jr SSK J r SSK J r SS K Jr SS KJr "S S \R"5rg) z*Waiter utility for api_lib.util.waiter.py.)absolute_import)division)unicode_literalsN) exceptions)util)cloud_console_url_helper)waiter)logc6\rSrSrSrSrSrSrSrSr Sr g ) RmBatchPollerzdPoller for resource manager batches. This should be used for spark version 3+, and Ray version 1+. cXlSUlg)NFdataprocfist_tick_message_printed)selfrs =lib/googlecloudsdk/api_lib/dataproc/poller/rm_batch_poller.py__init__RmBatchPoller.__init__%sM%*D"cfU(aURURRRRR URRRRR URRRRR4;agg)See base class.TF)statermessagesBatchStateValueValuesEnum SUCCEEDED CANCELLEDFAILED)rbatchs rIsDoneRmBatchPoller.IsDone)sz  $$99CC $$99CC $$99@@!  rclURRRUS9nURRRR U5$![ RaKn[R"SRU55 [R"U5(aeSnAgSnAff=f)r)namezGet Batch failed: {}N) rr*DataprocProjectsLocationsBatchesGetRequestclientprojects_locations_batchesGetapitools_exceptions HttpErrorr warningformatrIsClientHttpException)r batch_refrequesterrors rPollRmBatchPoller.Poll3smm$$OO PG ]] ! ! < < @ @ II  ( ( kk)0078 # #E * *  +s.AB3(AB..B3c,U(d[R"S5eURURRR R R:Xa[R"S5 g URURRR R R:Xa\SnUR(a)SRX!R5nUSS:waUS- nUS- nUS- n[R"U5eg ) a<Handles errors. Error handling for batch jobs. This happen after the batch reaches one of the complete states. Overrides. Args: batch: The batch resource. Returns: None. The result is directly output to log.err. Raises: JobTimeoutError: When waiter timed out. JobError: When remote batch job is failed. z&Timed out while waiting for batch job.zBatch job is CANCELLED.zBatch job is FAILED.z {} Detail: {}. zRunning auto diagnostics on the batch. It may take few minutes before diagnostics output is available. Please check diagnostics output by running 'gcloud dataproc batches describe' command.N)rJobTimeoutErrorrrrrrrr r+r stateMessager,JobError)rr err_messages r GetResultRmBatchPoller.GetResult@s$   & &'O PP  == ! ! ' ' < < F F G kk+,& #  t}}--33HHOOO*k   %,,[:L:LM r?c !  +Tk 'k    ,, rcUR(dSUl[R"U5n[R"U5n[R "5n[R "[R5 [R"SUU5 [R"S5 [R "U5 gg)z;Prints links to cloud console after the first success pull.TzPlease check the driver output in Cloud Logging: %s. (The log can take a few minutes to show up.) You can visit the batch resource at %sz"Waiting for the batch to complete.N) rrget_batch_logging_urlget_dataproc_batch_urlr GetVerbosity SetVerbosityloggingINFOinfo)rtracker poll_resultstatuscloud_logging_urldataproc_batch_url log_levels rTrackerUpdateFunction#RmBatchPoller.TrackerUpdateFunctionps  ) )'+d$2HH 4JJ ""$i w||$ hh     hh34 y!% *rrN) __name__ __module__ __qualname____firstlineno____doc__rr!r1r;rK__static_attributes__rrr r s! + .`"rr )rQ __future__rrrrBapitools.base.pyrr)googlecloudsdk.api_lib.dataprocr&googlecloudsdk.api_lib.dataproc.pollerrgooglecloudsdk.api_lib.utilr googlecloudsdk.corer OperationPollerr rSrrr[s< 1&'>60K.#e"F**e"r