dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl mZGd d e jZy ) z*Waiter utility for api_lib.util.waiter.py.)absolute_import)division)unicode_literals) exceptions)util)"abstract_operation_streamer_poller)logc(eZdZdZdZdZdZdZy)GceBatchPollerz$Poller for GCE based batch workload.cX|r|j|jjjjj |jjjjj |jjjjjfvryy)See base class.TF)statedataprocmessagesBatchStateValueValuesEnum SUCCEEDED CANCELLEDFAILEDselfbatchs >lib/googlecloudsdk/api_lib/dataproc/poller/gce_batch_poller.pyIsDonezGceBatchPoller.IsDone$sz  $$99CC $$99CC $$99@@!  cb|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 rPollzGceBatchPoller.Poll.smm$$OO PG ]] ! ! < < @ @ II  ( ( kk)0078 # #E *  +s.AB.*:B))B.c|stjd|j|jjj j jk(rN|jstjdy |jjrtjdy |j|jjj j jk(rtjdy d}|jr)dj||j}|ddk7r|dz }|d z }|d z }tj|) 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.z$Expected batch job output not found.z:Batch job terminated, but output did not finish streaming.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)rJobTimeoutErrorrrrrrrdriver_log_streamerr r$openr stateMessager%JobError)rr err_messages r _GetResultzGceBatchPoller._GetResult;s7$   & &'O PP  == ! ! ' ' < < F F G % % :;6 5  # # ( ( H 0 )  == ! ! ' ' < < F F G kk+," +k   %,,[%:L:LM r?c !  +Tk 'k    ,,rcx|r8|jr,|jjr|jjSy)r N) runtimeInfo outputUrirs r _GetOutputUrizGceBatchPoller._GetOutputUrits0 ""u'8'8'B'B    ( (( rN)__name__ __module__ __qualname____doc__rr*r5r9rrr r s- 7rrr N)r= __future__rrrapitools.base.pyrr"googlecloudsdk.api_lib.dataprocr&googlecloudsdk.api_lib.dataproc.pollerrdataproc_poller_basegooglecloudsdk.corer AbstractOperationStreamerPollerr r>rrrFs; 1&'>60$Y88Yr