dZddlmZddlmZddlmZddlZddlmZddl mZddl m Z ddl m Z dd l mZdd lmZGd d ej"Zy) z*Waiter utility for api_lib.util.waiter.py.)absolute_import)division)unicode_literalsN) exceptions)util)cloud_console_url_helper)waiter)logc.eZdZdZdZdZdZdZdZy) RmBatchPollerzhPoller for resource manager batches. This should be used for spark version 3+, and Ray version 1+. c ||_d|_y)NF)dataprocfist_tick_message_printed)selfrs =lib/googlecloudsdk/api_lib/dataproc/poller/rm_batch_poller.py__init__zRmBatchPoller.__init__%sDM%*D"cX|r|j|jjjjj |jjjjj |jjjjjfvryy)See base class.TF)statermessagesBatchStateValueValuesEnum SUCCEEDED CANCELLEDFAILED)rbatchs rIsDonezRmBatchPoller.IsDone)sz  $$99CC $$99CC $$99@@!  rcb|jjj|} |jjjj |S#t j$rD}tjdj|tj|rYd}~yd}~wwxYw)r)namezGet Batch failed: {}N) rr*DataprocProjectsLocationsBatchesGetRequestclientprojects_locations_batchesGetapitools_exceptions HttpErrorr warningformatrIsClientHttpException)r batch_refrequesterrors rPollzRmBatchPoller.Poll3smm$$OO PG ]] ! ! < < @ @ II  ( ( kk)0078 # #E *  +s.AB.*:B))B.c|stjd|j|jjj j jk(rtjdy |j|jjj j jk(rVd}|jr)dj||j}|ddk7r|dz }|dz }|dz }tj|y ) ahHandles 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 GetResultzRmBatchPoller.GetResult@s$   & &'O PP  == ! ! ' ' < < F F G kk+,& #  t}}--33HHOOO*k   %,,[%:L:LM r?c !  +Tk 'k    ,, rcp|jsd|_tj|}tj|}t j }t j tjt jd||t jdt j |yy)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 rTrackerUpdateFunctionz#RmBatchPoller.TrackerUpdateFunctionps  ) )'+d$2HH 4JJ ""$i w||$ hh     hh34 y!% *rN) __name__ __module__ __qualname____doc__rrr-r6rErrr r s! + .`"rr )rI __future__rrrr<apitools.base.pyrr%googlecloudsdk.api_lib.dataprocr&googlecloudsdk.api_lib.dataproc.pollerrgooglecloudsdk.api_lib.utilr googlecloudsdk.corer OperationPollerr rJrrrRs< 1&'>60K.#e"F**e"r